Mario Zechner
ba93da9a93
Stream implementations move from src/providers/ to src/api/, renamed by API id (anthropic.ts -> anthropic-messages.ts, google.ts -> google-generative-ai.ts, mistral.ts -> mistral-conversations.ts, amazon-bedrock.ts -> bedrock-converse-stream.ts). Every module now exports exactly stream/streamSimple; shared helpers move alongside. New ProviderStreams dispatch contract in types.ts, lazyApi() wrapper in api/lazy.ts, and one .lazy.ts wrapper per API. Bedrock's wrapper keeps the node-only variable-specifier import and setBedrockProviderModule() (now taking ProviderStreams). providers/register-builtins.ts deleted; interim until the compat entrypoint lands, builtin api-registry registration lives in stream.ts and lazy wrappers are exported from the root barrel. Old per-API lazy exports (streamAnthropic, ...) are gone; package.json subpaths retarget to dist/api/.
39 lines
1.7 KiB
TypeScript
39 lines
1.7 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { isThinkingPart, retainThoughtSignature } from "../src/api/google-shared.ts";
|
|
|
|
describe("Google thinking detection (thoughtSignature)", () => {
|
|
it("treats part.thought === true as thinking", () => {
|
|
expect(isThinkingPart({ thought: true, thoughtSignature: undefined })).toBe(true);
|
|
expect(isThinkingPart({ thought: true, thoughtSignature: "opaque-signature" })).toBe(true);
|
|
});
|
|
|
|
it("does not treat thoughtSignature alone as thinking", () => {
|
|
// Per Google docs, thoughtSignature is for context replay and can appear on any part type.
|
|
// Only thought === true indicates thinking content.
|
|
// See: https://ai.google.dev/gemini-api/docs/thought-signatures
|
|
expect(isThinkingPart({ thought: undefined, thoughtSignature: "opaque-signature" })).toBe(false);
|
|
expect(isThinkingPart({ thought: false, thoughtSignature: "opaque-signature" })).toBe(false);
|
|
});
|
|
|
|
it("does not treat empty/missing signatures as thinking if thought is not set", () => {
|
|
expect(isThinkingPart({ thought: undefined, thoughtSignature: undefined })).toBe(false);
|
|
expect(isThinkingPart({ thought: false, thoughtSignature: "" })).toBe(false);
|
|
});
|
|
|
|
it("preserves the existing signature when subsequent deltas omit thoughtSignature", () => {
|
|
const first = retainThoughtSignature(undefined, "sig-1");
|
|
expect(first).toBe("sig-1");
|
|
|
|
const second = retainThoughtSignature(first, undefined);
|
|
expect(second).toBe("sig-1");
|
|
|
|
const third = retainThoughtSignature(second, "");
|
|
expect(third).toBe("sig-1");
|
|
});
|
|
|
|
it("updates the signature when a new non-empty signature arrives", () => {
|
|
const updated = retainThoughtSignature("sig-1", "sig-2");
|
|
expect(updated).toBe("sig-2");
|
|
});
|
|
});
|