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

Add handleImage and multiTurn tests for vercel-ai-gateway in stream.test.ts

Browse source
This commit is contained in:
Mario Zechner 2026-01-13 16:56:48 +01:00
parent 19f3c23f6d
commit 1d8d5d3d88
2 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -599,7 +599,7 @@ describe("Generate E2E Tests", () => {
}); });
describe.skipIf(!process.env.AI_GATEWAY_API_KEY)( describe.skipIf(!process.env.AI_GATEWAY_API_KEY)(
"Vercel AI Gateway Provider (google/gemini-2.5-flash via OpenAI Completions)", "Vercel AI Gateway Provider (google/gemini-2.5-flash via Anthropic Messages)",
() => { () => {
const llm = getModel("vercel-ai-gateway", "google/gemini-2.5-flash"); const llm = getModel("vercel-ai-gateway", "google/gemini-2.5-flash");
@ -614,6 +614,14 @@ describe("Generate E2E Tests", () => {
it("should handle streaming", { retry: 3 }, async () => { it("should handle streaming", { retry: 3 }, async () => {
await handleStreaming(llm); await handleStreaming(llm);
}); });
it("should handle image input", { retry: 3 }, async () => {
await handleImage(llm);
});
it("should handle multi-turn with tools", { retry: 3 }, async () => {
await multiTurn(llm);
});
}, },
); );

1
packages/ai/test/tokens.test.ts
View file

@ -55,6 +55,7 @@ async function testTokensOnAbort<TApi extends Api>(llm: Model<TApi>, options: Op
llm.provider === "google-gemini-cli" || llm.provider === "google-gemini-cli" ||
llm.provider === "zai" || llm.provider === "zai" ||
llm.provider === "amazon-bedrock" || llm.provider === "amazon-bedrock" ||
llm.provider === "vercel-ai-gateway" ||
(llm.provider === "google-antigravity" && llm.id.includes("gpt-oss")) (llm.provider === "google-antigravity" && llm.id.includes("gpt-oss"))
) { ) {
expect(msg.usage.input).toBe(0); expect(msg.usage.input).toBe(0);