From 4b761f6e23d52d72f4fa192f8f6dc7d0a0d76293 Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Sun, 12 Apr 2026 16:37:53 +0100
Subject: [PATCH] test(agents): share gemini model fixture

---
 src/agents/google-transport-stream.test.ts | 48 +++++++++-------------
 1 file changed, 20 insertions(+), 28 deletions(-)

diff --git a/src/agents/google-transport-stream.test.ts b/src/agents/google-transport-stream.test.ts
index 093d22a0380..9bd597a95b2 100644
--- a/src/agents/google-transport-stream.test.ts
+++ b/src/agents/google-transport-stream.test.ts
@@ -14,6 +14,24 @@ vi.mock("./provider-transport-fetch.js", () => ({
 let buildGoogleGenerativeAiParams: typeof import("./google-transport-stream.js").buildGoogleGenerativeAiParams;
 let createGoogleGenerativeAiTransportStreamFn: typeof import("./google-transport-stream.js").createGoogleGenerativeAiTransportStreamFn;
 
+function buildGeminiModel(
+  overrides: Partial<Model<"google-generative-ai">> = {},
+): Model<"google-generative-ai"> {
+  return {
+    id: "gemini-2.5-pro",
+    name: "Gemini 2.5 Pro",
+    api: "google-generative-ai",
+    provider: "google",
+    baseUrl: "https://generativelanguage.googleapis.com/v1beta",
+    reasoning: true,
+    input: ["text"],
+    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    contextWindow: 128000,
+    maxTokens: 8192,
+    ...overrides,
+  };
+}
+
 function buildSseResponse(events: unknown[]): Response {
   const sse = `${events.map((event) => `data: ${JSON.stringify(event)}\n\n`).join("")}data: [DONE]\n\n`;
   const encoder = new TextEncoder();
@@ -218,20 +236,7 @@ describe("google transport stream", () => {
   });
 
   it("coerces replayed malformed tool-call args to an object for Google payloads", () => {
-    const model = {
-      id: "gemini-2.5-pro",
-      name: "Gemini 2.5 Pro",
-      api: "google-generative-ai",
-      provider: "google",
-      baseUrl: "https://generativelanguage.googleapis.com/v1beta",
-      reasoning: true,
-      input: ["text"],
-      cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-      contextWindow: 128000,
-      maxTokens: 8192,
-    } satisfies Model<"google-generative-ai">;
-
-    const params = buildGoogleGenerativeAiParams(model, {
+    const params = buildGoogleGenerativeAiParams(buildGeminiModel(), {
       messages: [
         {
           role: "assistant",
@@ -291,21 +296,8 @@ describe("google transport stream", () => {
   });
 
   it("includes cachedContent in direct Gemini payloads when requested", () => {
-    const model = {
-      id: "gemini-2.5-pro",
-      name: "Gemini 2.5 Pro",
-      api: "google-generative-ai",
-      provider: "google",
-      baseUrl: "https://generativelanguage.googleapis.com/v1beta",
-      reasoning: true,
-      input: ["text"],
-      cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-      contextWindow: 128000,
-      maxTokens: 8192,
-    } satisfies Model<"google-generative-ai">;
-
     const params = buildGoogleGenerativeAiParams(
-      model,
+      buildGeminiModel(),
       {
         messages: [{ role: "user", content: "hello", timestamp: 0 }],
       } as never,