getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-21 14:05:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clean up stream tests

Browse source
This commit is contained in:
Mario Zechner 2025-12-30 01:08:49 +01:00
parent ecd240f636
commit 2601c26761
1 changed files with 5 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

15
packages/ai/test/stream.test.ts
View file

@ -752,13 +752,11 @@ describe("Generate E2E Tests", () => {
const llm = getModel("google-gemini-cli", "gemini-3-flash-preview"); const llm = getModel("google-gemini-cli", "gemini-3-flash-preview");
it.skipIf(!geminiCliToken)("should handle thinking with thinkingLevel", { retry: 3 }, async () => { it.skipIf(!geminiCliToken)("should handle thinking with thinkingLevel", { retry: 3 }, async () => {
const { ThinkingLevel } = await import("@google/genai"); await handleThinking(llm, { apiKey: geminiCliToken, thinking: { enabled: true, level: "LOW" } });
await handleThinking(llm, { apiKey: geminiCliToken, thinking: { enabled: true, level: ThinkingLevel.LOW } });
}); });
it.skipIf(!geminiCliToken)("should handle multi-turn with thinking and tools", { retry: 3 }, async () => { it.skipIf(!geminiCliToken)("should handle multi-turn with thinking and tools", { retry: 3 }, async () => {
const { ThinkingLevel } = await import("@google/genai"); await multiTurn(llm, { apiKey: geminiCliToken, thinking: { enabled: true, level: "MEDIUM" } });
await multiTurn(llm, { apiKey: geminiCliToken, thinking: { enabled: true, level: ThinkingLevel.MEDIUM } });
}); });
}); });
@ -778,17 +776,15 @@ describe("Generate E2E Tests", () => {
}); });
it.skipIf(!antigravityToken)("should handle thinking with thinkingLevel", { retry: 3 }, async () => { it.skipIf(!antigravityToken)("should handle thinking with thinkingLevel", { retry: 3 }, async () => {
const { ThinkingLevel } = await import("@google/genai");
// gemini-3-flash supports all four levels: MINIMAL, LOW, MEDIUM, HIGH // gemini-3-flash supports all four levels: MINIMAL, LOW, MEDIUM, HIGH
await handleThinking(llm, { await handleThinking(llm, {
apiKey: antigravityToken, apiKey: antigravityToken,
thinking: { enabled: true, level: ThinkingLevel.LOW }, thinking: { enabled: true, level: "LOW" },
}); });
}); });
it.skipIf(!antigravityToken)("should handle multi-turn with thinking and tools", { retry: 3 }, async () => { it.skipIf(!antigravityToken)("should handle multi-turn with thinking and tools", { retry: 3 }, async () => {
const { ThinkingLevel } = await import("@google/genai"); await multiTurn(llm, { apiKey: antigravityToken, thinking: { enabled: true, level: "MEDIUM" } });
await multiTurn(llm, { apiKey: antigravityToken, thinking: { enabled: true, level: ThinkingLevel.MEDIUM } });
}); });
it.skipIf(!antigravityToken)("should handle image input", { retry: 3 }, async () => { it.skipIf(!antigravityToken)("should handle image input", { retry: 3 }, async () => {
@ -800,11 +796,10 @@ describe("Generate E2E Tests", () => {
const llm = getModel("google-antigravity", "gemini-3-pro-high"); const llm = getModel("google-antigravity", "gemini-3-pro-high");
it.skipIf(!antigravityToken)("should handle thinking with thinkingLevel HIGH", { retry: 3 }, async () => { it.skipIf(!antigravityToken)("should handle thinking with thinkingLevel HIGH", { retry: 3 }, async () => {
const { ThinkingLevel } = await import("@google/genai");
// gemini-3-pro only supports LOW/HIGH // gemini-3-pro only supports LOW/HIGH
await handleThinking(llm, { await handleThinking(llm, {
apiKey: antigravityToken, apiKey: antigravityToken,
thinking: { enabled: true, level: ThinkingLevel.HIGH }, thinking: { enabled: true, level: "HIGH" },
}); });
}); });
}); });