@copilotkit/aimock 1.8.0 → 1.9.0

package/dist/embeddings.js

@@ -1,4 +1,4 @@
-import { buildEmbeddingResponse, flattenHeaders, generateDeterministicEmbedding, isEmbeddingResponse, isErrorResponse } from "./helpers.js";
+import { buildEmbeddingResponse, flattenHeaders, generateDeterministicEmbedding, getTestId, isEmbeddingResponse, isErrorResponse } from "./helpers.js";
 import { matchFixture } from "./router.js";
 import { writeErrorResponse } from "./sse-writer.js";
 import { applyChaos } from "./chaos.js";
@@ -36,8 +36,9 @@ async function handleEmbeddings(req, res, raw, fixtures, journal, defaults, setC
 		messages: [],
 		embeddingInput: combinedInput
 	};
-	const fixture = matchFixture(fixtures, syntheticReq, journal.fixtureMatchCounts, defaults.requestTransform);
-	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures);
+	const testId = getTestId(req);
+	const fixture = matchFixture(fixtures, syntheticReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
+	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures, testId);
 	if (applyChaos(res, fixture, defaults.chaos, req.headers, journal, {
 		method: req.method ?? "POST",
 		path: req.url ?? "/v1/embeddings",

package/dist/embeddings.js.map

@@ -1 +1 @@
-{"version":3,"file":"embeddings.js","names":[],"sources":["../src/embeddings.ts"],"sourcesContent":["/**\n * OpenAI Embeddings API support for LLMock.\n *\n * Handles POST /v1/embeddings requests. Matches fixtures using the `inputText`\n * field, and falls back to generating a deterministic embedding from the input\n * text hash when no fixture matches.\n */\n\nimport type * as http from \"node:http\";\nimport type { ChatCompletionRequest, Fixture, HandlerDefaults } from \"./types.js\";\nimport {\n  isEmbeddingResponse,\n  isErrorResponse,\n  generateDeterministicEmbedding,\n  buildEmbeddingResponse,\n  flattenHeaders,\n} from \"./helpers.js\";\nimport { matchFixture } from \"./router.js\";\nimport { writeErrorResponse } from \"./sse-writer.js\";\nimport type { Journal } from \"./journal.js\";\nimport { applyChaos } from \"./chaos.js\";\nimport { proxyAndRecord } from \"./recorder.js\";\n\n// ─── Embeddings API request types ──────────────────────────────────────────\n\ninterface EmbeddingRequest {\n  input: string | string[];\n  model: string;\n  encoding_format?: \"float\" | \"base64\";\n  dimensions?: number;\n  [key: string]: unknown;\n}\n\n// ─── Request handler ───────────────────────────────────────────────────────\n\nexport async function handleEmbeddings(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  raw: string,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: HandlerDefaults,\n  setCorsHeaders: (res: http.ServerResponse) => void,\n): Promise<void> {\n  const { logger } = defaults;\n  setCorsHeaders(res);\n\n  let embeddingReq: EmbeddingRequest;\n  try {\n    embeddingReq = JSON.parse(raw) as EmbeddingRequest;\n  } catch {\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v1/embeddings\",\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 400, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      400,\n      JSON.stringify({\n        error: {\n          message: \"Malformed JSON\",\n          type: \"invalid_request_error\",\n          code: \"invalid_json\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Normalize input to array of strings\n  const inputs: string[] = Array.isArray(embeddingReq.input)\n    ? embeddingReq.input\n    : [embeddingReq.input];\n\n  // Concatenate all inputs for matching purposes\n  const combinedInput = inputs.join(\" \");\n\n  // Build a synthetic ChatCompletionRequest for the fixture router.\n  // We attach `embeddingInput` so the router's inputText matching can use it.\n  const syntheticReq: ChatCompletionRequest = {\n    model: embeddingReq.model,\n    messages: [],\n    embeddingInput: combinedInput,\n  };\n\n  const fixture = matchFixture(\n    fixtures,\n    syntheticReq,\n    journal.fixtureMatchCounts,\n    defaults.requestTransform,\n  );\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures);\n  }\n\n  if (\n    applyChaos(\n      res,\n      fixture,\n      defaults.chaos,\n      req.headers,\n      journal,\n      {\n        method: req.method ?? \"POST\",\n        path: req.url ?? \"/v1/embeddings\",\n        headers: flattenHeaders(req.headers),\n        body: syntheticReq,\n      },\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\n\n  if (fixture) {\n    const response = fixture.response;\n\n    // Error response\n    if (isErrorResponse(response)) {\n      const status = response.status ?? 500;\n      journal.add({\n        method: req.method ?? \"POST\",\n        path: req.url ?? \"/v1/embeddings\",\n        headers: flattenHeaders(req.headers),\n        body: syntheticReq,\n        response: { status, fixture },\n      });\n      writeErrorResponse(res, status, JSON.stringify(response));\n      return;\n    }\n\n    // Embedding response — use the fixture's embedding for each input\n    if (isEmbeddingResponse(response)) {\n      journal.add({\n        method: req.method ?? \"POST\",\n        path: req.url ?? \"/v1/embeddings\",\n        headers: flattenHeaders(req.headers),\n        body: syntheticReq,\n        response: { status: 200, fixture },\n      });\n      const embeddings = inputs.map(() => [...response.embedding]);\n      const body = buildEmbeddingResponse(embeddings, embeddingReq.model);\n      res.writeHead(200, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify(body));\n      return;\n    }\n\n    // Fixture matched but response type is not compatible with embeddings\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v1/embeddings\",\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: { status: 500, fixture },\n    });\n    writeErrorResponse(\n      res,\n      500,\n      JSON.stringify({\n        error: {\n          message:\n            \"Fixture response did not match any known embedding type (must have embedding or error)\",\n          type: \"server_error\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // No fixture match — try record-and-replay proxy if configured\n  if (defaults.record) {\n    const proxied = await proxyAndRecord(\n      req,\n      res,\n      syntheticReq,\n      \"openai\",\n      req.url ?? \"/v1/embeddings\",\n      fixtures,\n      defaults,\n      raw,\n    );\n    if (proxied) {\n      journal.add({\n        method: req.method ?? \"POST\",\n        path: req.url ?? \"/v1/embeddings\",\n        headers: flattenHeaders(req.headers),\n        body: syntheticReq,\n        response: { status: res.statusCode ?? 200, fixture: null },\n      });\n      return;\n    }\n  }\n\n  if (defaults.strict) {\n    logger.error(\n      `STRICT: No fixture matched for ${req.method ?? \"POST\"} ${req.url ?? \"/v1/embeddings\"}`,\n    );\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v1/embeddings\",\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: { status: 503, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      503,\n      JSON.stringify({\n        error: {\n          message: \"Strict mode: no fixture matched\",\n          type: \"invalid_request_error\",\n          code: \"no_fixture_match\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // No fixture match — generate deterministic embeddings from input text\n  logger.warn(\n    `No embedding fixture matched for \"${combinedInput.slice(0, 80)}\" — returning deterministic fallback`,\n  );\n  const dimensions = embeddingReq.dimensions ?? 1536;\n  const embeddings = inputs.map((input) => generateDeterministicEmbedding(input, dimensions));\n\n  journal.add({\n    method: req.method ?? \"POST\",\n    path: req.url ?? \"/v1/embeddings\",\n    headers: flattenHeaders(req.headers),\n    body: syntheticReq,\n    response: { status: 200, fixture: null },\n  });\n\n  const body = buildEmbeddingResponse(embeddings, embeddingReq.model);\n  res.writeHead(200, { \"Content-Type\": \"application/json\" });\n  res.end(JSON.stringify(body));\n}\n"],"mappings":";;;;;;;AAmCA,eAAsB,iBACpB,KACA,KACA,KACA,UACA,SACA,UACA,gBACe;CACf,MAAM,EAAE,WAAW;AACnB,gBAAe,IAAI;CAEnB,IAAI;AACJ,KAAI;AACF,iBAAe,KAAK,MAAM,IAAI;SACxB;AACN,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,MAAM;GACP,EACF,CAAC,CACH;AACD;;CAIF,MAAM,SAAmB,MAAM,QAAQ,aAAa,MAAM,GACtD,aAAa,QACb,CAAC,aAAa,MAAM;CAGxB,MAAM,gBAAgB,OAAO,KAAK,IAAI;CAItC,MAAM,eAAsC;EAC1C,OAAO,aAAa;EACpB,UAAU,EAAE;EACZ,gBAAgB;EACjB;CAED,MAAM,UAAU,aACd,UACA,cACA,QAAQ,oBACR,SAAS,iBACV;AAED,KAAI,QACF,SAAQ,2BAA2B,SAAS,SAAS;AAGvD,KACE,WACE,KACA,SACA,SAAS,OACT,IAAI,SACJ,SACA;EACE,QAAQ,IAAI,UAAU;EACtB,MAAM,IAAI,OAAO;EACjB,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACP,EACD,SAAS,UACT,SAAS,OACV,CAED;AAEF,KAAI,SAAS;EACX,MAAM,WAAW,QAAQ;AAGzB,MAAI,gBAAgB,SAAS,EAAE;GAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,WAAQ,IAAI;IACV,QAAQ,IAAI,UAAU;IACtB,MAAM,IAAI,OAAO;IACjB,SAAS,eAAe,IAAI,QAAQ;IACpC,MAAM;IACN,UAAU;KAAE;KAAQ;KAAS;IAC9B,CAAC;AACF,sBAAmB,KAAK,QAAQ,KAAK,UAAU,SAAS,CAAC;AACzD;;AAIF,MAAI,oBAAoB,SAAS,EAAE;AACjC,WAAQ,IAAI;IACV,QAAQ,IAAI,UAAU;IACtB,MAAM,IAAI,OAAO;IACjB,SAAS,eAAe,IAAI,QAAQ;IACpC,MAAM;IACN,UAAU;KAAE,QAAQ;KAAK;KAAS;IACnC,CAAC;GAEF,MAAM,OAAO,uBADM,OAAO,UAAU,CAAC,GAAG,SAAS,UAAU,CAAC,EACZ,aAAa,MAAM;AACnE,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,KAAK,CAAC;AAC7B;;AAIF,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SACE;GACF,MAAM;GACP,EACF,CAAC,CACH;AACD;;AAIF,KAAI,SAAS,QAWX;MAVgB,MAAM,eACpB,KACA,KACA,cACA,UACA,IAAI,OAAO,kBACX,UACA,UACA,IACD,EACY;AACX,WAAQ,IAAI;IACV,QAAQ,IAAI,UAAU;IACtB,MAAM,IAAI,OAAO;IACjB,SAAS,eAAe,IAAI,QAAQ;IACpC,MAAM;IACN,UAAU;KAAE,QAAQ,IAAI,cAAc;KAAK,SAAS;KAAM;IAC3D,CAAC;AACF;;;AAIJ,KAAI,SAAS,QAAQ;AACnB,SAAO,MACL,kCAAkC,IAAI,UAAU,OAAO,GAAG,IAAI,OAAO,mBACtE;AACD,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,MAAM;GACP,EACF,CAAC,CACH;AACD;;AAIF,QAAO,KACL,qCAAqC,cAAc,MAAM,GAAG,GAAG,CAAC,sCACjE;CACD,MAAM,aAAa,aAAa,cAAc;CAC9C,MAAM,aAAa,OAAO,KAAK,UAAU,+BAA+B,OAAO,WAAW,CAAC;AAE3F,SAAQ,IAAI;EACV,QAAQ,IAAI,UAAU;EACtB,MAAM,IAAI,OAAO;EACjB,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK,SAAS;GAAM;EACzC,CAAC;CAEF,MAAM,OAAO,uBAAuB,YAAY,aAAa,MAAM;AACnE,KAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,KAAI,IAAI,KAAK,UAAU,KAAK,CAAC"}
+{"version":3,"file":"embeddings.js","names":[],"sources":["../src/embeddings.ts"],"sourcesContent":["/**\n * OpenAI Embeddings API support for LLMock.\n *\n * Handles POST /v1/embeddings requests. Matches fixtures using the `inputText`\n * field, and falls back to generating a deterministic embedding from the input\n * text hash when no fixture matches.\n */\n\nimport type * as http from \"node:http\";\nimport type { ChatCompletionRequest, Fixture, HandlerDefaults } from \"./types.js\";\nimport {\n  isEmbeddingResponse,\n  isErrorResponse,\n  generateDeterministicEmbedding,\n  buildEmbeddingResponse,\n  flattenHeaders,\n  getTestId,\n} from \"./helpers.js\";\nimport { matchFixture } from \"./router.js\";\nimport { writeErrorResponse } from \"./sse-writer.js\";\nimport type { Journal } from \"./journal.js\";\nimport { applyChaos } from \"./chaos.js\";\nimport { proxyAndRecord } from \"./recorder.js\";\n\n// ─── Embeddings API request types ──────────────────────────────────────────\n\ninterface EmbeddingRequest {\n  input: string | string[];\n  model: string;\n  encoding_format?: \"float\" | \"base64\";\n  dimensions?: number;\n  [key: string]: unknown;\n}\n\n// ─── Request handler ───────────────────────────────────────────────────────\n\nexport async function handleEmbeddings(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  raw: string,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: HandlerDefaults,\n  setCorsHeaders: (res: http.ServerResponse) => void,\n): Promise<void> {\n  const { logger } = defaults;\n  setCorsHeaders(res);\n\n  let embeddingReq: EmbeddingRequest;\n  try {\n    embeddingReq = JSON.parse(raw) as EmbeddingRequest;\n  } catch {\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v1/embeddings\",\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 400, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      400,\n      JSON.stringify({\n        error: {\n          message: \"Malformed JSON\",\n          type: \"invalid_request_error\",\n          code: \"invalid_json\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Normalize input to array of strings\n  const inputs: string[] = Array.isArray(embeddingReq.input)\n    ? embeddingReq.input\n    : [embeddingReq.input];\n\n  // Concatenate all inputs for matching purposes\n  const combinedInput = inputs.join(\" \");\n\n  // Build a synthetic ChatCompletionRequest for the fixture router.\n  // We attach `embeddingInput` so the router's inputText matching can use it.\n  const syntheticReq: ChatCompletionRequest = {\n    model: embeddingReq.model,\n    messages: [],\n    embeddingInput: combinedInput,\n  };\n\n  const testId = getTestId(req);\n  const fixture = matchFixture(\n    fixtures,\n    syntheticReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (\n    applyChaos(\n      res,\n      fixture,\n      defaults.chaos,\n      req.headers,\n      journal,\n      {\n        method: req.method ?? \"POST\",\n        path: req.url ?? \"/v1/embeddings\",\n        headers: flattenHeaders(req.headers),\n        body: syntheticReq,\n      },\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\n\n  if (fixture) {\n    const response = fixture.response;\n\n    // Error response\n    if (isErrorResponse(response)) {\n      const status = response.status ?? 500;\n      journal.add({\n        method: req.method ?? \"POST\",\n        path: req.url ?? \"/v1/embeddings\",\n        headers: flattenHeaders(req.headers),\n        body: syntheticReq,\n        response: { status, fixture },\n      });\n      writeErrorResponse(res, status, JSON.stringify(response));\n      return;\n    }\n\n    // Embedding response — use the fixture's embedding for each input\n    if (isEmbeddingResponse(response)) {\n      journal.add({\n        method: req.method ?? \"POST\",\n        path: req.url ?? \"/v1/embeddings\",\n        headers: flattenHeaders(req.headers),\n        body: syntheticReq,\n        response: { status: 200, fixture },\n      });\n      const embeddings = inputs.map(() => [...response.embedding]);\n      const body = buildEmbeddingResponse(embeddings, embeddingReq.model);\n      res.writeHead(200, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify(body));\n      return;\n    }\n\n    // Fixture matched but response type is not compatible with embeddings\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v1/embeddings\",\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: { status: 500, fixture },\n    });\n    writeErrorResponse(\n      res,\n      500,\n      JSON.stringify({\n        error: {\n          message:\n            \"Fixture response did not match any known embedding type (must have embedding or error)\",\n          type: \"server_error\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // No fixture match — try record-and-replay proxy if configured\n  if (defaults.record) {\n    const proxied = await proxyAndRecord(\n      req,\n      res,\n      syntheticReq,\n      \"openai\",\n      req.url ?? \"/v1/embeddings\",\n      fixtures,\n      defaults,\n      raw,\n    );\n    if (proxied) {\n      journal.add({\n        method: req.method ?? \"POST\",\n        path: req.url ?? \"/v1/embeddings\",\n        headers: flattenHeaders(req.headers),\n        body: syntheticReq,\n        response: { status: res.statusCode ?? 200, fixture: null },\n      });\n      return;\n    }\n  }\n\n  if (defaults.strict) {\n    logger.error(\n      `STRICT: No fixture matched for ${req.method ?? \"POST\"} ${req.url ?? \"/v1/embeddings\"}`,\n    );\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v1/embeddings\",\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: { status: 503, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      503,\n      JSON.stringify({\n        error: {\n          message: \"Strict mode: no fixture matched\",\n          type: \"invalid_request_error\",\n          code: \"no_fixture_match\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // No fixture match — generate deterministic embeddings from input text\n  logger.warn(\n    `No embedding fixture matched for \"${combinedInput.slice(0, 80)}\" — returning deterministic fallback`,\n  );\n  const dimensions = embeddingReq.dimensions ?? 1536;\n  const embeddings = inputs.map((input) => generateDeterministicEmbedding(input, dimensions));\n\n  journal.add({\n    method: req.method ?? \"POST\",\n    path: req.url ?? \"/v1/embeddings\",\n    headers: flattenHeaders(req.headers),\n    body: syntheticReq,\n    response: { status: 200, fixture: null },\n  });\n\n  const body = buildEmbeddingResponse(embeddings, embeddingReq.model);\n  res.writeHead(200, { \"Content-Type\": \"application/json\" });\n  res.end(JSON.stringify(body));\n}\n"],"mappings":";;;;;;;AAoCA,eAAsB,iBACpB,KACA,KACA,KACA,UACA,SACA,UACA,gBACe;CACf,MAAM,EAAE,WAAW;AACnB,gBAAe,IAAI;CAEnB,IAAI;AACJ,KAAI;AACF,iBAAe,KAAK,MAAM,IAAI;SACxB;AACN,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,MAAM;GACP,EACF,CAAC,CACH;AACD;;CAIF,MAAM,SAAmB,MAAM,QAAQ,aAAa,MAAM,GACtD,aAAa,QACb,CAAC,aAAa,MAAM;CAGxB,MAAM,gBAAgB,OAAO,KAAK,IAAI;CAItC,MAAM,eAAsC;EAC1C,OAAO,aAAa;EACpB,UAAU,EAAE;EACZ,gBAAgB;EACjB;CAED,MAAM,SAAS,UAAU,IAAI;CAC7B,MAAM,UAAU,aACd,UACA,cACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;AAED,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KACE,WACE,KACA,SACA,SAAS,OACT,IAAI,SACJ,SACA;EACE,QAAQ,IAAI,UAAU;EACtB,MAAM,IAAI,OAAO;EACjB,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACP,EACD,SAAS,UACT,SAAS,OACV,CAED;AAEF,KAAI,SAAS;EACX,MAAM,WAAW,QAAQ;AAGzB,MAAI,gBAAgB,SAAS,EAAE;GAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,WAAQ,IAAI;IACV,QAAQ,IAAI,UAAU;IACtB,MAAM,IAAI,OAAO;IACjB,SAAS,eAAe,IAAI,QAAQ;IACpC,MAAM;IACN,UAAU;KAAE;KAAQ;KAAS;IAC9B,CAAC;AACF,sBAAmB,KAAK,QAAQ,KAAK,UAAU,SAAS,CAAC;AACzD;;AAIF,MAAI,oBAAoB,SAAS,EAAE;AACjC,WAAQ,IAAI;IACV,QAAQ,IAAI,UAAU;IACtB,MAAM,IAAI,OAAO;IACjB,SAAS,eAAe,IAAI,QAAQ;IACpC,MAAM;IACN,UAAU;KAAE,QAAQ;KAAK;KAAS;IACnC,CAAC;GAEF,MAAM,OAAO,uBADM,OAAO,UAAU,CAAC,GAAG,SAAS,UAAU,CAAC,EACZ,aAAa,MAAM;AACnE,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,KAAK,CAAC;AAC7B;;AAIF,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SACE;GACF,MAAM;GACP,EACF,CAAC,CACH;AACD;;AAIF,KAAI,SAAS,QAWX;MAVgB,MAAM,eACpB,KACA,KACA,cACA,UACA,IAAI,OAAO,kBACX,UACA,UACA,IACD,EACY;AACX,WAAQ,IAAI;IACV,QAAQ,IAAI,UAAU;IACtB,MAAM,IAAI,OAAO;IACjB,SAAS,eAAe,IAAI,QAAQ;IACpC,MAAM;IACN,UAAU;KAAE,QAAQ,IAAI,cAAc;KAAK,SAAS;KAAM;IAC3D,CAAC;AACF;;;AAIJ,KAAI,SAAS,QAAQ;AACnB,SAAO,MACL,kCAAkC,IAAI,UAAU,OAAO,GAAG,IAAI,OAAO,mBACtE;AACD,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,MAAM;GACP,EACF,CAAC,CACH;AACD;;AAIF,QAAO,KACL,qCAAqC,cAAc,MAAM,GAAG,GAAG,CAAC,sCACjE;CACD,MAAM,aAAa,aAAa,cAAc;CAC9C,MAAM,aAAa,OAAO,KAAK,UAAU,+BAA+B,OAAO,WAAW,CAAC;AAE3F,SAAQ,IAAI;EACV,QAAQ,IAAI,UAAU;EACtB,MAAM,IAAI,OAAO;EACjB,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK,SAAS;GAAM;EACzC,CAAC;CAEF,MAAM,OAAO,uBAAuB,YAAY,aAAa,MAAM;AACnE,KAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,KAAI,IAAI,KAAK,UAAU,KAAK,CAAC"}

package/dist/gemini.cjs

@@ -136,25 +136,26 @@ function buildGeminiTextStreamChunks(content, chunkSize, reasoning) {
 	});
 	return chunks;
 }
+function parseToolCallPart(tc, logger) {
+	let argsObj;
+	try {
+		argsObj = JSON.parse(tc.arguments || "{}");
+	} catch {
+		logger.warn(`Malformed JSON in fixture tool call arguments for "${tc.name}": ${tc.arguments}`);
+		argsObj = {};
+	}
+	return { functionCall: {
+		name: tc.name,
+		args: argsObj,
+		id: tc.id || require_helpers.generateToolCallId()
+	} };
+}
 function buildGeminiToolCallStreamChunks(toolCalls, logger) {
 	return [{
 		candidates: [{
 			content: {
 				role: "model",
-				parts: toolCalls.map((tc) => {
-					let argsObj;
-					try {
-						argsObj = JSON.parse(tc.arguments || "{}");
-					} catch {
-						logger.warn(`Malformed JSON in fixture tool call arguments for "${tc.name}": ${tc.arguments}`);
-						argsObj = {};
-					}
-					return { functionCall: {
-						name: tc.name,
-						args: argsObj,
-						id: tc.id || require_helpers.generateToolCallId()
-					} };
-				})
+				parts: toolCalls.map((tc) => parseToolCallPart(tc, logger))
 			},
 			finishReason: "FUNCTION_CALL",
 			index: 0
@@ -194,20 +195,61 @@ function buildGeminiToolCallResponse(toolCalls, logger) {
 		candidates: [{
 			content: {
 				role: "model",
-				parts: toolCalls.map((tc) => {
-					let argsObj;
-					try {
-						argsObj = JSON.parse(tc.arguments || "{}");
-					} catch {
-						logger.warn(`Malformed JSON in fixture tool call arguments for "${tc.name}": ${tc.arguments}`);
-						argsObj = {};
-					}
-					return { functionCall: {
-						name: tc.name,
-						args: argsObj,
-						id: tc.id || require_helpers.generateToolCallId()
-					} };
-				})
+				parts: toolCalls.map((tc) => parseToolCallPart(tc, logger))
+			},
+			finishReason: "FUNCTION_CALL",
+			index: 0
+		}],
+		usageMetadata: {
+			promptTokenCount: 0,
+			candidatesTokenCount: 0,
+			totalTokenCount: 0
+		}
+	};
+}
+function buildGeminiContentWithToolCallsStreamChunks(content, toolCalls, chunkSize, logger) {
+	const chunks = [];
+	if (content.length === 0) chunks.push({ candidates: [{
+		content: {
+			role: "model",
+			parts: [{ text: "" }]
+		},
+		index: 0
+	}] });
+	else for (let i = 0; i < content.length; i += chunkSize) {
+		const slice = content.slice(i, i + chunkSize);
+		chunks.push({ candidates: [{
+			content: {
+				role: "model",
+				parts: [{ text: slice }]
+			},
+			index: 0
+		}] });
+	}
+	const parts = toolCalls.map((tc) => parseToolCallPart(tc, logger));
+	chunks.push({
+		candidates: [{
+			content: {
+				role: "model",
+				parts
+			},
+			finishReason: "FUNCTION_CALL",
+			index: 0
+		}],
+		usageMetadata: {
+			promptTokenCount: 0,
+			candidatesTokenCount: 0,
+			totalTokenCount: 0
+		}
+	});
+	return chunks;
+}
+function buildGeminiContentWithToolCallsResponse(content, toolCalls, logger) {
+	return {
+		candidates: [{
+			content: {
+				role: "model",
+				parts: [{ text: content }, ...toolCalls.map((tc) => parseToolCallPart(tc, logger))]
 			},
 			finishReason: "FUNCTION_CALL",
 			index: 0
@@ -268,9 +310,10 @@ async function handleGemini(req, res, raw, model, streaming, fixtures, journal,
 		return;
 	}
 	const completionReq = geminiToCompletionRequest(geminiReq, model, streaming);
-	const fixture = require_router.matchFixture(fixtures, completionReq, journal.fixtureMatchCounts, defaults.requestTransform);
+	const testId = require_helpers.getTestId(req);
+	const fixture = require_router.matchFixture(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
 	const path = req.url ?? `/v1beta/models/${model}:generateContent`;
-	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures);
+	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures, testId);
 	if (require_chaos.applyChaos(res, fixture, defaults.chaos, req.headers, journal, {
 		method: req.method ?? "POST",
 		path,
@@ -336,6 +379,38 @@ async function handleGemini(req, res, raw, model, streaming, fixtures, journal,
 		require_sse_writer.writeErrorResponse(res, status, JSON.stringify(geminiError));
 		return;
 	}
+	if (require_helpers.isContentWithToolCallsResponse(response)) {
+		const journalEntry = journal.add({
+			method: req.method ?? "POST",
+			path,
+			headers: require_helpers.flattenHeaders(req.headers),
+			body: completionReq,
+			response: {
+				status: 200,
+				fixture
+			}
+		});
+		if (!streaming) {
+			const body = buildGeminiContentWithToolCallsResponse(response.content, response.toolCalls, logger);
+			res.writeHead(200, { "Content-Type": "application/json" });
+			res.end(JSON.stringify(body));
+		} else {
+			const chunks = buildGeminiContentWithToolCallsStreamChunks(response.content, response.toolCalls, chunkSize, logger);
+			const interruption = require_interruption.createInterruptionSignal(fixture);
+			if (!await writeGeminiSSEStream(res, chunks, {
+				latency,
+				streamingProfile: fixture.streamingProfile,
+				signal: interruption?.signal,
+				onChunkSent: interruption?.tick
+			})) {
+				if (!res.writableEnded) res.destroy();
+				journalEntry.response.interrupted = true;
+				journalEntry.response.interruptReason = interruption?.reason();
+			}
+			interruption?.cleanup();
+		}
+		return;
+	}
 	if (require_helpers.isTextResponse(response)) {
 		const journalEntry = journal.add({
 			method: req.method ?? "POST",

package/dist/gemini.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"gemini.cjs","names":["generateToolCallId","calculateDelay","delay","flattenHeaders","matchFixture","applyChaos","proxyAndRecord","isErrorResponse","isTextResponse","createInterruptionSignal","isToolCallResponse"],"sources":["../src/gemini.ts"],"sourcesContent":["/**\n * Google Gemini GenerateContent API support.\n *\n * Translates incoming Gemini requests into the ChatCompletionRequest format\n * used by the fixture router, and converts fixture responses back into the\n * Gemini GenerateContent streaming (or non-streaming) format.\n */\n\nimport type * as http from \"node:http\";\nimport type {\n  ChatCompletionRequest,\n  ChatMessage,\n  Fixture,\n  HandlerDefaults,\n  RecordProviderKey,\n  StreamingProfile,\n  ToolCall,\n  ToolDefinition,\n} from \"./types.js\";\nimport {\n  isTextResponse,\n  isToolCallResponse,\n  isErrorResponse,\n  generateToolCallId,\n  flattenHeaders,\n} from \"./helpers.js\";\nimport { matchFixture } from \"./router.js\";\nimport { writeErrorResponse, delay, calculateDelay } from \"./sse-writer.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport type { Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport { applyChaos } from \"./chaos.js\";\nimport { proxyAndRecord } from \"./recorder.js\";\n\n// ─── Gemini request types ───────────────────────────────────────────────────\n\ninterface GeminiPart {\n  text?: string;\n  thought?: boolean;\n  functionCall?: { name: string; args: Record<string, unknown>; id?: string };\n  functionResponse?: { name: string; response: unknown };\n}\n\ninterface GeminiContent {\n  role?: string;\n  parts: GeminiPart[];\n}\n\ninterface GeminiFunctionDeclaration {\n  name: string;\n  description?: string;\n  parameters?: object;\n}\n\ninterface GeminiToolDef {\n  functionDeclarations?: GeminiFunctionDeclaration[];\n}\n\ninterface GeminiRequest {\n  contents?: GeminiContent[];\n  systemInstruction?: GeminiContent;\n  tools?: GeminiToolDef[];\n  generationConfig?: {\n    temperature?: number;\n    maxOutputTokens?: number;\n    [key: string]: unknown;\n  };\n  [key: string]: unknown;\n}\n\n// ─── Input conversion: Gemini → ChatCompletions messages ────────────────────\n\nexport function geminiToCompletionRequest(\n  req: GeminiRequest,\n  model: string,\n  stream: boolean,\n): ChatCompletionRequest {\n  const messages: ChatMessage[] = [];\n\n  // systemInstruction → system message\n  if (req.systemInstruction) {\n    const text = req.systemInstruction.parts\n      .filter((p) => p.text !== undefined)\n      .map((p) => p.text!)\n      .join(\"\");\n    if (text) {\n      messages.push({ role: \"system\", content: text });\n    }\n  }\n\n  if (req.contents) {\n    for (const content of req.contents) {\n      const role = content.role ?? \"user\";\n\n      if (role === \"user\") {\n        // Check for functionResponse parts\n        const funcResponses = content.parts.filter((p) => p.functionResponse);\n        const textParts = content.parts.filter((p) => p.text !== undefined);\n\n        if (funcResponses.length > 0) {\n          // functionResponse → tool message\n          for (let i = 0; i < funcResponses.length; i++) {\n            const part = funcResponses[i];\n            messages.push({\n              role: \"tool\",\n              content:\n                typeof part.functionResponse!.response === \"string\"\n                  ? part.functionResponse!.response\n                  : JSON.stringify(part.functionResponse!.response),\n              tool_call_id: `call_gemini_${part.functionResponse!.name}_${i}`,\n            });\n          }\n          // Any text parts alongside → user message\n          if (textParts.length > 0) {\n            messages.push({\n              role: \"user\",\n              content: textParts.map((p) => p.text!).join(\"\"),\n            });\n          }\n        } else {\n          // Regular user text\n          const text = textParts.map((p) => p.text!).join(\"\");\n          messages.push({ role: \"user\", content: text });\n        }\n      } else if (role === \"model\") {\n        // Check for functionCall parts\n        const funcCalls = content.parts.filter((p) => p.functionCall);\n        const textParts = content.parts.filter((p) => p.text !== undefined);\n\n        if (funcCalls.length > 0) {\n          messages.push({\n            role: \"assistant\",\n            content: null,\n            tool_calls: funcCalls.map((p, i) => ({\n              id: `call_gemini_${p.functionCall!.name}_${i}`,\n              type: \"function\" as const,\n              function: {\n                name: p.functionCall!.name,\n                arguments: JSON.stringify(p.functionCall!.args),\n              },\n            })),\n          });\n        } else {\n          const text = textParts.map((p) => p.text!).join(\"\");\n          messages.push({ role: \"assistant\", content: text });\n        }\n      }\n    }\n  }\n\n  // Convert tools\n  let tools: ToolDefinition[] | undefined;\n  if (req.tools && req.tools.length > 0) {\n    const decls = req.tools.flatMap((t) => t.functionDeclarations ?? []);\n    if (decls.length > 0) {\n      tools = decls.map((d) => ({\n        type: \"function\" as const,\n        function: {\n          name: d.name,\n          description: d.description,\n          parameters: d.parameters,\n        },\n      }));\n    }\n  }\n\n  return {\n    model,\n    messages,\n    stream,\n    temperature: req.generationConfig?.temperature,\n    tools,\n  };\n}\n\n// ─── Response building: fixture → Gemini format ─────────────────────────────\n\ninterface GeminiResponseChunk {\n  candidates: {\n    content: { role: string; parts: GeminiPart[] };\n    finishReason?: string;\n    index: number;\n  }[];\n  usageMetadata?: {\n    promptTokenCount: number;\n    candidatesTokenCount: number;\n    totalTokenCount: number;\n  };\n}\n\nfunction buildGeminiTextStreamChunks(\n  content: string,\n  chunkSize: number,\n  reasoning?: string,\n): GeminiResponseChunk[] {\n  const chunks: GeminiResponseChunk[] = [];\n\n  // Reasoning chunks (thought: true)\n  if (reasoning) {\n    for (let i = 0; i < reasoning.length; i += chunkSize) {\n      const slice = reasoning.slice(i, i + chunkSize);\n      chunks.push({\n        candidates: [\n          {\n            content: { role: \"model\", parts: [{ text: slice, thought: true }] },\n            index: 0,\n          },\n        ],\n      });\n    }\n  }\n\n  // Content chunks (original logic unchanged)\n  for (let i = 0; i < content.length; i += chunkSize) {\n    const slice = content.slice(i, i + chunkSize);\n    const isLast = i + chunkSize >= content.length;\n    const chunk: GeminiResponseChunk = {\n      candidates: [\n        {\n          content: { role: \"model\", parts: [{ text: slice }] },\n          index: 0,\n          ...(isLast ? { finishReason: \"STOP\" } : {}),\n        },\n      ],\n      ...(isLast\n        ? {\n            usageMetadata: {\n              promptTokenCount: 0,\n              candidatesTokenCount: 0,\n              totalTokenCount: 0,\n            },\n          }\n        : {}),\n    };\n    chunks.push(chunk);\n  }\n\n  // Handle empty content (original logic unchanged)\n  if (content.length === 0) {\n    chunks.push({\n      candidates: [\n        {\n          content: { role: \"model\", parts: [{ text: \"\" }] },\n          finishReason: \"STOP\",\n          index: 0,\n        },\n      ],\n      usageMetadata: {\n        promptTokenCount: 0,\n        candidatesTokenCount: 0,\n        totalTokenCount: 0,\n      },\n    });\n  }\n\n  return chunks;\n}\n\nfunction buildGeminiToolCallStreamChunks(\n  toolCalls: ToolCall[],\n  logger: Logger,\n): GeminiResponseChunk[] {\n  const parts: GeminiPart[] = toolCalls.map((tc) => {\n    let argsObj: Record<string, unknown>;\n    try {\n      argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n    } catch {\n      logger.warn(\n        `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n      );\n      argsObj = {};\n    }\n    return {\n      functionCall: { name: tc.name, args: argsObj, id: tc.id || generateToolCallId() },\n    };\n  });\n\n  // Gemini sends all tool calls in a single response chunk\n  return [\n    {\n      candidates: [\n        {\n          content: { role: \"model\", parts },\n          finishReason: \"FUNCTION_CALL\",\n          index: 0,\n        },\n      ],\n      usageMetadata: {\n        promptTokenCount: 0,\n        candidatesTokenCount: 0,\n        totalTokenCount: 0,\n      },\n    },\n  ];\n}\n\n// Non-streaming response builders\n\nfunction buildGeminiTextResponse(content: string, reasoning?: string): GeminiResponseChunk {\n  const parts: GeminiPart[] = [];\n  if (reasoning) {\n    parts.push({ text: reasoning, thought: true });\n  }\n  parts.push({ text: content });\n\n  return {\n    candidates: [\n      {\n        content: { role: \"model\", parts },\n        finishReason: \"STOP\",\n        index: 0,\n      },\n    ],\n    usageMetadata: {\n      promptTokenCount: 0,\n      candidatesTokenCount: 0,\n      totalTokenCount: 0,\n    },\n  };\n}\n\nfunction buildGeminiToolCallResponse(toolCalls: ToolCall[], logger: Logger): GeminiResponseChunk {\n  const parts: GeminiPart[] = toolCalls.map((tc) => {\n    let argsObj: Record<string, unknown>;\n    try {\n      argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n    } catch {\n      logger.warn(\n        `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n      );\n      argsObj = {};\n    }\n    return {\n      functionCall: { name: tc.name, args: argsObj, id: tc.id || generateToolCallId() },\n    };\n  });\n\n  return {\n    candidates: [\n      {\n        content: { role: \"model\", parts },\n        finishReason: \"FUNCTION_CALL\",\n        index: 0,\n      },\n    ],\n    usageMetadata: {\n      promptTokenCount: 0,\n      candidatesTokenCount: 0,\n      totalTokenCount: 0,\n    },\n  };\n}\n\n// ─── SSE writer for Gemini streaming ────────────────────────────────────────\n\ninterface GeminiStreamOptions {\n  latency?: number;\n  streamingProfile?: StreamingProfile;\n  signal?: AbortSignal;\n  onChunkSent?: () => void;\n}\n\nasync function writeGeminiSSEStream(\n  res: http.ServerResponse,\n  chunks: GeminiResponseChunk[],\n  optionsOrLatency?: number | GeminiStreamOptions,\n): Promise<boolean> {\n  const opts: GeminiStreamOptions =\n    typeof optionsOrLatency === \"number\" ? { latency: optionsOrLatency } : (optionsOrLatency ?? {});\n  const latency = opts.latency ?? 0;\n  const profile = opts.streamingProfile;\n  const signal = opts.signal;\n  const onChunkSent = opts.onChunkSent;\n\n  if (res.writableEnded) return true;\n  res.setHeader(\"Content-Type\", \"text/event-stream\");\n  res.setHeader(\"Cache-Control\", \"no-cache\");\n  res.setHeader(\"Connection\", \"keep-alive\");\n\n  let chunkIndex = 0;\n  for (const chunk of chunks) {\n    const chunkDelay = calculateDelay(chunkIndex, profile, latency);\n    if (chunkDelay > 0) await delay(chunkDelay, signal);\n    if (signal?.aborted) return false;\n    if (res.writableEnded) return true;\n    // Gemini uses data-only SSE (no event: prefix, no [DONE])\n    res.write(`data: ${JSON.stringify(chunk)}\\n\\n`);\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n    chunkIndex++;\n  }\n\n  if (!res.writableEnded) {\n    res.end();\n  }\n  return true;\n}\n\n// ─── Request handler ────────────────────────────────────────────────────────\n\nexport async function handleGemini(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  raw: string,\n  model: string,\n  streaming: boolean,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: HandlerDefaults,\n  setCorsHeaders: (res: http.ServerResponse) => void,\n  providerKey: RecordProviderKey = \"gemini\",\n): Promise<void> {\n  const { logger } = defaults;\n  setCorsHeaders(res);\n\n  let geminiReq: GeminiRequest;\n  try {\n    geminiReq = JSON.parse(raw) as GeminiRequest;\n  } catch {\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? `/v1beta/models/${model}:generateContent`,\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 400, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      400,\n      JSON.stringify({\n        error: {\n          message: \"Malformed JSON\",\n          code: 400,\n          status: \"INVALID_ARGUMENT\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Convert to ChatCompletionRequest for fixture matching\n  const completionReq = geminiToCompletionRequest(geminiReq, model, streaming);\n\n  const fixture = matchFixture(\n    fixtures,\n    completionReq,\n    journal.fixtureMatchCounts,\n    defaults.requestTransform,\n  );\n  const path = req.url ?? `/v1beta/models/${model}:generateContent`;\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures);\n  }\n\n  if (\n    applyChaos(\n      res,\n      fixture,\n      defaults.chaos,\n      req.headers,\n      journal,\n      {\n        method: req.method ?? \"POST\",\n        path,\n        headers: flattenHeaders(req.headers),\n        body: completionReq,\n      },\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\n\n  if (!fixture) {\n    if (defaults.record) {\n      const proxied = await proxyAndRecord(\n        req,\n        res,\n        completionReq,\n        providerKey,\n        path,\n        fixtures,\n        defaults,\n        raw,\n      );\n      if (proxied) {\n        journal.add({\n          method: req.method ?? \"POST\",\n          path,\n          headers: flattenHeaders(req.headers),\n          body: completionReq,\n          response: { status: res.statusCode ?? 200, fixture: null },\n        });\n        return;\n      }\n    }\n    const strictStatus = defaults.strict ? 503 : 404;\n    const strictMessage = defaults.strict\n      ? \"Strict mode: no fixture matched\"\n      : \"No fixture matched\";\n    if (defaults.strict) {\n      logger.error(`STRICT: No fixture matched for ${req.method ?? \"POST\"} ${path}`);\n    }\n    journal.add({\n      method: req.method ?? \"POST\",\n      path,\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status: strictStatus, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      strictStatus,\n      JSON.stringify({\n        error: {\n          message: strictMessage,\n          code: strictStatus,\n          status: defaults.strict ? \"UNAVAILABLE\" : \"NOT_FOUND\",\n        },\n      }),\n    );\n    return;\n  }\n\n  const response = fixture.response;\n  const latency = fixture.latency ?? defaults.latency;\n  const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);\n\n  // Error response\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method: req.method ?? \"POST\",\n      path,\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    // Gemini-style error format: { error: { code, message, status } }\n    const geminiError = {\n      error: {\n        code: status,\n        message: response.error.message,\n        status: response.error.type ?? \"ERROR\",\n      },\n    };\n    writeErrorResponse(res, status, JSON.stringify(geminiError));\n    return;\n  }\n\n  // Text response\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: req.method ?? \"POST\",\n      path,\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    if (!streaming) {\n      const body = buildGeminiTextResponse(response.content, response.reasoning);\n      res.writeHead(200, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify(body));\n    } else {\n      const chunks = buildGeminiTextStreamChunks(response.content, chunkSize, response.reasoning);\n      const interruption = createInterruptionSignal(fixture);\n      const completed = await writeGeminiSSEStream(res, chunks, {\n        latency,\n        streamingProfile: fixture.streamingProfile,\n        signal: interruption?.signal,\n        onChunkSent: interruption?.tick,\n      });\n      if (!completed) {\n        if (!res.writableEnded) res.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n      }\n      interruption?.cleanup();\n    }\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: req.method ?? \"POST\",\n      path,\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    if (!streaming) {\n      const body = buildGeminiToolCallResponse(response.toolCalls, logger);\n      res.writeHead(200, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify(body));\n    } else {\n      const chunks = buildGeminiToolCallStreamChunks(response.toolCalls, logger);\n      const interruption = createInterruptionSignal(fixture);\n      const completed = await writeGeminiSSEStream(res, chunks, {\n        latency,\n        streamingProfile: fixture.streamingProfile,\n        signal: interruption?.signal,\n        onChunkSent: interruption?.tick,\n      });\n      if (!completed) {\n        if (!res.writableEnded) res.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n      }\n      interruption?.cleanup();\n    }\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: req.method ?? \"POST\",\n    path,\n    headers: flattenHeaders(req.headers),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  writeErrorResponse(\n    res,\n    500,\n    JSON.stringify({\n      error: {\n        message: \"Fixture response did not match any known type\",\n        code: 500,\n        status: \"INTERNAL\",\n      },\n    }),\n  );\n}\n"],"mappings":";;;;;;;;AAwEA,SAAgB,0BACd,KACA,OACA,QACuB;CACvB,MAAM,WAA0B,EAAE;AAGlC,KAAI,IAAI,mBAAmB;EACzB,MAAM,OAAO,IAAI,kBAAkB,MAChC,QAAQ,MAAM,EAAE,SAAS,OAAU,CACnC,KAAK,MAAM,EAAE,KAAM,CACnB,KAAK,GAAG;AACX,MAAI,KACF,UAAS,KAAK;GAAE,MAAM;GAAU,SAAS;GAAM,CAAC;;AAIpD,KAAI,IAAI,SACN,MAAK,MAAM,WAAW,IAAI,UAAU;EAClC,MAAM,OAAO,QAAQ,QAAQ;AAE7B,MAAI,SAAS,QAAQ;GAEnB,MAAM,gBAAgB,QAAQ,MAAM,QAAQ,MAAM,EAAE,iBAAiB;GACrE,MAAM,YAAY,QAAQ,MAAM,QAAQ,MAAM,EAAE,SAAS,OAAU;AAEnE,OAAI,cAAc,SAAS,GAAG;AAE5B,SAAK,IAAI,IAAI,GAAG,IAAI,cAAc,QAAQ,KAAK;KAC7C,MAAM,OAAO,cAAc;AAC3B,cAAS,KAAK;MACZ,MAAM;MACN,SACE,OAAO,KAAK,iBAAkB,aAAa,WACvC,KAAK,iBAAkB,WACvB,KAAK,UAAU,KAAK,iBAAkB,SAAS;MACrD,cAAc,eAAe,KAAK,iBAAkB,KAAK,GAAG;MAC7D,CAAC;;AAGJ,QAAI,UAAU,SAAS,EACrB,UAAS,KAAK;KACZ,MAAM;KACN,SAAS,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;KAChD,CAAC;UAEC;IAEL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAQ,SAAS;KAAM,CAAC;;aAEvC,SAAS,SAAS;GAE3B,MAAM,YAAY,QAAQ,MAAM,QAAQ,MAAM,EAAE,aAAa;GAC7D,MAAM,YAAY,QAAQ,MAAM,QAAQ,MAAM,EAAE,SAAS,OAAU;AAEnE,OAAI,UAAU,SAAS,EACrB,UAAS,KAAK;IACZ,MAAM;IACN,SAAS;IACT,YAAY,UAAU,KAAK,GAAG,OAAO;KACnC,IAAI,eAAe,EAAE,aAAc,KAAK,GAAG;KAC3C,MAAM;KACN,UAAU;MACR,MAAM,EAAE,aAAc;MACtB,WAAW,KAAK,UAAU,EAAE,aAAc,KAAK;MAChD;KACF,EAAE;IACJ,CAAC;QACG;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAa,SAAS;KAAM,CAAC;;;;CAO3D,IAAI;AACJ,KAAI,IAAI,SAAS,IAAI,MAAM,SAAS,GAAG;EACrC,MAAM,QAAQ,IAAI,MAAM,SAAS,MAAM,EAAE,wBAAwB,EAAE,CAAC;AACpE,MAAI,MAAM,SAAS,EACjB,SAAQ,MAAM,KAAK,OAAO;GACxB,MAAM;GACN,UAAU;IACR,MAAM,EAAE;IACR,aAAa,EAAE;IACf,YAAY,EAAE;IACf;GACF,EAAE;;AAIP,QAAO;EACL;EACA;EACA;EACA,aAAa,IAAI,kBAAkB;EACnC;EACD;;AAkBH,SAAS,4BACP,SACA,WACA,WACuB;CACvB,MAAM,SAAgC,EAAE;AAGxC,KAAI,UACF,MAAK,IAAI,IAAI,GAAG,IAAI,UAAU,QAAQ,KAAK,WAAW;EACpD,MAAM,QAAQ,UAAU,MAAM,GAAG,IAAI,UAAU;AAC/C,SAAO,KAAK,EACV,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS,OAAO,CAAC;KAAE,MAAM;KAAO,SAAS;KAAM,CAAC;IAAE;GACnE,OAAO;GACR,CACF,EACF,CAAC;;AAKN,MAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,WAAW;EAClD,MAAM,QAAQ,QAAQ,MAAM,GAAG,IAAI,UAAU;EAC7C,MAAM,SAAS,IAAI,aAAa,QAAQ;EACxC,MAAM,QAA6B;GACjC,YAAY,CACV;IACE,SAAS;KAAE,MAAM;KAAS,OAAO,CAAC,EAAE,MAAM,OAAO,CAAC;KAAE;IACpD,OAAO;IACP,GAAI,SAAS,EAAE,cAAc,QAAQ,GAAG,EAAE;IAC3C,CACF;GACD,GAAI,SACA,EACE,eAAe;IACb,kBAAkB;IAClB,sBAAsB;IACtB,iBAAiB;IAClB,EACF,GACD,EAAE;GACP;AACD,SAAO,KAAK,MAAM;;AAIpB,KAAI,QAAQ,WAAW,EACrB,QAAO,KAAK;EACV,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;IAAE;GACjD,cAAc;GACd,OAAO;GACR,CACF;EACD,eAAe;GACb,kBAAkB;GAClB,sBAAsB;GACtB,iBAAiB;GAClB;EACF,CAAC;AAGJ,QAAO;;AAGT,SAAS,gCACP,WACA,QACuB;AAiBvB,QAAO,CACL;EACE,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS,OApBN,UAAU,KAAK,OAAO;KAChD,IAAI;AACJ,SAAI;AACF,gBAAU,KAAK,MAAM,GAAG,aAAa,KAAK;aACpC;AACN,aAAO,KACL,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,gBAAU,EAAE;;AAEd,YAAO,EACL,cAAc;MAAE,MAAM,GAAG;MAAM,MAAM;MAAS,IAAI,GAAG,MAAMA,oCAAoB;MAAE,EAClF;MACD;IAOuC;GACjC,cAAc;GACd,OAAO;GACR,CACF;EACD,eAAe;GACb,kBAAkB;GAClB,sBAAsB;GACtB,iBAAiB;GAClB;EACF,CACF;;AAKH,SAAS,wBAAwB,SAAiB,WAAyC;CACzF,MAAM,QAAsB,EAAE;AAC9B,KAAI,UACF,OAAM,KAAK;EAAE,MAAM;EAAW,SAAS;EAAM,CAAC;AAEhD,OAAM,KAAK,EAAE,MAAM,SAAS,CAAC;AAE7B,QAAO;EACL,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS;IAAO;GACjC,cAAc;GACd,OAAO;GACR,CACF;EACD,eAAe;GACb,kBAAkB;GAClB,sBAAsB;GACtB,iBAAiB;GAClB;EACF;;AAGH,SAAS,4BAA4B,WAAuB,QAAqC;AAgB/F,QAAO;EACL,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS,OAlBJ,UAAU,KAAK,OAAO;KAChD,IAAI;AACJ,SAAI;AACF,gBAAU,KAAK,MAAM,GAAG,aAAa,KAAK;aACpC;AACN,aAAO,KACL,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,gBAAU,EAAE;;AAEd,YAAO,EACL,cAAc;MAAE,MAAM,GAAG;MAAM,MAAM;MAAS,IAAI,GAAG,MAAMA,oCAAoB;MAAE,EAClF;MACD;IAKqC;GACjC,cAAc;GACd,OAAO;GACR,CACF;EACD,eAAe;GACb,kBAAkB;GAClB,sBAAsB;GACtB,iBAAiB;GAClB;EACF;;AAYH,eAAe,qBACb,KACA,QACA,kBACkB;CAClB,MAAM,OACJ,OAAO,qBAAqB,WAAW,EAAE,SAAS,kBAAkB,GAAI,oBAAoB,EAAE;CAChG,MAAM,UAAU,KAAK,WAAW;CAChC,MAAM,UAAU,KAAK;CACrB,MAAM,SAAS,KAAK;CACpB,MAAM,cAAc,KAAK;AAEzB,KAAI,IAAI,cAAe,QAAO;AAC9B,KAAI,UAAU,gBAAgB,oBAAoB;AAClD,KAAI,UAAU,iBAAiB,WAAW;AAC1C,KAAI,UAAU,cAAc,aAAa;CAEzC,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,aAAaC,kCAAe,YAAY,SAAS,QAAQ;AAC/D,MAAI,aAAa,EAAG,OAAMC,yBAAM,YAAY,OAAO;AACnD,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,IAAI,cAAe,QAAO;AAE9B,MAAI,MAAM,SAAS,KAAK,UAAU,MAAM,CAAC,MAAM;AAC/C,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;AAC5B;;AAGF,KAAI,CAAC,IAAI,cACP,KAAI,KAAK;AAEX,QAAO;;AAKT,eAAsB,aACpB,KACA,KACA,KACA,OACA,WACA,UACA,SACA,UACA,gBACA,cAAiC,UAClB;CACf,MAAM,EAAE,WAAW;AACnB,gBAAe,IAAI;CAEnB,IAAI;AACJ,KAAI;AACF,cAAY,KAAK,MAAM,IAAI;SACrB;AACN,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO,kBAAkB,MAAM;GACzC,SAASC,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,wCACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,QAAQ;GACT,EACF,CAAC,CACH;AACD;;CAIF,MAAM,gBAAgB,0BAA0B,WAAW,OAAO,UAAU;CAE5E,MAAM,UAAUC,4BACd,UACA,eACA,QAAQ,oBACR,SAAS,iBACV;CACD,MAAM,OAAO,IAAI,OAAO,kBAAkB,MAAM;AAEhD,KAAI,QACF,SAAQ,2BAA2B,SAAS,SAAS;AAGvD,KACEC,yBACE,KACA,SACA,SAAS,OACT,IAAI,SACJ,SACA;EACE,QAAQ,IAAI,UAAU;EACtB;EACA,SAASF,+BAAe,IAAI,QAAQ;EACpC,MAAM;EACP,EACD,SAAS,UACT,SAAS,OACV,CAED;AAEF,KAAI,CAAC,SAAS;AACZ,MAAI,SAAS,QAWX;OAVgB,MAAMG,gCACpB,KACA,KACA,eACA,aACA,MACA,UACA,UACA,IACD,EACY;AACX,YAAQ,IAAI;KACV,QAAQ,IAAI,UAAU;KACtB;KACA,SAASH,+BAAe,IAAI,QAAQ;KACpC,MAAM;KACN,UAAU;MAAE,QAAQ,IAAI,cAAc;MAAK,SAAS;MAAM;KAC3D,CAAC;AACF;;;EAGJ,MAAM,eAAe,SAAS,SAAS,MAAM;EAC7C,MAAM,gBAAgB,SAAS,SAC3B,oCACA;AACJ,MAAI,SAAS,OACX,QAAO,MAAM,kCAAkC,IAAI,UAAU,OAAO,GAAG,OAAO;AAEhF,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB;GACA,SAASA,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAc,SAAS;IAAM;GAClD,CAAC;AACF,wCACE,KACA,cACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,QAAQ,SAAS,SAAS,gBAAgB;GAC3C,EACF,CAAC,CACH;AACD;;CAGF,MAAM,WAAW,QAAQ;CACzB,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;AAGtE,KAAII,gCAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB;GACA,SAASJ,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;EAEF,MAAM,cAAc,EAClB,OAAO;GACL,MAAM;GACN,SAAS,SAAS,MAAM;GACxB,QAAQ,SAAS,MAAM,QAAQ;GAChC,EACF;AACD,wCAAmB,KAAK,QAAQ,KAAK,UAAU,YAAY,CAAC;AAC5D;;AAIF,KAAIK,+BAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ,IAAI,UAAU;GACtB;GACA,SAASL,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,MAAI,CAAC,WAAW;GACd,MAAM,OAAO,wBAAwB,SAAS,SAAS,SAAS,UAAU;AAC1E,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,KAAK,CAAC;SACxB;GACL,MAAM,SAAS,4BAA4B,SAAS,SAAS,WAAW,SAAS,UAAU;GAC3F,MAAM,eAAeM,8CAAyB,QAAQ;AAOtD,OAAI,CANc,MAAM,qBAAqB,KAAK,QAAQ;IACxD;IACA,kBAAkB,QAAQ;IAC1B,QAAQ,cAAc;IACtB,aAAa,cAAc;IAC5B,CAAC,EACc;AACd,QAAI,CAAC,IAAI,cAAe,KAAI,SAAS;AACrC,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,iBAAc,SAAS;;AAEzB;;AAIF,KAAIC,mCAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ,IAAI,UAAU;GACtB;GACA,SAASP,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,MAAI,CAAC,WAAW;GACd,MAAM,OAAO,4BAA4B,SAAS,WAAW,OAAO;AACpE,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,KAAK,CAAC;SACxB;GACL,MAAM,SAAS,gCAAgC,SAAS,WAAW,OAAO;GAC1E,MAAM,eAAeM,8CAAyB,QAAQ;AAOtD,OAAI,CANc,MAAM,qBAAqB,KAAK,QAAQ;IACxD;IACA,kBAAkB,QAAQ;IAC1B,QAAQ,cAAc;IACtB,aAAa,cAAc;IAC5B,CAAC,EACc;AACd,QAAI,CAAC,IAAI,cAAe,KAAI,SAAS;AACrC,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,iBAAc,SAAS;;AAEzB;;AAIF,SAAQ,IAAI;EACV,QAAQ,IAAI,UAAU;EACtB;EACA,SAASN,+BAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,uCACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;EACL,SAAS;EACT,MAAM;EACN,QAAQ;EACT,EACF,CAAC,CACH"}
+{"version":3,"file":"gemini.cjs","names":["generateToolCallId","calculateDelay","delay","flattenHeaders","getTestId","matchFixture","applyChaos","proxyAndRecord","isErrorResponse","isContentWithToolCallsResponse","createInterruptionSignal","isTextResponse","isToolCallResponse"],"sources":["../src/gemini.ts"],"sourcesContent":["/**\n * Google Gemini GenerateContent API support.\n *\n * Translates incoming Gemini requests into the ChatCompletionRequest format\n * used by the fixture router, and converts fixture responses back into the\n * Gemini GenerateContent streaming (or non-streaming) format.\n */\n\nimport type * as http from \"node:http\";\nimport type {\n  ChatCompletionRequest,\n  ChatMessage,\n  Fixture,\n  HandlerDefaults,\n  RecordProviderKey,\n  StreamingProfile,\n  ToolCall,\n  ToolDefinition,\n} from \"./types.js\";\nimport {\n  isTextResponse,\n  isToolCallResponse,\n  isContentWithToolCallsResponse,\n  isErrorResponse,\n  generateToolCallId,\n  flattenHeaders,\n  getTestId,\n} from \"./helpers.js\";\nimport { matchFixture } from \"./router.js\";\nimport { writeErrorResponse, delay, calculateDelay } from \"./sse-writer.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport type { Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport { applyChaos } from \"./chaos.js\";\nimport { proxyAndRecord } from \"./recorder.js\";\n\n// ─── Gemini request types ───────────────────────────────────────────────────\n\ninterface GeminiPart {\n  text?: string;\n  thought?: boolean;\n  functionCall?: { name: string; args: Record<string, unknown>; id?: string };\n  functionResponse?: { name: string; response: unknown };\n}\n\ninterface GeminiContent {\n  role?: string;\n  parts: GeminiPart[];\n}\n\ninterface GeminiFunctionDeclaration {\n  name: string;\n  description?: string;\n  parameters?: object;\n}\n\ninterface GeminiToolDef {\n  functionDeclarations?: GeminiFunctionDeclaration[];\n}\n\ninterface GeminiRequest {\n  contents?: GeminiContent[];\n  systemInstruction?: GeminiContent;\n  tools?: GeminiToolDef[];\n  generationConfig?: {\n    temperature?: number;\n    maxOutputTokens?: number;\n    [key: string]: unknown;\n  };\n  [key: string]: unknown;\n}\n\n// ─── Input conversion: Gemini → ChatCompletions messages ────────────────────\n\nexport function geminiToCompletionRequest(\n  req: GeminiRequest,\n  model: string,\n  stream: boolean,\n): ChatCompletionRequest {\n  const messages: ChatMessage[] = [];\n\n  // systemInstruction → system message\n  if (req.systemInstruction) {\n    const text = req.systemInstruction.parts\n      .filter((p) => p.text !== undefined)\n      .map((p) => p.text!)\n      .join(\"\");\n    if (text) {\n      messages.push({ role: \"system\", content: text });\n    }\n  }\n\n  if (req.contents) {\n    for (const content of req.contents) {\n      const role = content.role ?? \"user\";\n\n      if (role === \"user\") {\n        // Check for functionResponse parts\n        const funcResponses = content.parts.filter((p) => p.functionResponse);\n        const textParts = content.parts.filter((p) => p.text !== undefined);\n\n        if (funcResponses.length > 0) {\n          // functionResponse → tool message\n          for (let i = 0; i < funcResponses.length; i++) {\n            const part = funcResponses[i];\n            messages.push({\n              role: \"tool\",\n              content:\n                typeof part.functionResponse!.response === \"string\"\n                  ? part.functionResponse!.response\n                  : JSON.stringify(part.functionResponse!.response),\n              tool_call_id: `call_gemini_${part.functionResponse!.name}_${i}`,\n            });\n          }\n          // Any text parts alongside → user message\n          if (textParts.length > 0) {\n            messages.push({\n              role: \"user\",\n              content: textParts.map((p) => p.text!).join(\"\"),\n            });\n          }\n        } else {\n          // Regular user text\n          const text = textParts.map((p) => p.text!).join(\"\");\n          messages.push({ role: \"user\", content: text });\n        }\n      } else if (role === \"model\") {\n        // Check for functionCall parts\n        const funcCalls = content.parts.filter((p) => p.functionCall);\n        const textParts = content.parts.filter((p) => p.text !== undefined);\n\n        if (funcCalls.length > 0) {\n          messages.push({\n            role: \"assistant\",\n            content: null,\n            tool_calls: funcCalls.map((p, i) => ({\n              id: `call_gemini_${p.functionCall!.name}_${i}`,\n              type: \"function\" as const,\n              function: {\n                name: p.functionCall!.name,\n                arguments: JSON.stringify(p.functionCall!.args),\n              },\n            })),\n          });\n        } else {\n          const text = textParts.map((p) => p.text!).join(\"\");\n          messages.push({ role: \"assistant\", content: text });\n        }\n      }\n    }\n  }\n\n  // Convert tools\n  let tools: ToolDefinition[] | undefined;\n  if (req.tools && req.tools.length > 0) {\n    const decls = req.tools.flatMap((t) => t.functionDeclarations ?? []);\n    if (decls.length > 0) {\n      tools = decls.map((d) => ({\n        type: \"function\" as const,\n        function: {\n          name: d.name,\n          description: d.description,\n          parameters: d.parameters,\n        },\n      }));\n    }\n  }\n\n  return {\n    model,\n    messages,\n    stream,\n    temperature: req.generationConfig?.temperature,\n    tools,\n  };\n}\n\n// ─── Response building: fixture → Gemini format ─────────────────────────────\n\ninterface GeminiResponseChunk {\n  candidates: {\n    content: { role: string; parts: GeminiPart[] };\n    finishReason?: string;\n    index: number;\n  }[];\n  usageMetadata?: {\n    promptTokenCount: number;\n    candidatesTokenCount: number;\n    totalTokenCount: number;\n  };\n}\n\nfunction buildGeminiTextStreamChunks(\n  content: string,\n  chunkSize: number,\n  reasoning?: string,\n): GeminiResponseChunk[] {\n  const chunks: GeminiResponseChunk[] = [];\n\n  // Reasoning chunks (thought: true)\n  if (reasoning) {\n    for (let i = 0; i < reasoning.length; i += chunkSize) {\n      const slice = reasoning.slice(i, i + chunkSize);\n      chunks.push({\n        candidates: [\n          {\n            content: { role: \"model\", parts: [{ text: slice, thought: true }] },\n            index: 0,\n          },\n        ],\n      });\n    }\n  }\n\n  // Content chunks (original logic unchanged)\n  for (let i = 0; i < content.length; i += chunkSize) {\n    const slice = content.slice(i, i + chunkSize);\n    const isLast = i + chunkSize >= content.length;\n    const chunk: GeminiResponseChunk = {\n      candidates: [\n        {\n          content: { role: \"model\", parts: [{ text: slice }] },\n          index: 0,\n          ...(isLast ? { finishReason: \"STOP\" } : {}),\n        },\n      ],\n      ...(isLast\n        ? {\n            usageMetadata: {\n              promptTokenCount: 0,\n              candidatesTokenCount: 0,\n              totalTokenCount: 0,\n            },\n          }\n        : {}),\n    };\n    chunks.push(chunk);\n  }\n\n  // Handle empty content (original logic unchanged)\n  if (content.length === 0) {\n    chunks.push({\n      candidates: [\n        {\n          content: { role: \"model\", parts: [{ text: \"\" }] },\n          finishReason: \"STOP\",\n          index: 0,\n        },\n      ],\n      usageMetadata: {\n        promptTokenCount: 0,\n        candidatesTokenCount: 0,\n        totalTokenCount: 0,\n      },\n    });\n  }\n\n  return chunks;\n}\n\nfunction parseToolCallPart(tc: ToolCall, logger: Logger): GeminiPart {\n  let argsObj: Record<string, unknown>;\n  try {\n    argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n  } catch {\n    logger.warn(`Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`);\n    argsObj = {};\n  }\n  return { functionCall: { name: tc.name, args: argsObj, id: tc.id || generateToolCallId() } };\n}\n\nfunction buildGeminiToolCallStreamChunks(\n  toolCalls: ToolCall[],\n  logger: Logger,\n): GeminiResponseChunk[] {\n  const parts: GeminiPart[] = toolCalls.map((tc) => parseToolCallPart(tc, logger));\n\n  // Gemini sends all tool calls in a single response chunk\n  return [\n    {\n      candidates: [\n        {\n          content: { role: \"model\", parts },\n          finishReason: \"FUNCTION_CALL\",\n          index: 0,\n        },\n      ],\n      usageMetadata: {\n        promptTokenCount: 0,\n        candidatesTokenCount: 0,\n        totalTokenCount: 0,\n      },\n    },\n  ];\n}\n\n// Non-streaming response builders\n\nfunction buildGeminiTextResponse(content: string, reasoning?: string): GeminiResponseChunk {\n  const parts: GeminiPart[] = [];\n  if (reasoning) {\n    parts.push({ text: reasoning, thought: true });\n  }\n  parts.push({ text: content });\n\n  return {\n    candidates: [\n      {\n        content: { role: \"model\", parts },\n        finishReason: \"STOP\",\n        index: 0,\n      },\n    ],\n    usageMetadata: {\n      promptTokenCount: 0,\n      candidatesTokenCount: 0,\n      totalTokenCount: 0,\n    },\n  };\n}\n\nfunction buildGeminiToolCallResponse(toolCalls: ToolCall[], logger: Logger): GeminiResponseChunk {\n  const parts: GeminiPart[] = toolCalls.map((tc) => parseToolCallPart(tc, logger));\n\n  return {\n    candidates: [\n      {\n        content: { role: \"model\", parts },\n        finishReason: \"FUNCTION_CALL\",\n        index: 0,\n      },\n    ],\n    usageMetadata: {\n      promptTokenCount: 0,\n      candidatesTokenCount: 0,\n      totalTokenCount: 0,\n    },\n  };\n}\n\nfunction buildGeminiContentWithToolCallsStreamChunks(\n  content: string,\n  toolCalls: ToolCall[],\n  chunkSize: number,\n  logger: Logger,\n): GeminiResponseChunk[] {\n  const chunks: GeminiResponseChunk[] = [];\n\n  if (content.length === 0) {\n    chunks.push({\n      candidates: [\n        {\n          content: { role: \"model\", parts: [{ text: \"\" }] },\n          index: 0,\n        },\n      ],\n    });\n  } else {\n    for (let i = 0; i < content.length; i += chunkSize) {\n      const slice = content.slice(i, i + chunkSize);\n      chunks.push({\n        candidates: [\n          {\n            content: { role: \"model\", parts: [{ text: slice }] },\n            index: 0,\n          },\n        ],\n      });\n    }\n  }\n\n  const parts: GeminiPart[] = toolCalls.map((tc) => parseToolCallPart(tc, logger));\n\n  chunks.push({\n    candidates: [\n      {\n        content: { role: \"model\", parts },\n        finishReason: \"FUNCTION_CALL\",\n        index: 0,\n      },\n    ],\n    usageMetadata: {\n      promptTokenCount: 0,\n      candidatesTokenCount: 0,\n      totalTokenCount: 0,\n    },\n  });\n\n  return chunks;\n}\n\nfunction buildGeminiContentWithToolCallsResponse(\n  content: string,\n  toolCalls: ToolCall[],\n  logger: Logger,\n): GeminiResponseChunk {\n  const parts: GeminiPart[] = [\n    { text: content },\n    ...toolCalls.map((tc) => parseToolCallPart(tc, logger)),\n  ];\n\n  return {\n    candidates: [\n      {\n        content: { role: \"model\", parts },\n        finishReason: \"FUNCTION_CALL\",\n        index: 0,\n      },\n    ],\n    usageMetadata: {\n      promptTokenCount: 0,\n      candidatesTokenCount: 0,\n      totalTokenCount: 0,\n    },\n  };\n}\n\n// ─── SSE writer for Gemini streaming ────────────────────────────────────────\n\ninterface GeminiStreamOptions {\n  latency?: number;\n  streamingProfile?: StreamingProfile;\n  signal?: AbortSignal;\n  onChunkSent?: () => void;\n}\n\nasync function writeGeminiSSEStream(\n  res: http.ServerResponse,\n  chunks: GeminiResponseChunk[],\n  optionsOrLatency?: number | GeminiStreamOptions,\n): Promise<boolean> {\n  const opts: GeminiStreamOptions =\n    typeof optionsOrLatency === \"number\" ? { latency: optionsOrLatency } : (optionsOrLatency ?? {});\n  const latency = opts.latency ?? 0;\n  const profile = opts.streamingProfile;\n  const signal = opts.signal;\n  const onChunkSent = opts.onChunkSent;\n\n  if (res.writableEnded) return true;\n  res.setHeader(\"Content-Type\", \"text/event-stream\");\n  res.setHeader(\"Cache-Control\", \"no-cache\");\n  res.setHeader(\"Connection\", \"keep-alive\");\n\n  let chunkIndex = 0;\n  for (const chunk of chunks) {\n    const chunkDelay = calculateDelay(chunkIndex, profile, latency);\n    if (chunkDelay > 0) await delay(chunkDelay, signal);\n    if (signal?.aborted) return false;\n    if (res.writableEnded) return true;\n    // Gemini uses data-only SSE (no event: prefix, no [DONE])\n    res.write(`data: ${JSON.stringify(chunk)}\\n\\n`);\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n    chunkIndex++;\n  }\n\n  if (!res.writableEnded) {\n    res.end();\n  }\n  return true;\n}\n\n// ─── Request handler ────────────────────────────────────────────────────────\n\nexport async function handleGemini(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  raw: string,\n  model: string,\n  streaming: boolean,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: HandlerDefaults,\n  setCorsHeaders: (res: http.ServerResponse) => void,\n  providerKey: RecordProviderKey = \"gemini\",\n): Promise<void> {\n  const { logger } = defaults;\n  setCorsHeaders(res);\n\n  let geminiReq: GeminiRequest;\n  try {\n    geminiReq = JSON.parse(raw) as GeminiRequest;\n  } catch {\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? `/v1beta/models/${model}:generateContent`,\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 400, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      400,\n      JSON.stringify({\n        error: {\n          message: \"Malformed JSON\",\n          code: 400,\n          status: \"INVALID_ARGUMENT\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Convert to ChatCompletionRequest for fixture matching\n  const completionReq = geminiToCompletionRequest(geminiReq, model, streaming);\n\n  const testId = getTestId(req);\n  const fixture = matchFixture(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n  const path = req.url ?? `/v1beta/models/${model}:generateContent`;\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (\n    applyChaos(\n      res,\n      fixture,\n      defaults.chaos,\n      req.headers,\n      journal,\n      {\n        method: req.method ?? \"POST\",\n        path,\n        headers: flattenHeaders(req.headers),\n        body: completionReq,\n      },\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\n\n  if (!fixture) {\n    if (defaults.record) {\n      const proxied = await proxyAndRecord(\n        req,\n        res,\n        completionReq,\n        providerKey,\n        path,\n        fixtures,\n        defaults,\n        raw,\n      );\n      if (proxied) {\n        journal.add({\n          method: req.method ?? \"POST\",\n          path,\n          headers: flattenHeaders(req.headers),\n          body: completionReq,\n          response: { status: res.statusCode ?? 200, fixture: null },\n        });\n        return;\n      }\n    }\n    const strictStatus = defaults.strict ? 503 : 404;\n    const strictMessage = defaults.strict\n      ? \"Strict mode: no fixture matched\"\n      : \"No fixture matched\";\n    if (defaults.strict) {\n      logger.error(`STRICT: No fixture matched for ${req.method ?? \"POST\"} ${path}`);\n    }\n    journal.add({\n      method: req.method ?? \"POST\",\n      path,\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status: strictStatus, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      strictStatus,\n      JSON.stringify({\n        error: {\n          message: strictMessage,\n          code: strictStatus,\n          status: defaults.strict ? \"UNAVAILABLE\" : \"NOT_FOUND\",\n        },\n      }),\n    );\n    return;\n  }\n\n  const response = fixture.response;\n  const latency = fixture.latency ?? defaults.latency;\n  const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);\n\n  // Error response\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method: req.method ?? \"POST\",\n      path,\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    // Gemini-style error format: { error: { code, message, status } }\n    const geminiError = {\n      error: {\n        code: status,\n        message: response.error.message,\n        status: response.error.type ?? \"ERROR\",\n      },\n    };\n    writeErrorResponse(res, status, JSON.stringify(geminiError));\n    return;\n  }\n\n  // Content + tool calls response (must be checked before isTextResponse / isToolCallResponse)\n  if (isContentWithToolCallsResponse(response)) {\n    const journalEntry = journal.add({\n      method: req.method ?? \"POST\",\n      path,\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    if (!streaming) {\n      const body = buildGeminiContentWithToolCallsResponse(\n        response.content,\n        response.toolCalls,\n        logger,\n      );\n      res.writeHead(200, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify(body));\n    } else {\n      const chunks = buildGeminiContentWithToolCallsStreamChunks(\n        response.content,\n        response.toolCalls,\n        chunkSize,\n        logger,\n      );\n      const interruption = createInterruptionSignal(fixture);\n      const completed = await writeGeminiSSEStream(res, chunks, {\n        latency,\n        streamingProfile: fixture.streamingProfile,\n        signal: interruption?.signal,\n        onChunkSent: interruption?.tick,\n      });\n      if (!completed) {\n        if (!res.writableEnded) res.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n      }\n      interruption?.cleanup();\n    }\n    return;\n  }\n\n  // Text response\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: req.method ?? \"POST\",\n      path,\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    if (!streaming) {\n      const body = buildGeminiTextResponse(response.content, response.reasoning);\n      res.writeHead(200, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify(body));\n    } else {\n      const chunks = buildGeminiTextStreamChunks(response.content, chunkSize, response.reasoning);\n      const interruption = createInterruptionSignal(fixture);\n      const completed = await writeGeminiSSEStream(res, chunks, {\n        latency,\n        streamingProfile: fixture.streamingProfile,\n        signal: interruption?.signal,\n        onChunkSent: interruption?.tick,\n      });\n      if (!completed) {\n        if (!res.writableEnded) res.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n      }\n      interruption?.cleanup();\n    }\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: req.method ?? \"POST\",\n      path,\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    if (!streaming) {\n      const body = buildGeminiToolCallResponse(response.toolCalls, logger);\n      res.writeHead(200, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify(body));\n    } else {\n      const chunks = buildGeminiToolCallStreamChunks(response.toolCalls, logger);\n      const interruption = createInterruptionSignal(fixture);\n      const completed = await writeGeminiSSEStream(res, chunks, {\n        latency,\n        streamingProfile: fixture.streamingProfile,\n        signal: interruption?.signal,\n        onChunkSent: interruption?.tick,\n      });\n      if (!completed) {\n        if (!res.writableEnded) res.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n      }\n      interruption?.cleanup();\n    }\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: req.method ?? \"POST\",\n    path,\n    headers: flattenHeaders(req.headers),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  writeErrorResponse(\n    res,\n    500,\n    JSON.stringify({\n      error: {\n        message: \"Fixture response did not match any known type\",\n        code: 500,\n        status: \"INTERNAL\",\n      },\n    }),\n  );\n}\n"],"mappings":";;;;;;;;AA0EA,SAAgB,0BACd,KACA,OACA,QACuB;CACvB,MAAM,WAA0B,EAAE;AAGlC,KAAI,IAAI,mBAAmB;EACzB,MAAM,OAAO,IAAI,kBAAkB,MAChC,QAAQ,MAAM,EAAE,SAAS,OAAU,CACnC,KAAK,MAAM,EAAE,KAAM,CACnB,KAAK,GAAG;AACX,MAAI,KACF,UAAS,KAAK;GAAE,MAAM;GAAU,SAAS;GAAM,CAAC;;AAIpD,KAAI,IAAI,SACN,MAAK,MAAM,WAAW,IAAI,UAAU;EAClC,MAAM,OAAO,QAAQ,QAAQ;AAE7B,MAAI,SAAS,QAAQ;GAEnB,MAAM,gBAAgB,QAAQ,MAAM,QAAQ,MAAM,EAAE,iBAAiB;GACrE,MAAM,YAAY,QAAQ,MAAM,QAAQ,MAAM,EAAE,SAAS,OAAU;AAEnE,OAAI,cAAc,SAAS,GAAG;AAE5B,SAAK,IAAI,IAAI,GAAG,IAAI,cAAc,QAAQ,KAAK;KAC7C,MAAM,OAAO,cAAc;AAC3B,cAAS,KAAK;MACZ,MAAM;MACN,SACE,OAAO,KAAK,iBAAkB,aAAa,WACvC,KAAK,iBAAkB,WACvB,KAAK,UAAU,KAAK,iBAAkB,SAAS;MACrD,cAAc,eAAe,KAAK,iBAAkB,KAAK,GAAG;MAC7D,CAAC;;AAGJ,QAAI,UAAU,SAAS,EACrB,UAAS,KAAK;KACZ,MAAM;KACN,SAAS,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;KAChD,CAAC;UAEC;IAEL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAQ,SAAS;KAAM,CAAC;;aAEvC,SAAS,SAAS;GAE3B,MAAM,YAAY,QAAQ,MAAM,QAAQ,MAAM,EAAE,aAAa;GAC7D,MAAM,YAAY,QAAQ,MAAM,QAAQ,MAAM,EAAE,SAAS,OAAU;AAEnE,OAAI,UAAU,SAAS,EACrB,UAAS,KAAK;IACZ,MAAM;IACN,SAAS;IACT,YAAY,UAAU,KAAK,GAAG,OAAO;KACnC,IAAI,eAAe,EAAE,aAAc,KAAK,GAAG;KAC3C,MAAM;KACN,UAAU;MACR,MAAM,EAAE,aAAc;MACtB,WAAW,KAAK,UAAU,EAAE,aAAc,KAAK;MAChD;KACF,EAAE;IACJ,CAAC;QACG;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAa,SAAS;KAAM,CAAC;;;;CAO3D,IAAI;AACJ,KAAI,IAAI,SAAS,IAAI,MAAM,SAAS,GAAG;EACrC,MAAM,QAAQ,IAAI,MAAM,SAAS,MAAM,EAAE,wBAAwB,EAAE,CAAC;AACpE,MAAI,MAAM,SAAS,EACjB,SAAQ,MAAM,KAAK,OAAO;GACxB,MAAM;GACN,UAAU;IACR,MAAM,EAAE;IACR,aAAa,EAAE;IACf,YAAY,EAAE;IACf;GACF,EAAE;;AAIP,QAAO;EACL;EACA;EACA;EACA,aAAa,IAAI,kBAAkB;EACnC;EACD;;AAkBH,SAAS,4BACP,SACA,WACA,WACuB;CACvB,MAAM,SAAgC,EAAE;AAGxC,KAAI,UACF,MAAK,IAAI,IAAI,GAAG,IAAI,UAAU,QAAQ,KAAK,WAAW;EACpD,MAAM,QAAQ,UAAU,MAAM,GAAG,IAAI,UAAU;AAC/C,SAAO,KAAK,EACV,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS,OAAO,CAAC;KAAE,MAAM;KAAO,SAAS;KAAM,CAAC;IAAE;GACnE,OAAO;GACR,CACF,EACF,CAAC;;AAKN,MAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,WAAW;EAClD,MAAM,QAAQ,QAAQ,MAAM,GAAG,IAAI,UAAU;EAC7C,MAAM,SAAS,IAAI,aAAa,QAAQ;EACxC,MAAM,QAA6B;GACjC,YAAY,CACV;IACE,SAAS;KAAE,MAAM;KAAS,OAAO,CAAC,EAAE,MAAM,OAAO,CAAC;KAAE;IACpD,OAAO;IACP,GAAI,SAAS,EAAE,cAAc,QAAQ,GAAG,EAAE;IAC3C,CACF;GACD,GAAI,SACA,EACE,eAAe;IACb,kBAAkB;IAClB,sBAAsB;IACtB,iBAAiB;IAClB,EACF,GACD,EAAE;GACP;AACD,SAAO,KAAK,MAAM;;AAIpB,KAAI,QAAQ,WAAW,EACrB,QAAO,KAAK;EACV,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;IAAE;GACjD,cAAc;GACd,OAAO;GACR,CACF;EACD,eAAe;GACb,kBAAkB;GAClB,sBAAsB;GACtB,iBAAiB;GAClB;EACF,CAAC;AAGJ,QAAO;;AAGT,SAAS,kBAAkB,IAAc,QAA4B;CACnE,IAAI;AACJ,KAAI;AACF,YAAU,KAAK,MAAM,GAAG,aAAa,KAAK;SACpC;AACN,SAAO,KAAK,sDAAsD,GAAG,KAAK,KAAK,GAAG,YAAY;AAC9F,YAAU,EAAE;;AAEd,QAAO,EAAE,cAAc;EAAE,MAAM,GAAG;EAAM,MAAM;EAAS,IAAI,GAAG,MAAMA,oCAAoB;EAAE,EAAE;;AAG9F,SAAS,gCACP,WACA,QACuB;AAIvB,QAAO,CACL;EACE,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS,OAPN,UAAU,KAAK,OAAO,kBAAkB,IAAI,OAAO,CAAC;IAOvC;GACjC,cAAc;GACd,OAAO;GACR,CACF;EACD,eAAe;GACb,kBAAkB;GAClB,sBAAsB;GACtB,iBAAiB;GAClB;EACF,CACF;;AAKH,SAAS,wBAAwB,SAAiB,WAAyC;CACzF,MAAM,QAAsB,EAAE;AAC9B,KAAI,UACF,OAAM,KAAK;EAAE,MAAM;EAAW,SAAS;EAAM,CAAC;AAEhD,OAAM,KAAK,EAAE,MAAM,SAAS,CAAC;AAE7B,QAAO;EACL,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS;IAAO;GACjC,cAAc;GACd,OAAO;GACR,CACF;EACD,eAAe;GACb,kBAAkB;GAClB,sBAAsB;GACtB,iBAAiB;GAClB;EACF;;AAGH,SAAS,4BAA4B,WAAuB,QAAqC;AAG/F,QAAO;EACL,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS,OALJ,UAAU,KAAK,OAAO,kBAAkB,IAAI,OAAO,CAAC;IAKzC;GACjC,cAAc;GACd,OAAO;GACR,CACF;EACD,eAAe;GACb,kBAAkB;GAClB,sBAAsB;GACtB,iBAAiB;GAClB;EACF;;AAGH,SAAS,4CACP,SACA,WACA,WACA,QACuB;CACvB,MAAM,SAAgC,EAAE;AAExC,KAAI,QAAQ,WAAW,EACrB,QAAO,KAAK,EACV,YAAY,CACV;EACE,SAAS;GAAE,MAAM;GAAS,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;GAAE;EACjD,OAAO;EACR,CACF,EACF,CAAC;KAEF,MAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,WAAW;EAClD,MAAM,QAAQ,QAAQ,MAAM,GAAG,IAAI,UAAU;AAC7C,SAAO,KAAK,EACV,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS,OAAO,CAAC,EAAE,MAAM,OAAO,CAAC;IAAE;GACpD,OAAO;GACR,CACF,EACF,CAAC;;CAIN,MAAM,QAAsB,UAAU,KAAK,OAAO,kBAAkB,IAAI,OAAO,CAAC;AAEhF,QAAO,KAAK;EACV,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS;IAAO;GACjC,cAAc;GACd,OAAO;GACR,CACF;EACD,eAAe;GACb,kBAAkB;GAClB,sBAAsB;GACtB,iBAAiB;GAClB;EACF,CAAC;AAEF,QAAO;;AAGT,SAAS,wCACP,SACA,WACA,QACqB;AAMrB,QAAO;EACL,YAAY,CACV;GACE,SAAS;IAAE,MAAM;IAAS,OARJ,CAC1B,EAAE,MAAM,SAAS,EACjB,GAAG,UAAU,KAAK,OAAO,kBAAkB,IAAI,OAAO,CAAC,CACxD;IAKsC;GACjC,cAAc;GACd,OAAO;GACR,CACF;EACD,eAAe;GACb,kBAAkB;GAClB,sBAAsB;GACtB,iBAAiB;GAClB;EACF;;AAYH,eAAe,qBACb,KACA,QACA,kBACkB;CAClB,MAAM,OACJ,OAAO,qBAAqB,WAAW,EAAE,SAAS,kBAAkB,GAAI,oBAAoB,EAAE;CAChG,MAAM,UAAU,KAAK,WAAW;CAChC,MAAM,UAAU,KAAK;CACrB,MAAM,SAAS,KAAK;CACpB,MAAM,cAAc,KAAK;AAEzB,KAAI,IAAI,cAAe,QAAO;AAC9B,KAAI,UAAU,gBAAgB,oBAAoB;AAClD,KAAI,UAAU,iBAAiB,WAAW;AAC1C,KAAI,UAAU,cAAc,aAAa;CAEzC,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,aAAaC,kCAAe,YAAY,SAAS,QAAQ;AAC/D,MAAI,aAAa,EAAG,OAAMC,yBAAM,YAAY,OAAO;AACnD,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,IAAI,cAAe,QAAO;AAE9B,MAAI,MAAM,SAAS,KAAK,UAAU,MAAM,CAAC,MAAM;AAC/C,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;AAC5B;;AAGF,KAAI,CAAC,IAAI,cACP,KAAI,KAAK;AAEX,QAAO;;AAKT,eAAsB,aACpB,KACA,KACA,KACA,OACA,WACA,UACA,SACA,UACA,gBACA,cAAiC,UAClB;CACf,MAAM,EAAE,WAAW;AACnB,gBAAe,IAAI;CAEnB,IAAI;AACJ,KAAI;AACF,cAAY,KAAK,MAAM,IAAI;SACrB;AACN,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO,kBAAkB,MAAM;GACzC,SAASC,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,wCACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,QAAQ;GACT,EACF,CAAC,CACH;AACD;;CAIF,MAAM,gBAAgB,0BAA0B,WAAW,OAAO,UAAU;CAE5E,MAAM,SAASC,0BAAU,IAAI;CAC7B,MAAM,UAAUC,4BACd,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;CACD,MAAM,OAAO,IAAI,OAAO,kBAAkB,MAAM;AAEhD,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KACEC,yBACE,KACA,SACA,SAAS,OACT,IAAI,SACJ,SACA;EACE,QAAQ,IAAI,UAAU;EACtB;EACA,SAASH,+BAAe,IAAI,QAAQ;EACpC,MAAM;EACP,EACD,SAAS,UACT,SAAS,OACV,CAED;AAEF,KAAI,CAAC,SAAS;AACZ,MAAI,SAAS,QAWX;OAVgB,MAAMI,gCACpB,KACA,KACA,eACA,aACA,MACA,UACA,UACA,IACD,EACY;AACX,YAAQ,IAAI;KACV,QAAQ,IAAI,UAAU;KACtB;KACA,SAASJ,+BAAe,IAAI,QAAQ;KACpC,MAAM;KACN,UAAU;MAAE,QAAQ,IAAI,cAAc;MAAK,SAAS;MAAM;KAC3D,CAAC;AACF;;;EAGJ,MAAM,eAAe,SAAS,SAAS,MAAM;EAC7C,MAAM,gBAAgB,SAAS,SAC3B,oCACA;AACJ,MAAI,SAAS,OACX,QAAO,MAAM,kCAAkC,IAAI,UAAU,OAAO,GAAG,OAAO;AAEhF,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB;GACA,SAASA,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAc,SAAS;IAAM;GAClD,CAAC;AACF,wCACE,KACA,cACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,QAAQ,SAAS,SAAS,gBAAgB;GAC3C,EACF,CAAC,CACH;AACD;;CAGF,MAAM,WAAW,QAAQ;CACzB,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;AAGtE,KAAIK,gCAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB;GACA,SAASL,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;EAEF,MAAM,cAAc,EAClB,OAAO;GACL,MAAM;GACN,SAAS,SAAS,MAAM;GACxB,QAAQ,SAAS,MAAM,QAAQ;GAChC,EACF;AACD,wCAAmB,KAAK,QAAQ,KAAK,UAAU,YAAY,CAAC;AAC5D;;AAIF,KAAIM,+CAA+B,SAAS,EAAE;EAC5C,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ,IAAI,UAAU;GACtB;GACA,SAASN,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,MAAI,CAAC,WAAW;GACd,MAAM,OAAO,wCACX,SAAS,SACT,SAAS,WACT,OACD;AACD,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,KAAK,CAAC;SACxB;GACL,MAAM,SAAS,4CACb,SAAS,SACT,SAAS,WACT,WACA,OACD;GACD,MAAM,eAAeO,8CAAyB,QAAQ;AAOtD,OAAI,CANc,MAAM,qBAAqB,KAAK,QAAQ;IACxD;IACA,kBAAkB,QAAQ;IAC1B,QAAQ,cAAc;IACtB,aAAa,cAAc;IAC5B,CAAC,EACc;AACd,QAAI,CAAC,IAAI,cAAe,KAAI,SAAS;AACrC,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,iBAAc,SAAS;;AAEzB;;AAIF,KAAIC,+BAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ,IAAI,UAAU;GACtB;GACA,SAASR,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,MAAI,CAAC,WAAW;GACd,MAAM,OAAO,wBAAwB,SAAS,SAAS,SAAS,UAAU;AAC1E,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,KAAK,CAAC;SACxB;GACL,MAAM,SAAS,4BAA4B,SAAS,SAAS,WAAW,SAAS,UAAU;GAC3F,MAAM,eAAeO,8CAAyB,QAAQ;AAOtD,OAAI,CANc,MAAM,qBAAqB,KAAK,QAAQ;IACxD;IACA,kBAAkB,QAAQ;IAC1B,QAAQ,cAAc;IACtB,aAAa,cAAc;IAC5B,CAAC,EACc;AACd,QAAI,CAAC,IAAI,cAAe,KAAI,SAAS;AACrC,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,iBAAc,SAAS;;AAEzB;;AAIF,KAAIE,mCAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ,IAAI,UAAU;GACtB;GACA,SAAST,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,MAAI,CAAC,WAAW;GACd,MAAM,OAAO,4BAA4B,SAAS,WAAW,OAAO;AACpE,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,KAAK,CAAC;SACxB;GACL,MAAM,SAAS,gCAAgC,SAAS,WAAW,OAAO;GAC1E,MAAM,eAAeO,8CAAyB,QAAQ;AAOtD,OAAI,CANc,MAAM,qBAAqB,KAAK,QAAQ;IACxD;IACA,kBAAkB,QAAQ;IAC1B,QAAQ,cAAc;IACtB,aAAa,cAAc;IAC5B,CAAC,EACc;AACd,QAAI,CAAC,IAAI,cAAe,KAAI,SAAS;AACrC,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,iBAAc,SAAS;;AAEzB;;AAIF,SAAQ,IAAI;EACV,QAAQ,IAAI,UAAU;EACtB;EACA,SAASP,+BAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,uCACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;EACL,SAAS;EACT,MAAM;EACN,QAAQ;EACT,EACF,CAAC,CACH"}

package/dist/gemini.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"gemini.d.cts","names":[],"sources":["../src/gemini.ts"],"sourcesContent":[],"mappings":";;;;;;iBAgZsB,YAAA,MACf,IAAA,CAAK,sBACL,IAAA,CAAK,0EAIA,oBACD,mBACC,uCACY,IAAA,CAAK,uCACd,oBACZ"}
+{"version":3,"file":"gemini.d.cts","names":[],"sources":["../src/gemini.ts"],"sourcesContent":[],"mappings":";;;;;;iBAgdsB,YAAA,MACf,IAAA,CAAK,sBACL,IAAA,CAAK,0EAIA,oBACD,mBACC,uCACY,IAAA,CAAK,uCACd,oBACZ"}

package/dist/gemini.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"gemini.d.ts","names":[],"sources":["../src/gemini.ts"],"sourcesContent":[],"mappings":";;;;;;iBAgZsB,YAAA,MACf,IAAA,CAAK,sBACL,IAAA,CAAK,0EAIA,oBACD,mBACC,uCACY,IAAA,CAAK,uCACd,oBACZ"}
+{"version":3,"file":"gemini.d.ts","names":[],"sources":["../src/gemini.ts"],"sourcesContent":[],"mappings":";;;;;;iBAgdsB,YAAA,MACf,IAAA,CAAK,sBACL,IAAA,CAAK,0EAIA,oBACD,mBACC,uCACY,IAAA,CAAK,uCACd,oBACZ"}

package/dist/gemini.js

@@ -1,4 +1,4 @@
-import { flattenHeaders, generateToolCallId, isErrorResponse, isTextResponse, isToolCallResponse } from "./helpers.js";
+import { flattenHeaders, generateToolCallId, getTestId, isContentWithToolCallsResponse, isErrorResponse, isTextResponse, isToolCallResponse } from "./helpers.js";
 import { matchFixture } from "./router.js";
 import { calculateDelay, delay, writeErrorResponse } from "./sse-writer.js";
 import { createInterruptionSignal } from "./interruption.js";
@@ -136,25 +136,26 @@ function buildGeminiTextStreamChunks(content, chunkSize, reasoning) {
 	});
 	return chunks;
 }
+function parseToolCallPart(tc, logger) {
+	let argsObj;
+	try {
+		argsObj = JSON.parse(tc.arguments || "{}");
+	} catch {
+		logger.warn(`Malformed JSON in fixture tool call arguments for "${tc.name}": ${tc.arguments}`);
+		argsObj = {};
+	}
+	return { functionCall: {
+		name: tc.name,
+		args: argsObj,
+		id: tc.id || generateToolCallId()
+	} };
+}
 function buildGeminiToolCallStreamChunks(toolCalls, logger) {
 	return [{
 		candidates: [{
 			content: {
 				role: "model",
-				parts: toolCalls.map((tc) => {
-					let argsObj;
-					try {
-						argsObj = JSON.parse(tc.arguments || "{}");
-					} catch {
-						logger.warn(`Malformed JSON in fixture tool call arguments for "${tc.name}": ${tc.arguments}`);
-						argsObj = {};
-					}
-					return { functionCall: {
-						name: tc.name,
-						args: argsObj,
-						id: tc.id || generateToolCallId()
-					} };
-				})
+				parts: toolCalls.map((tc) => parseToolCallPart(tc, logger))
 			},
 			finishReason: "FUNCTION_CALL",
 			index: 0
@@ -194,20 +195,61 @@ function buildGeminiToolCallResponse(toolCalls, logger) {
 		candidates: [{
 			content: {
 				role: "model",
-				parts: toolCalls.map((tc) => {
-					let argsObj;
-					try {
-						argsObj = JSON.parse(tc.arguments || "{}");
-					} catch {
-						logger.warn(`Malformed JSON in fixture tool call arguments for "${tc.name}": ${tc.arguments}`);
-						argsObj = {};
-					}
-					return { functionCall: {
-						name: tc.name,
-						args: argsObj,
-						id: tc.id || generateToolCallId()
-					} };
-				})
+				parts: toolCalls.map((tc) => parseToolCallPart(tc, logger))
+			},
+			finishReason: "FUNCTION_CALL",
+			index: 0
+		}],
+		usageMetadata: {
+			promptTokenCount: 0,
+			candidatesTokenCount: 0,
+			totalTokenCount: 0
+		}
+	};
+}
+function buildGeminiContentWithToolCallsStreamChunks(content, toolCalls, chunkSize, logger) {
+	const chunks = [];
+	if (content.length === 0) chunks.push({ candidates: [{
+		content: {
+			role: "model",
+			parts: [{ text: "" }]
+		},
+		index: 0
+	}] });
+	else for (let i = 0; i < content.length; i += chunkSize) {
+		const slice = content.slice(i, i + chunkSize);
+		chunks.push({ candidates: [{
+			content: {
+				role: "model",
+				parts: [{ text: slice }]
+			},
+			index: 0
+		}] });
+	}
+	const parts = toolCalls.map((tc) => parseToolCallPart(tc, logger));
+	chunks.push({
+		candidates: [{
+			content: {
+				role: "model",
+				parts
+			},
+			finishReason: "FUNCTION_CALL",
+			index: 0
+		}],
+		usageMetadata: {
+			promptTokenCount: 0,
+			candidatesTokenCount: 0,
+			totalTokenCount: 0
+		}
+	});
+	return chunks;
+}
+function buildGeminiContentWithToolCallsResponse(content, toolCalls, logger) {
+	return {
+		candidates: [{
+			content: {
+				role: "model",
+				parts: [{ text: content }, ...toolCalls.map((tc) => parseToolCallPart(tc, logger))]
 			},
 			finishReason: "FUNCTION_CALL",
 			index: 0
@@ -268,9 +310,10 @@ async function handleGemini(req, res, raw, model, streaming, fixtures, journal,
 		return;
 	}
 	const completionReq = geminiToCompletionRequest(geminiReq, model, streaming);
-	const fixture = matchFixture(fixtures, completionReq, journal.fixtureMatchCounts, defaults.requestTransform);
+	const testId = getTestId(req);
+	const fixture = matchFixture(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
 	const path = req.url ?? `/v1beta/models/${model}:generateContent`;
-	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures);
+	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures, testId);
 	if (applyChaos(res, fixture, defaults.chaos, req.headers, journal, {
 		method: req.method ?? "POST",
 		path,
@@ -336,6 +379,38 @@ async function handleGemini(req, res, raw, model, streaming, fixtures, journal,
 		writeErrorResponse(res, status, JSON.stringify(geminiError));
 		return;
 	}
+	if (isContentWithToolCallsResponse(response)) {
+		const journalEntry = journal.add({
+			method: req.method ?? "POST",
+			path,
+			headers: flattenHeaders(req.headers),
+			body: completionReq,
+			response: {
+				status: 200,
+				fixture
+			}
+		});
+		if (!streaming) {
+			const body = buildGeminiContentWithToolCallsResponse(response.content, response.toolCalls, logger);
+			res.writeHead(200, { "Content-Type": "application/json" });
+			res.end(JSON.stringify(body));
+		} else {
+			const chunks = buildGeminiContentWithToolCallsStreamChunks(response.content, response.toolCalls, chunkSize, logger);
+			const interruption = createInterruptionSignal(fixture);
+			if (!await writeGeminiSSEStream(res, chunks, {
+				latency,
+				streamingProfile: fixture.streamingProfile,
+				signal: interruption?.signal,
+				onChunkSent: interruption?.tick
+			})) {
+				if (!res.writableEnded) res.destroy();
+				journalEntry.response.interrupted = true;
+				journalEntry.response.interruptReason = interruption?.reason();
+			}
+			interruption?.cleanup();
+		}
+		return;
+	}
 	if (isTextResponse(response)) {
 		const journalEntry = journal.add({
 			method: req.method ?? "POST",