From 1d8d5d3d8854701aa90b281820e27c73ca036a66 Mon Sep 17 00:00:00 2001
From: Mario Zechner <badlogicgames@gmail.com>
Date: Tue, 13 Jan 2026 16:56:48 +0100
Subject: [PATCH] Add handleImage and multiTurn tests for vercel-ai-gateway in
 stream.test.ts

---
 packages/ai/test/stream.test.ts | 10 +++++++++-
 packages/ai/test/tokens.test.ts |  1 +
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/packages/ai/test/stream.test.ts b/packages/ai/test/stream.test.ts
index e854c130..bbbba237 100644
--- a/packages/ai/test/stream.test.ts
+++ b/packages/ai/test/stream.test.ts
@@ -599,7 +599,7 @@ describe("Generate E2E Tests", () => {
 	});
 
 	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");
 
@@ -614,6 +614,14 @@ describe("Generate E2E Tests", () => {
 			it("should handle streaming", { retry: 3 }, async () => {
 				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);
+			});
 		},
 	);
 
diff --git a/packages/ai/test/tokens.test.ts b/packages/ai/test/tokens.test.ts
index 6d10abcf..10586d12 100644
--- a/packages/ai/test/tokens.test.ts
+++ b/packages/ai/test/tokens.test.ts
@@ -55,6 +55,7 @@ async function testTokensOnAbort<TApi extends Api>(llm: Model<TApi>, options: Op
 		llm.provider === "google-gemini-cli" ||
 		llm.provider === "zai" ||
 		llm.provider === "amazon-bedrock" ||
+		llm.provider === "vercel-ai-gateway" ||
 		(llm.provider === "google-antigravity" && llm.id.includes("gpt-oss"))
 	) {
 		expect(msg.usage.input).toBe(0);