mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-15 19:05:11 +00:00
5d3e7d5aaa
Instead of skipping unsigned tool calls entirely (which lobotomizes context), convert them to text with an explicit note telling the model this is historical context from a different model and not a format to mimic. This preserves tool call/result context when switching from providers without thought signatures (e.g. Claude via Antigravity) to Gemini 3.
72 lines
2 KiB
TypeScript
72 lines
2 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { convertMessages } from "../src/providers/google-shared.js";
|
|
import type { Context, Model } from "../src/types.js";
|
|
|
|
describe("google-shared convertMessages", () => {
|
|
it("converts unsigned tool calls to text for Gemini 3", () => {
|
|
const model: Model<"google-generative-ai"> = {
|
|
id: "gemini-3-pro-preview",
|
|
name: "Gemini 3 Pro Preview",
|
|
api: "google-generative-ai",
|
|
provider: "google",
|
|
baseUrl: "https://generativelanguage.googleapis.com",
|
|
reasoning: true,
|
|
input: ["text"],
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
|
|
contextWindow: 128000,
|
|
maxTokens: 8192,
|
|
};
|
|
|
|
const now = Date.now();
|
|
const context: Context = {
|
|
messages: [
|
|
{ role: "user", content: "Hi", timestamp: now },
|
|
{
|
|
role: "assistant",
|
|
content: [
|
|
{
|
|
type: "toolCall",
|
|
id: "call_1",
|
|
name: "bash",
|
|
arguments: { command: "ls -la" },
|
|
// No thoughtSignature: simulates Claude via Antigravity.
|
|
},
|
|
],
|
|
api: "google-gemini-cli",
|
|
provider: "google-antigravity",
|
|
model: "claude-sonnet-4-20250514",
|
|
usage: {
|
|
input: 0,
|
|
output: 0,
|
|
cacheRead: 0,
|
|
cacheWrite: 0,
|
|
totalTokens: 0,
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
|
|
},
|
|
stopReason: "stop",
|
|
timestamp: now,
|
|
},
|
|
],
|
|
};
|
|
|
|
const contents = convertMessages(model, context);
|
|
|
|
let toolTurn: (typeof contents)[number] | undefined;
|
|
for (let i = contents.length - 1; i >= 0; i -= 1) {
|
|
if (contents[i]?.role === "model") {
|
|
toolTurn = contents[i];
|
|
break;
|
|
}
|
|
}
|
|
|
|
expect(toolTurn).toBeTruthy();
|
|
expect(toolTurn?.parts?.some((p) => p.functionCall !== undefined)).toBe(false);
|
|
|
|
const text = toolTurn?.parts?.map((p) => p.text ?? "").join("\n");
|
|
// Should contain historical context note to prevent mimicry
|
|
expect(text).toContain("Historical context");
|
|
expect(text).toContain("bash");
|
|
expect(text).toContain("ls -la");
|
|
expect(text).toContain("Do not mimic this format");
|
|
});
|
|
});
|