@copilotkit/aimock 1.26.0 → 1.26.1

package/dist/video.js.map

@@ -1 +1 @@
-{"version":3,"file":"video.js","names":[],"sources":["../src/video.ts"],"sourcesContent":["import type * as http from \"node:http\";\nimport type { ChatCompletionRequest, Fixture, HandlerDefaults, VideoResponse } from \"./types.js\";\nimport {\n  isVideoResponse,\n  isErrorResponse,\n  serializeErrorResponse,\n  flattenHeaders,\n  getTestId,\n  resolveResponse,\n  resolveStrictMode,\n  strictOverrideField,\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\ninterface VideoRequest {\n  model?: string;\n  prompt: string;\n  [key: string]: unknown;\n}\n\n// ─── VideoStateMap with TTL and size bound ────────────────────────────────\n\nconst VIDEO_STATE_MAX_ENTRIES = 10_000;\nconst VIDEO_STATE_TTL_MS = 3_600_000; // 1 hour\n\ninterface VideoStateEntry {\n  video: VideoResponse[\"video\"];\n  createdAt: number;\n}\n\n/**\n * A Map wrapper for video state that enforces a maximum size and per-entry TTL.\n * Entries older than VIDEO_STATE_TTL_MS are lazily evicted on `get`.\n * When the map exceeds VIDEO_STATE_MAX_ENTRIES on `set`, the oldest entries\n * are removed to stay within bounds.\n */\nexport class VideoStateMap {\n  private readonly entries = new Map<string, VideoStateEntry>();\n  private readonly sweepTimer: ReturnType<typeof setInterval>;\n\n  constructor() {\n    // Proactive sweep every 60 seconds to evict expired entries\n    this.sweepTimer = setInterval(() => {\n      const now = Date.now();\n      for (const [key, entry] of this.entries) {\n        if (now - entry.createdAt > VIDEO_STATE_TTL_MS) {\n          this.entries.delete(key);\n        }\n      }\n    }, 60_000);\n    // Allow the process to exit even if the timer is still running\n    if (this.sweepTimer.unref) {\n      this.sweepTimer.unref();\n    }\n  }\n\n  getEntry(key: string): { video: VideoResponse[\"video\"]; createdAtUnix: number } | undefined {\n    const entry = this.entries.get(key);\n    if (!entry) return undefined;\n    if (Date.now() - entry.createdAt > VIDEO_STATE_TTL_MS) {\n      this.entries.delete(key);\n      return undefined;\n    }\n    return { video: entry.video, createdAtUnix: Math.floor(entry.createdAt / 1000) };\n  }\n\n  getCreatedAtUnix(key: string): number | undefined {\n    const e = this.getEntry(key);\n    return e?.createdAtUnix;\n  }\n\n  set(key: string, video: VideoResponse[\"video\"]): void {\n    this.entries.set(key, { video, createdAt: Date.now() });\n    // Evict oldest entries if over capacity\n    if (this.entries.size > VIDEO_STATE_MAX_ENTRIES) {\n      const excess = this.entries.size - VIDEO_STATE_MAX_ENTRIES;\n      const iter = this.entries.keys();\n      for (let i = 0; i < excess; i++) {\n        const next = iter.next();\n        if (!next.done) this.entries.delete(next.value);\n      }\n    }\n  }\n\n  delete(key: string): boolean {\n    return this.entries.delete(key);\n  }\n\n  clear(): void {\n    this.entries.clear();\n  }\n\n  destroy(): void {\n    clearInterval(this.sweepTimer);\n    this.entries.clear();\n  }\n\n  get size(): number {\n    return this.entries.size;\n  }\n}\n\nexport async function handleVideoCreate(\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  videoStates: VideoStateMap,\n): Promise<void> {\n  setCorsHeaders(res);\n  const path = req.url ?? \"/v1/videos\";\n  const method = req.method ?? \"POST\";\n\n  let videoReq: VideoRequest;\n  try {\n    videoReq = JSON.parse(raw) as VideoRequest;\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    journal.add({\n      method,\n      path,\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: ${detail}`,\n          type: \"invalid_request_error\",\n          code: \"invalid_json\",\n        },\n      }),\n    );\n    return;\n  }\n\n  if (!videoReq.prompt) {\n    journal.add({\n      method,\n      path,\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: { message: \"Missing required parameter: 'prompt'\", type: \"invalid_request_error\" },\n      }),\n    );\n    return;\n  }\n\n  const syntheticReq: ChatCompletionRequest = {\n    model: videoReq.model ?? \"sora-2\",\n    messages: [{ role: \"user\", content: videoReq.prompt }],\n    _endpointType: \"video\",\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    defaults.logger.debug(`Fixture matched: ${JSON.stringify(fixture.match).slice(0, 120)}`);\n  } else {\n    defaults.logger.debug(`No fixture matched for request`);\n  }\n\n  if (\n    applyChaos(\n      res,\n      fixture,\n      defaults.chaos,\n      req.headers,\n      journal,\n      { method, path, headers: flattenHeaders(req.headers), body: syntheticReq },\n      fixture ? \"fixture\" : \"proxy\",\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\n\n  if (!fixture) {\n    const effectiveStrict = resolveStrictMode(defaults.strict, req.headers);\n    if (effectiveStrict) {\n      journal.add({\n        method,\n        path,\n        headers: flattenHeaders(req.headers),\n        body: syntheticReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, req.headers),\n        },\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    if (defaults.record) {\n      const outcome = await proxyAndRecord(\n        req,\n        res,\n        syntheticReq,\n        \"openai\",\n        req.url ?? \"/v1/videos\",\n        fixtures,\n        defaults,\n        raw,\n      );\n      if (outcome === \"handled_by_hook\") return;\n      if (outcome !== \"not_configured\") {\n        journal.add({\n          method,\n          path,\n          headers: flattenHeaders(req.headers),\n          body: syntheticReq,\n          response: { status: res.statusCode ?? 200, fixture: null, source: \"proxy\" },\n        });\n        return;\n      }\n    }\n\n    journal.add({\n      method,\n      path,\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, req.headers),\n      },\n    });\n    writeErrorResponse(\n      res,\n      404,\n      JSON.stringify({\n        error: {\n          message: \"No fixture matched\",\n          type: \"invalid_request_error\",\n          code: \"no_fixture_match\",\n        },\n      }),\n    );\n    return;\n  }\n\n  const response = await resolveResponse(fixture, syntheticReq);\n\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method,\n      path,\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: { status, fixture },\n    });\n    writeErrorResponse(res, status, serializeErrorResponse(response), {\n      retryAfter: response.retryAfter,\n    });\n    return;\n  }\n\n  if (!isVideoResponse(response)) {\n    journal.add({\n      method,\n      path,\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: { message: \"Fixture response is not a video type\", type: \"server_error\" },\n      }),\n    );\n    return;\n  }\n\n  journal.add({\n    method,\n    path,\n    headers: flattenHeaders(req.headers),\n    body: syntheticReq,\n    response: { status: 200, fixture },\n  });\n\n  const video = response.video;\n\n  // Store for GET status checks\n  const stateKey = `${testId}:${video.id}`;\n  videoStates.set(stateKey, video);\n  const created_at = videoStates.getCreatedAtUnix(stateKey)!;\n\n  if (video.status === \"completed\") {\n    res.writeHead(200, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ id: video.id, status: video.status, url: video.url, created_at }));\n  } else {\n    res.writeHead(200, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ id: video.id, status: video.status, created_at }));\n  }\n}\n\nexport function handleVideoStatus(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  videoId: string,\n  journal: Journal,\n  defaults: HandlerDefaults,\n  setCorsHeaders: (res: http.ServerResponse) => void,\n  videoStates: VideoStateMap,\n): void {\n  setCorsHeaders(res);\n  const path = req.url ?? `/v1/videos/${videoId}`;\n  const method = req.method ?? \"GET\";\n\n  if (\n    applyChaos(\n      res,\n      null,\n      defaults.chaos,\n      req.headers,\n      journal,\n      { method, path, headers: flattenHeaders(req.headers), body: null },\n      \"internal\",\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\n\n  const testId = getTestId(req);\n  const stateKey = `${testId}:${videoId}`;\n  const entry = videoStates.getEntry(stateKey);\n\n  if (!entry) {\n    journal.add({\n      method,\n      path,\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 404, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      404,\n      JSON.stringify({ error: { message: `Video ${videoId} not found`, type: \"not_found\" } }),\n    );\n    return;\n  }\n\n  const { video, createdAtUnix: created_at } = entry;\n\n  journal.add({\n    method,\n    path,\n    headers: flattenHeaders(req.headers),\n    body: null,\n    response: { status: 200, fixture: null },\n  });\n\n  const body: Record<string, unknown> = {\n    id: video.id,\n    status: video.status,\n    created_at,\n  };\n  if (video.url) body.url = video.url;\n\n  res.writeHead(200, { \"Content-Type\": \"application/json\" });\n  res.end(JSON.stringify(body));\n}\n"],"mappings":";;;;;;;AA0BA,MAAM,0BAA0B;AAChC,MAAM,qBAAqB;;;;;;;AAa3B,IAAa,gBAAb,MAA2B;CACzB,AAAiB,0BAAU,IAAI,KAA8B;CAC7D,AAAiB;CAEjB,cAAc;AAEZ,OAAK,aAAa,kBAAkB;GAClC,MAAM,MAAM,KAAK,KAAK;AACtB,QAAK,MAAM,CAAC,KAAK,UAAU,KAAK,QAC9B,KAAI,MAAM,MAAM,YAAY,mBAC1B,MAAK,QAAQ,OAAO,IAAI;KAG3B,IAAO;AAEV,MAAI,KAAK,WAAW,MAClB,MAAK,WAAW,OAAO;;CAI3B,SAAS,KAAmF;EAC1F,MAAM,QAAQ,KAAK,QAAQ,IAAI,IAAI;AACnC,MAAI,CAAC,MAAO,QAAO;AACnB,MAAI,KAAK,KAAK,GAAG,MAAM,YAAY,oBAAoB;AACrD,QAAK,QAAQ,OAAO,IAAI;AACxB;;AAEF,SAAO;GAAE,OAAO,MAAM;GAAO,eAAe,KAAK,MAAM,MAAM,YAAY,IAAK;GAAE;;CAGlF,iBAAiB,KAAiC;AAEhD,SADU,KAAK,SAAS,IAAI,EAClB;;CAGZ,IAAI,KAAa,OAAqC;AACpD,OAAK,QAAQ,IAAI,KAAK;GAAE;GAAO,WAAW,KAAK,KAAK;GAAE,CAAC;AAEvD,MAAI,KAAK,QAAQ,OAAO,yBAAyB;GAC/C,MAAM,SAAS,KAAK,QAAQ,OAAO;GACnC,MAAM,OAAO,KAAK,QAAQ,MAAM;AAChC,QAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,KAAK;IAC/B,MAAM,OAAO,KAAK,MAAM;AACxB,QAAI,CAAC,KAAK,KAAM,MAAK,QAAQ,OAAO,KAAK,MAAM;;;;CAKrD,OAAO,KAAsB;AAC3B,SAAO,KAAK,QAAQ,OAAO,IAAI;;CAGjC,QAAc;AACZ,OAAK,QAAQ,OAAO;;CAGtB,UAAgB;AACd,gBAAc,KAAK,WAAW;AAC9B,OAAK,QAAQ,OAAO;;CAGtB,IAAI,OAAe;AACjB,SAAO,KAAK,QAAQ;;;AAIxB,eAAsB,kBACpB,KACA,KACA,KACA,UACA,SACA,UACA,gBACA,aACe;AACf,gBAAe,IAAI;CACnB,MAAM,OAAO,IAAI,OAAO;CACxB,MAAM,SAAS,IAAI,UAAU;CAE7B,IAAI;AACJ,KAAI;AACF,aAAW,KAAK,MAAM,IAAI;UACnB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,UAAQ,IAAI;GACV;GACA;GACA,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,mBAAmB;GAC5B,MAAM;GACN,MAAM;GACP,EACF,CAAC,CACH;AACD;;AAGF,KAAI,CAAC,SAAS,QAAQ;AACpB,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GAAE,SAAS;GAAwC,MAAM;GAAyB,EAC1F,CAAC,CACH;AACD;;CAGF,MAAM,eAAsC;EAC1C,OAAO,SAAS,SAAS;EACzB,UAAU,CAAC;GAAE,MAAM;GAAQ,SAAS,SAAS;GAAQ,CAAC;EACtD,eAAe;EAChB;CAED,MAAM,SAAS,UAAU,IAAI;CAC7B,MAAM,UAAU,aACd,UACA,cACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;AAED,KAAI,SAAS;AACX,UAAQ,2BAA2B,SAAS,UAAU,OAAO;AAC7D,WAAS,OAAO,MAAM,oBAAoB,KAAK,UAAU,QAAQ,MAAM,CAAC,MAAM,GAAG,IAAI,GAAG;OAExF,UAAS,OAAO,MAAM,iCAAiC;AAGzD,KACE,WACE,KACA,SACA,SAAS,OACT,IAAI,SACJ,SACA;EAAE;EAAQ;EAAM,SAAS,eAAe,IAAI,QAAQ;EAAE,MAAM;EAAc,EAC1E,UAAU,YAAY,SACtB,SAAS,UACT,SAAS,OACV,CAED;AAEF,KAAI,CAAC,SAAS;AAEZ,MADwB,kBAAkB,SAAS,QAAQ,IAAI,QAAQ,EAClD;AACnB,WAAQ,IAAI;IACV;IACA;IACA,SAAS,eAAe,IAAI,QAAQ;IACpC,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAG,oBAAoB,SAAS,QAAQ,IAAI,QAAQ;KACrD;IACF,CAAC;AACF,sBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;IACL,SAAS;IACT,MAAM;IACN,MAAM;IACP,EACF,CAAC,CACH;AACD;;AAEF,MAAI,SAAS,QAAQ;GACnB,MAAM,UAAU,MAAM,eACpB,KACA,KACA,cACA,UACA,IAAI,OAAO,cACX,UACA,UACA,IACD;AACD,OAAI,YAAY,kBAAmB;AACnC,OAAI,YAAY,kBAAkB;AAChC,YAAQ,IAAI;KACV;KACA;KACA,SAAS,eAAe,IAAI,QAAQ;KACpC,MAAM;KACN,UAAU;MAAE,QAAQ,IAAI,cAAc;MAAK,SAAS;MAAM,QAAQ;MAAS;KAC5E,CAAC;AACF;;;AAIJ,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAG,oBAAoB,SAAS,QAAQ,IAAI,QAAQ;IACrD;GACF,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,MAAM;GACP,EACF,CAAC,CACH;AACD;;CAGF,MAAM,WAAW,MAAM,gBAAgB,SAAS,aAAa;AAE7D,KAAI,gBAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,qBAAmB,KAAK,QAAQ,uBAAuB,SAAS,EAAE,EAChE,YAAY,SAAS,YACtB,CAAC;AACF;;AAGF,KAAI,CAAC,gBAAgB,SAAS,EAAE;AAC9B,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GAAE,SAAS;GAAwC,MAAM;GAAgB,EACjF,CAAC,CACH;AACD;;AAGF,SAAQ,IAAI;EACV;EACA;EACA,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;CAEF,MAAM,QAAQ,SAAS;CAGvB,MAAM,WAAW,GAAG,OAAO,GAAG,MAAM;AACpC,aAAY,IAAI,UAAU,MAAM;CAChC,MAAM,aAAa,YAAY,iBAAiB,SAAS;AAEzD,KAAI,MAAM,WAAW,aAAa;AAChC,MAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,MAAI,IAAI,KAAK,UAAU;GAAE,IAAI,MAAM;GAAI,QAAQ,MAAM;GAAQ,KAAK,MAAM;GAAK;GAAY,CAAC,CAAC;QACtF;AACL,MAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,MAAI,IAAI,KAAK,UAAU;GAAE,IAAI,MAAM;GAAI,QAAQ,MAAM;GAAQ;GAAY,CAAC,CAAC;;;AAI/E,SAAgB,kBACd,KACA,KACA,SACA,SACA,UACA,gBACA,aACM;AACN,gBAAe,IAAI;CACnB,MAAM,OAAO,IAAI,OAAO,cAAc;CACtC,MAAM,SAAS,IAAI,UAAU;AAE7B,KACE,WACE,KACA,MACA,SAAS,OACT,IAAI,SACJ,SACA;EAAE;EAAQ;EAAM,SAAS,eAAe,IAAI,QAAQ;EAAE,MAAM;EAAM,EAClE,YACA,SAAS,UACT,SAAS,OACV,CAED;CAGF,MAAM,WAAW,GADF,UAAU,IAAI,CACF,GAAG;CAC9B,MAAM,QAAQ,YAAY,SAAS,SAAS;AAE5C,KAAI,CAAC,OAAO;AACV,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EAAE,OAAO;GAAE,SAAS,SAAS,QAAQ;GAAa,MAAM;GAAa,EAAE,CAAC,CACxF;AACD;;CAGF,MAAM,EAAE,OAAO,eAAe,eAAe;AAE7C,SAAQ,IAAI;EACV;EACA;EACA,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK,SAAS;GAAM;EACzC,CAAC;CAEF,MAAM,OAAgC;EACpC,IAAI,MAAM;EACV,QAAQ,MAAM;EACd;EACD;AACD,KAAI,MAAM,IAAK,MAAK,MAAM,MAAM;AAEhC,KAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,KAAI,IAAI,KAAK,UAAU,KAAK,CAAC"}
+{"version":3,"file":"video.js","names":[],"sources":["../src/video.ts"],"sourcesContent":["import type * as http from \"node:http\";\nimport type { ChatCompletionRequest, Fixture, HandlerDefaults, VideoResponse } from \"./types.js\";\nimport {\n  isVideoResponse,\n  isErrorResponse,\n  serializeErrorResponse,\n  flattenHeaders,\n  getTestId,\n  resolveResponse,\n  resolveStrictMode,\n  strictOverrideField,\n  getContext,\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\ninterface VideoRequest {\n  model?: string;\n  prompt: string;\n  [key: string]: unknown;\n}\n\n// ─── VideoStateMap with TTL and size bound ────────────────────────────────\n\nconst VIDEO_STATE_MAX_ENTRIES = 10_000;\nconst VIDEO_STATE_TTL_MS = 3_600_000; // 1 hour\n\ninterface VideoStateEntry {\n  video: VideoResponse[\"video\"];\n  createdAt: number;\n}\n\n/**\n * A Map wrapper for video state that enforces a maximum size and per-entry TTL.\n * Entries older than VIDEO_STATE_TTL_MS are lazily evicted on `get`.\n * When the map exceeds VIDEO_STATE_MAX_ENTRIES on `set`, the oldest entries\n * are removed to stay within bounds.\n */\nexport class VideoStateMap {\n  private readonly entries = new Map<string, VideoStateEntry>();\n  private readonly sweepTimer: ReturnType<typeof setInterval>;\n\n  constructor() {\n    // Proactive sweep every 60 seconds to evict expired entries\n    this.sweepTimer = setInterval(() => {\n      const now = Date.now();\n      for (const [key, entry] of this.entries) {\n        if (now - entry.createdAt > VIDEO_STATE_TTL_MS) {\n          this.entries.delete(key);\n        }\n      }\n    }, 60_000);\n    // Allow the process to exit even if the timer is still running\n    if (this.sweepTimer.unref) {\n      this.sweepTimer.unref();\n    }\n  }\n\n  getEntry(key: string): { video: VideoResponse[\"video\"]; createdAtUnix: number } | undefined {\n    const entry = this.entries.get(key);\n    if (!entry) return undefined;\n    if (Date.now() - entry.createdAt > VIDEO_STATE_TTL_MS) {\n      this.entries.delete(key);\n      return undefined;\n    }\n    return { video: entry.video, createdAtUnix: Math.floor(entry.createdAt / 1000) };\n  }\n\n  getCreatedAtUnix(key: string): number | undefined {\n    const e = this.getEntry(key);\n    return e?.createdAtUnix;\n  }\n\n  set(key: string, video: VideoResponse[\"video\"]): void {\n    this.entries.set(key, { video, createdAt: Date.now() });\n    // Evict oldest entries if over capacity\n    if (this.entries.size > VIDEO_STATE_MAX_ENTRIES) {\n      const excess = this.entries.size - VIDEO_STATE_MAX_ENTRIES;\n      const iter = this.entries.keys();\n      for (let i = 0; i < excess; i++) {\n        const next = iter.next();\n        if (!next.done) this.entries.delete(next.value);\n      }\n    }\n  }\n\n  delete(key: string): boolean {\n    return this.entries.delete(key);\n  }\n\n  clear(): void {\n    this.entries.clear();\n  }\n\n  destroy(): void {\n    clearInterval(this.sweepTimer);\n    this.entries.clear();\n  }\n\n  get size(): number {\n    return this.entries.size;\n  }\n}\n\nexport async function handleVideoCreate(\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  videoStates: VideoStateMap,\n): Promise<void> {\n  setCorsHeaders(res);\n  const path = req.url ?? \"/v1/videos\";\n  const method = req.method ?? \"POST\";\n\n  let videoReq: VideoRequest;\n  try {\n    videoReq = JSON.parse(raw) as VideoRequest;\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    journal.add({\n      method,\n      path,\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: ${detail}`,\n          type: \"invalid_request_error\",\n          code: \"invalid_json\",\n        },\n      }),\n    );\n    return;\n  }\n\n  if (!videoReq.prompt) {\n    journal.add({\n      method,\n      path,\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: { message: \"Missing required parameter: 'prompt'\", type: \"invalid_request_error\" },\n      }),\n    );\n    return;\n  }\n\n  const syntheticReq: ChatCompletionRequest = {\n    model: videoReq.model ?? \"sora-2\",\n    messages: [{ role: \"user\", content: videoReq.prompt }],\n    _endpointType: \"video\",\n    _context: getContext(req),\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    defaults.logger.debug(`Fixture matched: ${JSON.stringify(fixture.match).slice(0, 120)}`);\n  } else {\n    defaults.logger.debug(`No fixture matched for request`);\n  }\n\n  if (\n    applyChaos(\n      res,\n      fixture,\n      defaults.chaos,\n      req.headers,\n      journal,\n      { method, path, headers: flattenHeaders(req.headers), body: syntheticReq },\n      fixture ? \"fixture\" : \"proxy\",\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\n\n  if (!fixture) {\n    const effectiveStrict = resolveStrictMode(defaults.strict, req.headers);\n    if (effectiveStrict) {\n      journal.add({\n        method,\n        path,\n        headers: flattenHeaders(req.headers),\n        body: syntheticReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, req.headers),\n        },\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    if (defaults.record) {\n      const outcome = await proxyAndRecord(\n        req,\n        res,\n        syntheticReq,\n        \"openai\",\n        req.url ?? \"/v1/videos\",\n        fixtures,\n        defaults,\n        raw,\n      );\n      if (outcome === \"handled_by_hook\") return;\n      if (outcome !== \"not_configured\") {\n        journal.add({\n          method,\n          path,\n          headers: flattenHeaders(req.headers),\n          body: syntheticReq,\n          response: { status: res.statusCode ?? 200, fixture: null, source: \"proxy\" },\n        });\n        return;\n      }\n    }\n\n    journal.add({\n      method,\n      path,\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, req.headers),\n      },\n    });\n    writeErrorResponse(\n      res,\n      404,\n      JSON.stringify({\n        error: {\n          message: \"No fixture matched\",\n          type: \"invalid_request_error\",\n          code: \"no_fixture_match\",\n        },\n      }),\n    );\n    return;\n  }\n\n  const response = await resolveResponse(fixture, syntheticReq);\n\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method,\n      path,\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: { status, fixture },\n    });\n    writeErrorResponse(res, status, serializeErrorResponse(response), {\n      retryAfter: response.retryAfter,\n    });\n    return;\n  }\n\n  if (!isVideoResponse(response)) {\n    journal.add({\n      method,\n      path,\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: { message: \"Fixture response is not a video type\", type: \"server_error\" },\n      }),\n    );\n    return;\n  }\n\n  journal.add({\n    method,\n    path,\n    headers: flattenHeaders(req.headers),\n    body: syntheticReq,\n    response: { status: 200, fixture },\n  });\n\n  const video = response.video;\n\n  // Store for GET status checks\n  const stateKey = `${testId}:${video.id}`;\n  videoStates.set(stateKey, video);\n  const created_at = videoStates.getCreatedAtUnix(stateKey)!;\n\n  if (video.status === \"completed\") {\n    res.writeHead(200, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ id: video.id, status: video.status, url: video.url, created_at }));\n  } else {\n    res.writeHead(200, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ id: video.id, status: video.status, created_at }));\n  }\n}\n\nexport function handleVideoStatus(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  videoId: string,\n  journal: Journal,\n  defaults: HandlerDefaults,\n  setCorsHeaders: (res: http.ServerResponse) => void,\n  videoStates: VideoStateMap,\n): void {\n  setCorsHeaders(res);\n  const path = req.url ?? `/v1/videos/${videoId}`;\n  const method = req.method ?? \"GET\";\n\n  if (\n    applyChaos(\n      res,\n      null,\n      defaults.chaos,\n      req.headers,\n      journal,\n      { method, path, headers: flattenHeaders(req.headers), body: null },\n      \"internal\",\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\n\n  const testId = getTestId(req);\n  const stateKey = `${testId}:${videoId}`;\n  const entry = videoStates.getEntry(stateKey);\n\n  if (!entry) {\n    journal.add({\n      method,\n      path,\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 404, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      404,\n      JSON.stringify({ error: { message: `Video ${videoId} not found`, type: \"not_found\" } }),\n    );\n    return;\n  }\n\n  const { video, createdAtUnix: created_at } = entry;\n\n  journal.add({\n    method,\n    path,\n    headers: flattenHeaders(req.headers),\n    body: null,\n    response: { status: 200, fixture: null },\n  });\n\n  const body: Record<string, unknown> = {\n    id: video.id,\n    status: video.status,\n    created_at,\n  };\n  if (video.url) body.url = video.url;\n\n  res.writeHead(200, { \"Content-Type\": \"application/json\" });\n  res.end(JSON.stringify(body));\n}\n"],"mappings":";;;;;;;AA2BA,MAAM,0BAA0B;AAChC,MAAM,qBAAqB;;;;;;;AAa3B,IAAa,gBAAb,MAA2B;CACzB,AAAiB,0BAAU,IAAI,KAA8B;CAC7D,AAAiB;CAEjB,cAAc;AAEZ,OAAK,aAAa,kBAAkB;GAClC,MAAM,MAAM,KAAK,KAAK;AACtB,QAAK,MAAM,CAAC,KAAK,UAAU,KAAK,QAC9B,KAAI,MAAM,MAAM,YAAY,mBAC1B,MAAK,QAAQ,OAAO,IAAI;KAG3B,IAAO;AAEV,MAAI,KAAK,WAAW,MAClB,MAAK,WAAW,OAAO;;CAI3B,SAAS,KAAmF;EAC1F,MAAM,QAAQ,KAAK,QAAQ,IAAI,IAAI;AACnC,MAAI,CAAC,MAAO,QAAO;AACnB,MAAI,KAAK,KAAK,GAAG,MAAM,YAAY,oBAAoB;AACrD,QAAK,QAAQ,OAAO,IAAI;AACxB;;AAEF,SAAO;GAAE,OAAO,MAAM;GAAO,eAAe,KAAK,MAAM,MAAM,YAAY,IAAK;GAAE;;CAGlF,iBAAiB,KAAiC;AAEhD,SADU,KAAK,SAAS,IAAI,EAClB;;CAGZ,IAAI,KAAa,OAAqC;AACpD,OAAK,QAAQ,IAAI,KAAK;GAAE;GAAO,WAAW,KAAK,KAAK;GAAE,CAAC;AAEvD,MAAI,KAAK,QAAQ,OAAO,yBAAyB;GAC/C,MAAM,SAAS,KAAK,QAAQ,OAAO;GACnC,MAAM,OAAO,KAAK,QAAQ,MAAM;AAChC,QAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,KAAK;IAC/B,MAAM,OAAO,KAAK,MAAM;AACxB,QAAI,CAAC,KAAK,KAAM,MAAK,QAAQ,OAAO,KAAK,MAAM;;;;CAKrD,OAAO,KAAsB;AAC3B,SAAO,KAAK,QAAQ,OAAO,IAAI;;CAGjC,QAAc;AACZ,OAAK,QAAQ,OAAO;;CAGtB,UAAgB;AACd,gBAAc,KAAK,WAAW;AAC9B,OAAK,QAAQ,OAAO;;CAGtB,IAAI,OAAe;AACjB,SAAO,KAAK,QAAQ;;;AAIxB,eAAsB,kBACpB,KACA,KACA,KACA,UACA,SACA,UACA,gBACA,aACe;AACf,gBAAe,IAAI;CACnB,MAAM,OAAO,IAAI,OAAO;CACxB,MAAM,SAAS,IAAI,UAAU;CAE7B,IAAI;AACJ,KAAI;AACF,aAAW,KAAK,MAAM,IAAI;UACnB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,UAAQ,IAAI;GACV;GACA;GACA,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,mBAAmB;GAC5B,MAAM;GACN,MAAM;GACP,EACF,CAAC,CACH;AACD;;AAGF,KAAI,CAAC,SAAS,QAAQ;AACpB,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GAAE,SAAS;GAAwC,MAAM;GAAyB,EAC1F,CAAC,CACH;AACD;;CAGF,MAAM,eAAsC;EAC1C,OAAO,SAAS,SAAS;EACzB,UAAU,CAAC;GAAE,MAAM;GAAQ,SAAS,SAAS;GAAQ,CAAC;EACtD,eAAe;EACf,UAAU,WAAW,IAAI;EAC1B;CAED,MAAM,SAAS,UAAU,IAAI;CAC7B,MAAM,UAAU,aACd,UACA,cACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;AAED,KAAI,SAAS;AACX,UAAQ,2BAA2B,SAAS,UAAU,OAAO;AAC7D,WAAS,OAAO,MAAM,oBAAoB,KAAK,UAAU,QAAQ,MAAM,CAAC,MAAM,GAAG,IAAI,GAAG;OAExF,UAAS,OAAO,MAAM,iCAAiC;AAGzD,KACE,WACE,KACA,SACA,SAAS,OACT,IAAI,SACJ,SACA;EAAE;EAAQ;EAAM,SAAS,eAAe,IAAI,QAAQ;EAAE,MAAM;EAAc,EAC1E,UAAU,YAAY,SACtB,SAAS,UACT,SAAS,OACV,CAED;AAEF,KAAI,CAAC,SAAS;AAEZ,MADwB,kBAAkB,SAAS,QAAQ,IAAI,QAAQ,EAClD;AACnB,WAAQ,IAAI;IACV;IACA;IACA,SAAS,eAAe,IAAI,QAAQ;IACpC,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAG,oBAAoB,SAAS,QAAQ,IAAI,QAAQ;KACrD;IACF,CAAC;AACF,sBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;IACL,SAAS;IACT,MAAM;IACN,MAAM;IACP,EACF,CAAC,CACH;AACD;;AAEF,MAAI,SAAS,QAAQ;GACnB,MAAM,UAAU,MAAM,eACpB,KACA,KACA,cACA,UACA,IAAI,OAAO,cACX,UACA,UACA,IACD;AACD,OAAI,YAAY,kBAAmB;AACnC,OAAI,YAAY,kBAAkB;AAChC,YAAQ,IAAI;KACV;KACA;KACA,SAAS,eAAe,IAAI,QAAQ;KACpC,MAAM;KACN,UAAU;MAAE,QAAQ,IAAI,cAAc;MAAK,SAAS;MAAM,QAAQ;MAAS;KAC5E,CAAC;AACF;;;AAIJ,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAG,oBAAoB,SAAS,QAAQ,IAAI,QAAQ;IACrD;GACF,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,MAAM;GACP,EACF,CAAC,CACH;AACD;;CAGF,MAAM,WAAW,MAAM,gBAAgB,SAAS,aAAa;AAE7D,KAAI,gBAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,qBAAmB,KAAK,QAAQ,uBAAuB,SAAS,EAAE,EAChE,YAAY,SAAS,YACtB,CAAC;AACF;;AAGF,KAAI,CAAC,gBAAgB,SAAS,EAAE;AAC9B,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GAAE,SAAS;GAAwC,MAAM;GAAgB,EACjF,CAAC,CACH;AACD;;AAGF,SAAQ,IAAI;EACV;EACA;EACA,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;CAEF,MAAM,QAAQ,SAAS;CAGvB,MAAM,WAAW,GAAG,OAAO,GAAG,MAAM;AACpC,aAAY,IAAI,UAAU,MAAM;CAChC,MAAM,aAAa,YAAY,iBAAiB,SAAS;AAEzD,KAAI,MAAM,WAAW,aAAa;AAChC,MAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,MAAI,IAAI,KAAK,UAAU;GAAE,IAAI,MAAM;GAAI,QAAQ,MAAM;GAAQ,KAAK,MAAM;GAAK;GAAY,CAAC,CAAC;QACtF;AACL,MAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,MAAI,IAAI,KAAK,UAAU;GAAE,IAAI,MAAM;GAAI,QAAQ,MAAM;GAAQ;GAAY,CAAC,CAAC;;;AAI/E,SAAgB,kBACd,KACA,KACA,SACA,SACA,UACA,gBACA,aACM;AACN,gBAAe,IAAI;CACnB,MAAM,OAAO,IAAI,OAAO,cAAc;CACtC,MAAM,SAAS,IAAI,UAAU;AAE7B,KACE,WACE,KACA,MACA,SAAS,OACT,IAAI,SACJ,SACA;EAAE;EAAQ;EAAM,SAAS,eAAe,IAAI,QAAQ;EAAE,MAAM;EAAM,EAClE,YACA,SAAS,UACT,SAAS,OACV,CAED;CAGF,MAAM,WAAW,GADF,UAAU,IAAI,CACF,GAAG;CAC9B,MAAM,QAAQ,YAAY,SAAS,SAAS;AAE5C,KAAI,CAAC,OAAO;AACV,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EAAE,OAAO;GAAE,SAAS,SAAS,QAAQ;GAAa,MAAM;GAAa,EAAE,CAAC,CACxF;AACD;;CAGF,MAAM,EAAE,OAAO,eAAe,eAAe;AAE7C,SAAQ,IAAI;EACV;EACA;EACA,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK,SAAS;GAAM;EACzC,CAAC;CAEF,MAAM,OAAgC;EACpC,IAAI,MAAM;EACV,QAAQ,MAAM;EACd;EACD;AACD,KAAI,MAAM,IAAK,MAAK,MAAM,MAAM;AAEhC,KAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,KAAI,IAAI,KAAK,UAAU,KAAK,CAAC"}

package/dist/ws-gemini-live.cjs

@@ -189,12 +189,15 @@ async function processMessage(raw, ws, fixtures, journal, defaults, session) {
 		} }));
 		return;
 	}
+	const geminiContextHeader = defaults.upgradeHeaders?.["x-aimock-context"];
+	const geminiContext = typeof geminiContextHeader === "string" ? geminiContextHeader : Array.isArray(geminiContextHeader) && geminiContextHeader.length > 0 ? geminiContextHeader[0] : void 0;
 	const completionReq = {
 		model: session.model,
 		messages: [...session.conversationHistory, ...newMessages],
 		stream: true,
 		tools: session.tools.length > 0 ? session.tools : void 0,
-		_endpointType: "chat"
+		_endpointType: "chat",
+		_context: geminiContext
 	};
 	const testId = defaults.testId ?? require_constants.DEFAULT_TEST_ID;
 	const fixture = require_router.matchFixture(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);

package/dist/ws-gemini-live.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"ws-gemini-live.cjs","names":["DEFAULT_TEST_ID","matchFixture","resolveStrictMode","flattenHeaders","strictOverrideField","resolveResponse","isErrorResponse","isAudioResponse","formatToMime","isContentWithToolCallsResponse","createInterruptionSignal","calculateDelay","delay","generateToolCallId","isTextResponse","isToolCallResponse"],"sources":["../src/ws-gemini-live.ts"],"sourcesContent":["/**\n * WebSocket handler for Gemini Live BidiGenerateContent API.\n *\n * Accepts setup, clientContent, and toolResponse messages over WebSocket\n * and responds with setupComplete, serverContent, toolCall, and error\n * messages in the Gemini Live streaming format.\n */\n\nimport type {\n  Fixture,\n  ChatMessage,\n  ChatCompletionRequest,\n  ToolDefinition,\n  AudioResponse,\n} from \"./types.js\";\nimport { matchFixture } from \"./router.js\";\nimport {\n  isTextResponse,\n  isToolCallResponse,\n  isContentWithToolCallsResponse,\n  isErrorResponse,\n  isAudioResponse,\n  flattenHeaders,\n  formatToMime,\n  generateToolCallId,\n  resolveResponse,\n  resolveStrictMode,\n  strictOverrideField,\n} from \"./helpers.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\nimport { DEFAULT_TEST_ID, type Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport type { WebSocketConnection } from \"./ws-framing.js\";\n\n// ─── Gemini Live protocol types ─────────────────────────────────────────────\n\ninterface GeminiLivePart {\n  text?: string;\n  thought?: boolean;\n  functionCall?: { name: string; args: Record<string, unknown> };\n  functionResponse?: { name: string; response: unknown; id?: string };\n  inlineData?: { mimeType: string; data: string };\n}\n\ninterface GeminiLiveTurn {\n  role: string;\n  parts: GeminiLivePart[];\n}\n\ninterface GeminiLiveFunctionDeclaration {\n  name: string;\n  description?: string;\n  parameters?: object;\n}\n\ninterface GeminiLiveToolDef {\n  functionDeclarations?: GeminiLiveFunctionDeclaration[];\n}\n\ninterface GeminiLiveSetup {\n  model?: string;\n  generationConfig?: Record<string, unknown>;\n  tools?: GeminiLiveToolDef[];\n}\n\ninterface GeminiLiveClientContent {\n  turns: GeminiLiveTurn[];\n  turnComplete?: boolean;\n}\n\ninterface GeminiLiveFunctionResponse {\n  id?: string;\n  name: string;\n  response: unknown;\n}\n\ninterface GeminiLiveToolResponse {\n  functionResponses: GeminiLiveFunctionResponse[];\n}\n\ninterface GeminiLiveMessage {\n  setup?: GeminiLiveSetup;\n  config?: GeminiLiveSetup;\n  clientContent?: GeminiLiveClientContent;\n  toolResponse?: GeminiLiveToolResponse;\n}\n\n// ─── Session state ──────────────────────────────────────────────────────────\n\ninterface SessionState {\n  setupDone: boolean;\n  model: string;\n  tools: ToolDefinition[];\n  conversationHistory: ChatMessage[];\n}\n\n// ─── Helpers ────────────────────────────────────────────────────────────────\n\nconst WS_PATH = \"/ws/google.ai.generativelanguage.v1beta.GenerativeService.BidiGenerateContent\";\n\n/**\n * Map HTTP status codes to gRPC error codes.\n * Gemini Live uses gRPC codes, not HTTP status codes.\n */\nfunction httpToGrpc(httpCode: number): number {\n  switch (httpCode) {\n    case 400:\n      return 3; // INVALID_ARGUMENT\n    case 401:\n      return 16; // UNAUTHENTICATED\n    case 403:\n      return 7; // PERMISSION_DENIED\n    case 404:\n      return 5; // NOT_FOUND\n    case 409:\n      return 10; // ABORTED\n    case 429:\n      return 8; // RESOURCE_EXHAUSTED\n    case 501:\n      return 12; // UNIMPLEMENTED\n    case 503:\n      return 14; // UNAVAILABLE\n    default:\n      return 13; // INTERNAL\n  }\n}\n\n/**\n * Convert Gemini Live turns into ChatMessage[] for fixture matching.\n */\nfunction geminiTurnsToMessages(turns: GeminiLiveTurn[]): ChatMessage[] {\n  const messages: ChatMessage[] = [];\n\n  for (const turn of turns) {\n    const role = turn.role ?? \"user\";\n\n    if (role === \"user\") {\n      const funcResponses = turn.parts.filter((p) => p.functionResponse);\n      // inlineData parts (e.g. client audio input) are silently skipped —\n      // only text and functionResponse parts are relevant for fixture matching.\n      const textParts = turn.parts.filter((p) => p.text !== undefined && !p.thought);\n\n      if (funcResponses.length > 0) {\n        for (let i = 0; i < funcResponses.length; i++) {\n          const part = funcResponses[i];\n          const fr = part.functionResponse!;\n          messages.push({\n            role: \"tool\",\n            content: typeof fr.response === \"string\" ? fr.response : JSON.stringify(fr.response),\n            tool_call_id: fr.id ?? `call_gemini_${fr.name}_${i}`,\n          });\n        }\n        if (textParts.length > 0) {\n          messages.push({\n            role: \"user\",\n            content: textParts.map((p) => p.text!).join(\"\"),\n          });\n        }\n      } else {\n        const text = textParts.map((p) => p.text!).join(\"\");\n        messages.push({ role: \"user\", content: text });\n      }\n    } else if (role === \"model\") {\n      const funcCalls = turn.parts.filter((p) => p.functionCall);\n      const textParts = turn.parts.filter((p) => p.text !== undefined && !p.thought);\n\n      if (funcCalls.length > 0) {\n        const text = textParts.map((p) => p.text!).join(\"\");\n        messages.push({\n          role: \"assistant\",\n          content: text || 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  return messages;\n}\n\n/**\n * Convert toolResponse messages into ChatMessage[] for fixture matching.\n */\nfunction toolResponseToMessages(toolResponse: GeminiLiveToolResponse): ChatMessage[] {\n  return toolResponse.functionResponses.map((fr, i) => ({\n    role: \"tool\" as const,\n    content: typeof fr.response === \"string\" ? fr.response : JSON.stringify(fr.response),\n    tool_call_id: fr.id ?? `call_gemini_${fr.name}_${i}`,\n  }));\n}\n\n/**\n * Convert Gemini tool definitions to ChatCompletion ToolDefinition[].\n */\nfunction convertTools(geminiTools?: GeminiLiveToolDef[]): ToolDefinition[] {\n  if (!geminiTools || geminiTools.length === 0) return [];\n  const decls = geminiTools.flatMap((t) => t.functionDeclarations ?? []);\n  return 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// ─── Main handler ───────────────────────────────────────────────────────────\n\nexport function handleWebSocketGeminiLive(\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): void {\n  const { logger } = defaults;\n  const session: SessionState = {\n    setupDone: false,\n    model: defaults.model,\n    tools: [],\n    conversationHistory: [],\n  };\n\n  let pending = Promise.resolve();\n  ws.on(\"message\", (raw: string) => {\n    pending = pending.then(() =>\n      processMessage(raw, ws, fixtures, journal, defaults, session).catch((err: unknown) => {\n        const msg = err instanceof Error ? err.message : \"Internal error\";\n        logger.error(`WebSocket Gemini Live error: ${msg}`);\n        try {\n          ws.send(\n            JSON.stringify({\n              error: { code: 13, message: msg, status: \"INTERNAL\" },\n            }),\n          );\n        } catch (sendErr) {\n          defaults.logger.debug(\n            `Failed to send error to client: ${sendErr instanceof Error ? sendErr.message : \"unknown\"}`,\n          );\n        }\n      }),\n    );\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n  session: SessionState,\n): Promise<void> {\n  let parsed: GeminiLiveMessage;\n  try {\n    parsed = JSON.parse(raw) as GeminiLiveMessage;\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    ws.send(\n      JSON.stringify({\n        error: { code: 3, message: `Malformed JSON: ${detail}`, status: \"INVALID_ARGUMENT\" },\n      }),\n    );\n    return;\n  }\n\n  // Handle setup message (accept both `setup` and `config` as aliases)\n  const setupMsg = parsed.setup ?? parsed.config;\n  if (setupMsg) {\n    session.setupDone = true;\n    session.model = setupMsg.model ?? defaults.model;\n    session.tools = convertTools(setupMsg.tools);\n    ws.send(JSON.stringify({ setupComplete: {} }));\n    return;\n  }\n\n  // Reject messages before setup\n  if (!session.setupDone) {\n    ws.send(\n      JSON.stringify({\n        error: { code: 9, message: \"Setup required\", status: \"FAILED_PRECONDITION\" },\n      }),\n    );\n    return;\n  }\n\n  // Build messages from this interaction\n  let newMessages: ChatMessage[];\n\n  if (parsed.clientContent) {\n    if (!parsed.clientContent.turns || !Array.isArray(parsed.clientContent.turns)) {\n      ws.send(\n        JSON.stringify({\n          error: {\n            code: 3,\n            message: \"Missing 'turns' in clientContent\",\n            status: \"INVALID_ARGUMENT\",\n          },\n        }),\n      );\n      return;\n    }\n    newMessages = geminiTurnsToMessages(parsed.clientContent.turns);\n  } else if (parsed.toolResponse) {\n    if (\n      !parsed.toolResponse.functionResponses ||\n      !Array.isArray(parsed.toolResponse.functionResponses)\n    ) {\n      ws.send(\n        JSON.stringify({\n          error: {\n            code: 3,\n            message: \"Missing 'functionResponses' in toolResponse\",\n            status: \"INVALID_ARGUMENT\",\n          },\n        }),\n      );\n      return;\n    }\n    newMessages = toolResponseToMessages(parsed.toolResponse);\n  } else {\n    ws.send(\n      JSON.stringify({\n        error: {\n          code: 3,\n          message: \"Expected clientContent or toolResponse\",\n          status: \"INVALID_ARGUMENT\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Build completion request for fixture matching (include new messages speculatively)\n  const completionReq: ChatCompletionRequest = {\n    model: session.model,\n    messages: [...session.conversationHistory, ...newMessages],\n    stream: true,\n    tools: session.tools.length > 0 ? session.tools : undefined,\n    _endpointType: \"chat\",\n  };\n\n  const testId = defaults.testId ?? DEFAULT_TEST_ID;\n  const fixture = matchFixture(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n  const path = WS_PATH;\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (!fixture) {\n    if (resolveStrictMode(defaults.strict, defaults.upgradeHeaders)) {\n      defaults.logger.warn(`STRICT: No fixture matched for WebSocket message`);\n      journal.add({\n        method: \"WS\",\n        path,\n        headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n        body: completionReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n        },\n      });\n      ws.close(1008, \"Strict mode: no fixture matched\");\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n      },\n    });\n    ws.send(\n      JSON.stringify({\n        error: { code: 5, message: \"No fixture matched\", status: \"NOT_FOUND\" },\n      }),\n    );\n    return;\n  }\n\n  // Commit messages to conversation history only after successful fixture match\n  session.conversationHistory.push(...newMessages);\n\n  const response = await resolveResponse(fixture, completionReq);\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: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    ws.send(\n      JSON.stringify({\n        error: {\n          code: httpToGrpc(status),\n          message: response.error.message,\n          status: response.error.type ?? \"INTERNAL\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Audio response — single frame with inlineData and turnComplete: true\n  if (isAudioResponse(response)) {\n    journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const audioResp = response as AudioResponse;\n    let mimeType: string;\n    let data: string;\n\n    if (typeof audioResp.audio === \"string\") {\n      mimeType = formatToMime(audioResp.format ?? \"mp3\");\n      data = audioResp.audio;\n    } else {\n      mimeType = audioResp.audio.contentType ?? \"audio/mpeg\";\n      data = audioResp.audio.b64Json;\n    }\n\n    ws.send(\n      JSON.stringify({\n        serverContent: {\n          modelTurn: {\n            parts: [{ inlineData: { mimeType, data } }],\n          },\n          turnComplete: true,\n        },\n      }),\n    );\n\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: \"[audio]\",\n    });\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: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const content = response.content;\n    const chunkList: string[] = [];\n    for (let i = 0; i < content.length; i += chunkSize) {\n      chunkList.push(content.slice(i, i + chunkSize));\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n    let interrupted = false;\n\n    // Stream text content chunks (turnComplete omitted — sent as a separate message later)\n    if (content.length === 0) {\n      if (!ws.isClosed) {\n        ws.send(\n          JSON.stringify({\n            serverContent: {\n              modelTurn: { parts: [{ text: \"\" }] },\n            },\n          }),\n        );\n      }\n    } else {\n      for (let i = 0; i < chunkList.length; i++) {\n        if (ws.isClosed) break;\n        const chunkDelay = calculateDelay(i, undefined, latency, recordedTimings, replaySpeed);\n        if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);\n        if (interruption?.signal.aborted) {\n          interrupted = true;\n          break;\n        }\n        if (ws.isClosed) break;\n\n        ws.send(\n          JSON.stringify({\n            serverContent: {\n              modelTurn: { parts: [{ text: chunkList[i] }] },\n            },\n          }),\n        );\n        interruption?.tick();\n        if (interruption?.signal.aborted) {\n          interrupted = true;\n          break;\n        }\n      }\n    }\n\n    if (interrupted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    // Pre-compute tool calls with stable IDs so wire message and history match\n    const resolvedToolCalls = response.toolCalls.map((tc) => ({\n      ...tc,\n      resolvedId: tc.id ?? generateToolCallId(),\n    }));\n\n    // Send tool calls\n    if (!ws.isClosed) {\n      const tcDelay = calculateDelay(\n        chunkList.length,\n        undefined,\n        latency,\n        recordedTimings,\n        replaySpeed,\n      );\n      if (tcDelay > 0) await delay(tcDelay, interruption?.signal);\n      if (interruption?.signal.aborted) {\n        ws.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n        interruption?.cleanup();\n        return;\n      }\n\n      const functionCalls = resolvedToolCalls.map((tc) => {\n        let argsObj: Record<string, unknown>;\n        try {\n          argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n        } catch {\n          defaults.logger.warn(\n            `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n          );\n          argsObj = {};\n        }\n        return {\n          name: tc.name,\n          args: argsObj,\n          id: tc.resolvedId,\n        };\n      });\n\n      ws.send(JSON.stringify({ toolCall: { functionCalls } }));\n      interruption?.tick();\n    }\n\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send turnComplete\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add to conversation history using the same resolved IDs from the wire message\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: content || null,\n      tool_calls: resolvedToolCalls.map((tc) => ({\n        id: tc.resolvedId,\n        type: \"function\" as const,\n        function: {\n          name: tc.name,\n          arguments: tc.arguments,\n        },\n      })),\n    });\n    return;\n  }\n\n  // Text response — stream chunks with serverContent\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const content = response.content;\n\n    if (content.length === 0) {\n      if (ws.isClosed) return;\n      // Empty content: send empty modelTurn, then separate turnComplete\n      ws.send(\n        JSON.stringify({\n          serverContent: {\n            modelTurn: { parts: [{ text: \"\" }] },\n          },\n        }),\n      );\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n      return;\n    }\n\n    // Chunk the content\n    const chunks: string[] = [];\n    for (let i = 0; i < content.length; i += chunkSize) {\n      chunks.push(content.slice(i, i + chunkSize));\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n    let interrupted = false;\n\n    // Stream content chunks without turnComplete (sent separately after)\n    for (let i = 0; i < chunks.length; i++) {\n      if (ws.isClosed) break;\n      const chunkDelay = calculateDelay(i, undefined, latency, recordedTimings, replaySpeed);\n      if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);\n      if (interruption?.signal.aborted) {\n        interrupted = true;\n        break;\n      }\n      if (ws.isClosed) break;\n\n      ws.send(\n        JSON.stringify({\n          serverContent: {\n            modelTurn: { parts: [{ text: chunks[i] }] },\n          },\n        }),\n      );\n      interruption?.tick();\n      if (interruption?.signal.aborted) {\n        interrupted = true;\n        break;\n      }\n    }\n\n    if (interrupted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send separate turnComplete message\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add assistant response to conversation history\n    session.conversationHistory.push({ role: \"assistant\", content });\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n\n    if (ws.isClosed) {\n      interruption?.cleanup();\n      return;\n    }\n    const tcDelay = calculateDelay(0, undefined, latency, recordedTimings, replaySpeed);\n    if (tcDelay > 0) await delay(tcDelay, interruption?.signal);\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n    if (ws.isClosed) {\n      interruption?.cleanup();\n      return;\n    }\n\n    const functionCalls = response.toolCalls.map((tc, i) => {\n      let argsObj: Record<string, unknown>;\n      try {\n        argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n      } catch {\n        defaults.logger.warn(\n          `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n        );\n        argsObj = {};\n      }\n      return {\n        name: tc.name,\n        args: argsObj,\n        id: tc.id ?? `call_gemini_${tc.name}_${i}`,\n      };\n    });\n\n    ws.send(JSON.stringify({ toolCall: { functionCalls } }));\n    interruption?.tick();\n\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send turnComplete after tool call\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add assistant tool_calls to conversation history\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: null,\n      tool_calls: response.toolCalls.map((tc, i) => ({\n        id: tc.id ?? `call_gemini_${tc.name}_${i}`,\n        type: \"function\" as const,\n        function: {\n          name: tc.name,\n          arguments: tc.arguments,\n        },\n      })),\n    });\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: \"WS\",\n    path,\n    headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  ws.send(\n    JSON.stringify({\n      error: {\n        code: 13,\n        message: \"Fixture response did not match any known type\",\n        status: \"INTERNAL\",\n      },\n    }),\n  );\n}\n"],"mappings":";;;;;;;;AAmGA,MAAM,UAAU;;;;;AAMhB,SAAS,WAAW,UAA0B;AAC5C,SAAQ,UAAR;EACE,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,QACE,QAAO;;;;;;AAOb,SAAS,sBAAsB,OAAwC;CACrE,MAAM,WAA0B,EAAE;AAElC,MAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,OAAO,KAAK,QAAQ;AAE1B,MAAI,SAAS,QAAQ;GACnB,MAAM,gBAAgB,KAAK,MAAM,QAAQ,MAAM,EAAE,iBAAiB;GAGlE,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,SAAS,UAAa,CAAC,EAAE,QAAQ;AAE9E,OAAI,cAAc,SAAS,GAAG;AAC5B,SAAK,IAAI,IAAI,GAAG,IAAI,cAAc,QAAQ,KAAK;KAE7C,MAAM,KADO,cAAc,GACX;AAChB,cAAS,KAAK;MACZ,MAAM;MACN,SAAS,OAAO,GAAG,aAAa,WAAW,GAAG,WAAW,KAAK,UAAU,GAAG,SAAS;MACpF,cAAc,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;MAClD,CAAC;;AAEJ,QAAI,UAAU,SAAS,EACrB,UAAS,KAAK;KACZ,MAAM;KACN,SAAS,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;KAChD,CAAC;UAEC;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAQ,SAAS;KAAM,CAAC;;aAEvC,SAAS,SAAS;GAC3B,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,aAAa;GAC1D,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,SAAS,UAAa,CAAC,EAAE,QAAQ;AAE9E,OAAI,UAAU,SAAS,GAAG;IACxB,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KACZ,MAAM;KACN,SAAS,QAAQ;KACjB,YAAY,UAAU,KAAK,GAAG,OAAO;MACnC,IAAI,eAAe,EAAE,aAAc,KAAK,GAAG;MAC3C,MAAM;MACN,UAAU;OACR,MAAM,EAAE,aAAc;OACtB,WAAW,KAAK,UAAU,EAAE,aAAc,QAAQ,EAAE,CAAC;OACtD;MACF,EAAE;KACJ,CAAC;UACG;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAa,SAAS;KAAM,CAAC;;;;AAKzD,QAAO;;;;;AAMT,SAAS,uBAAuB,cAAqD;AACnF,QAAO,aAAa,kBAAkB,KAAK,IAAI,OAAO;EACpD,MAAM;EACN,SAAS,OAAO,GAAG,aAAa,WAAW,GAAG,WAAW,KAAK,UAAU,GAAG,SAAS;EACpF,cAAc,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;EAClD,EAAE;;;;;AAML,SAAS,aAAa,aAAqD;AACzE,KAAI,CAAC,eAAe,YAAY,WAAW,EAAG,QAAO,EAAE;AAEvD,QADc,YAAY,SAAS,MAAM,EAAE,wBAAwB,EAAE,CAAC,CACzD,KAAK,OAAO;EACvB,MAAM;EACN,UAAU;GACR,MAAM,EAAE;GACR,aAAa,EAAE;GACf,YAAY,EAAE;GACf;EACF,EAAE;;AAKL,SAAgB,0BACd,IACA,UACA,SACA,UAWM;CACN,MAAM,EAAE,WAAW;CACnB,MAAM,UAAwB;EAC5B,WAAW;EACX,OAAO,SAAS;EAChB,OAAO,EAAE;EACT,qBAAqB,EAAE;EACxB;CAED,IAAI,UAAU,QAAQ,SAAS;AAC/B,IAAG,GAAG,YAAY,QAAgB;AAChC,YAAU,QAAQ,WAChB,eAAe,KAAK,IAAI,UAAU,SAAS,UAAU,QAAQ,CAAC,OAAO,QAAiB;GACpF,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,UAAO,MAAM,gCAAgC,MAAM;AACnD,OAAI;AACF,OAAG,KACD,KAAK,UAAU,EACb,OAAO;KAAE,MAAM;KAAI,SAAS;KAAK,QAAQ;KAAY,EACtD,CAAC,CACH;YACM,SAAS;AAChB,aAAS,OAAO,MACd,mCAAmC,mBAAmB,QAAQ,QAAQ,UAAU,YACjF;;IAEH,CACH;GACD;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UAWA,SACe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;UACjB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS,mBAAmB;GAAU,QAAQ;GAAoB,EACrF,CAAC,CACH;AACD;;CAIF,MAAM,WAAW,OAAO,SAAS,OAAO;AACxC,KAAI,UAAU;AACZ,UAAQ,YAAY;AACpB,UAAQ,QAAQ,SAAS,SAAS,SAAS;AAC3C,UAAQ,QAAQ,aAAa,SAAS,MAAM;AAC5C,KAAG,KAAK,KAAK,UAAU,EAAE,eAAe,EAAE,EAAE,CAAC,CAAC;AAC9C;;AAIF,KAAI,CAAC,QAAQ,WAAW;AACtB,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS;GAAkB,QAAQ;GAAuB,EAC7E,CAAC,CACH;AACD;;CAIF,IAAI;AAEJ,KAAI,OAAO,eAAe;AACxB,MAAI,CAAC,OAAO,cAAc,SAAS,CAAC,MAAM,QAAQ,OAAO,cAAc,MAAM,EAAE;AAC7E,MAAG,KACD,KAAK,UAAU,EACb,OAAO;IACL,MAAM;IACN,SAAS;IACT,QAAQ;IACT,EACF,CAAC,CACH;AACD;;AAEF,gBAAc,sBAAsB,OAAO,cAAc,MAAM;YACtD,OAAO,cAAc;AAC9B,MACE,CAAC,OAAO,aAAa,qBACrB,CAAC,MAAM,QAAQ,OAAO,aAAa,kBAAkB,EACrD;AACA,MAAG,KACD,KAAK,UAAU,EACb,OAAO;IACL,MAAM;IACN,SAAS;IACT,QAAQ;IACT,EACF,CAAC,CACH;AACD;;AAEF,gBAAc,uBAAuB,OAAO,aAAa;QACpD;AACL,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GACL,MAAM;GACN,SAAS;GACT,QAAQ;GACT,EACF,CAAC,CACH;AACD;;CAIF,MAAM,gBAAuC;EAC3C,OAAO,QAAQ;EACf,UAAU,CAAC,GAAG,QAAQ,qBAAqB,GAAG,YAAY;EAC1D,QAAQ;EACR,OAAO,QAAQ,MAAM,SAAS,IAAI,QAAQ,QAAQ;EAClD,eAAe;EAChB;CAED,MAAM,SAAS,SAAS,UAAUA;CAClC,MAAM,UAAUC,4BACd,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;CACD,MAAM,OAAO;AAEb,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KAAI,CAAC,SAAS;AACZ,MAAIC,kCAAkB,SAAS,QAAQ,SAAS,eAAe,EAAE;AAC/D,YAAS,OAAO,KAAK,mDAAmD;AACxE,WAAQ,IAAI;IACV,QAAQ;IACR;IACA,SAASC,+BAAe,SAAS,kBAAkB,EAAE,CAAC;IACtD,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAGC,oCAAoB,SAAS,QAAQ,SAAS,eAAe;KACjE;IACF,CAAC;AACF,MAAG,MAAM,MAAM,kCAAkC;AACjD;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAASD,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAGC,oCAAoB,SAAS,QAAQ,SAAS,eAAe;IACjE;GACF,CAAC;AACF,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS;GAAsB,QAAQ;GAAa,EACvE,CAAC,CACH;AACD;;AAIF,SAAQ,oBAAoB,KAAK,GAAG,YAAY;CAEhD,MAAM,WAAW,MAAMC,gCAAgB,SAAS,cAAc;CAC9D,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;AAGtE,KAAIC,gCAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAASH,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GACL,MAAM,WAAW,OAAO;GACxB,SAAS,SAAS,MAAM;GACxB,QAAQ,SAAS,MAAM,QAAQ;GAChC,EACF,CAAC,CACH;AACD;;AAIF,KAAII,gCAAgB,SAAS,EAAE;AAC7B,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAASJ,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,YAAY;EAClB,IAAI;EACJ,IAAI;AAEJ,MAAI,OAAO,UAAU,UAAU,UAAU;AACvC,cAAWK,6BAAa,UAAU,UAAU,MAAM;AAClD,UAAO,UAAU;SACZ;AACL,cAAW,UAAU,MAAM,eAAe;AAC1C,UAAO,UAAU,MAAM;;AAGzB,KAAG,KACD,KAAK,UAAU,EACb,eAAe;GACb,WAAW,EACT,OAAO,CAAC,EAAE,YAAY;IAAE;IAAU;IAAM,EAAE,CAAC,EAC5C;GACD,cAAc;GACf,EACF,CAAC,CACH;AAED,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS;GACV,CAAC;AACF;;AAIF,KAAIC,+CAA+B,SAAS,EAAE;EAC5C,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAASN,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,UAAU,SAAS;EACzB,MAAM,YAAsB,EAAE;AAC9B,OAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,UACvC,WAAU,KAAK,QAAQ,MAAM,GAAG,IAAI,UAAU,CAAC;EAGjD,MAAM,eAAeO,8CAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;EAC5B,IAAI,cAAc;AAGlB,MAAI,QAAQ,WAAW,GACrB;OAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,EAAE,EACrC,EACF,CAAC,CACH;QAGH,MAAK,IAAI,IAAI,GAAG,IAAI,UAAU,QAAQ,KAAK;AACzC,OAAI,GAAG,SAAU;GACjB,MAAM,aAAaC,kCAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACtF,OAAI,aAAa,EAAG,OAAMC,yBAAM,YAAY,cAAc,OAAO;AACjE,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;AAEF,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,UAAU,IAAI,CAAC,EAAE,EAC/C,EACF,CAAC,CACH;AACD,iBAAc,MAAM;AACpB,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;;AAKN,MAAI,aAAa;AACf,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;EAIF,MAAM,oBAAoB,SAAS,UAAU,KAAK,QAAQ;GACxD,GAAG;GACH,YAAY,GAAG,MAAMC,oCAAoB;GAC1C,EAAE;AAGH,MAAI,CAAC,GAAG,UAAU;GAChB,MAAM,UAAUF,kCACd,UAAU,QACV,QACA,SACA,iBACA,YACD;AACD,OAAI,UAAU,EAAG,OAAMC,yBAAM,SAAS,cAAc,OAAO;AAC3D,OAAI,cAAc,OAAO,SAAS;AAChC,OAAG,SAAS;AACZ,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,kBAAc,SAAS;AACvB;;GAGF,MAAM,gBAAgB,kBAAkB,KAAK,OAAO;IAClD,IAAI;AACJ,QAAI;AACF,eAAU,KAAK,MAAM,GAAG,aAAa,KAAK;YACpC;AACN,cAAS,OAAO,KACd,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,eAAU,EAAE;;AAEd,WAAO;KACL,MAAM,GAAG;KACT,MAAM;KACN,IAAI,GAAG;KACR;KACD;AAEF,MAAG,KAAK,KAAK,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;AACxD,iBAAc,MAAM;;AAGtB,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS,WAAW;GACpB,YAAY,kBAAkB,KAAK,QAAQ;IACzC,IAAI,GAAG;IACP,MAAM;IACN,UAAU;KACR,MAAM,GAAG;KACT,WAAW,GAAG;KACf;IACF,EAAE;GACJ,CAAC;AACF;;AAIF,KAAIE,+BAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAASX,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,UAAU,SAAS;AAEzB,MAAI,QAAQ,WAAW,GAAG;AACxB,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,EAAE,EACrC,EACF,CAAC,CACH;AACD,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AACD;;EAIF,MAAM,SAAmB,EAAE;AAC3B,OAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,UACvC,QAAO,KAAK,QAAQ,MAAM,GAAG,IAAI,UAAU,CAAC;EAG9C,MAAM,eAAeO,8CAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;EAC5B,IAAI,cAAc;AAGlB,OAAK,IAAI,IAAI,GAAG,IAAI,OAAO,QAAQ,KAAK;AACtC,OAAI,GAAG,SAAU;GACjB,MAAM,aAAaC,kCAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACtF,OAAI,aAAa,EAAG,OAAMC,yBAAM,YAAY,cAAc,OAAO;AACjE,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;AAEF,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,OAAO,IAAI,CAAC,EAAE,EAC5C,EACF,CAAC,CACH;AACD,iBAAc,MAAM;AACpB,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;;AAIJ,MAAI,aAAa;AACf,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAAE,MAAM;GAAa;GAAS,CAAC;AAChE;;AAIF,KAAIG,mCAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAASZ,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,eAAeO,8CAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;AAE5B,MAAI,GAAG,UAAU;AACf,iBAAc,SAAS;AACvB;;EAEF,MAAM,UAAUC,kCAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACnF,MAAI,UAAU,EAAG,OAAMC,yBAAM,SAAS,cAAc,OAAO;AAC3D,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAEF,MAAI,GAAG,UAAU;AACf,iBAAc,SAAS;AACvB;;EAGF,MAAM,gBAAgB,SAAS,UAAU,KAAK,IAAI,MAAM;GACtD,IAAI;AACJ,OAAI;AACF,cAAU,KAAK,MAAM,GAAG,aAAa,KAAK;WACpC;AACN,aAAS,OAAO,KACd,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,cAAU,EAAE;;AAEd,UAAO;IACL,MAAM,GAAG;IACT,MAAM;IACN,IAAI,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;IACxC;IACD;AAEF,KAAG,KAAK,KAAK,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;AACxD,gBAAc,MAAM;AAEpB,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS;GACT,YAAY,SAAS,UAAU,KAAK,IAAI,OAAO;IAC7C,IAAI,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;IACvC,MAAM;IACN,UAAU;KACR,MAAM,GAAG;KACT,WAAW,GAAG;KACf;IACF,EAAE;GACJ,CAAC;AACF;;AAIF,SAAQ,IAAI;EACV,QAAQ;EACR;EACA,SAAST,+BAAe,SAAS,kBAAkB,EAAE,CAAC;EACtD,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,IAAG,KACD,KAAK,UAAU,EACb,OAAO;EACL,MAAM;EACN,SAAS;EACT,QAAQ;EACT,EACF,CAAC,CACH"}
+{"version":3,"file":"ws-gemini-live.cjs","names":["DEFAULT_TEST_ID","matchFixture","resolveStrictMode","flattenHeaders","strictOverrideField","resolveResponse","isErrorResponse","isAudioResponse","formatToMime","isContentWithToolCallsResponse","createInterruptionSignal","calculateDelay","delay","generateToolCallId","isTextResponse","isToolCallResponse"],"sources":["../src/ws-gemini-live.ts"],"sourcesContent":["/**\n * WebSocket handler for Gemini Live BidiGenerateContent API.\n *\n * Accepts setup, clientContent, and toolResponse messages over WebSocket\n * and responds with setupComplete, serverContent, toolCall, and error\n * messages in the Gemini Live streaming format.\n */\n\nimport type {\n  Fixture,\n  ChatMessage,\n  ChatCompletionRequest,\n  ToolDefinition,\n  AudioResponse,\n} from \"./types.js\";\nimport { matchFixture } from \"./router.js\";\nimport {\n  isTextResponse,\n  isToolCallResponse,\n  isContentWithToolCallsResponse,\n  isErrorResponse,\n  isAudioResponse,\n  flattenHeaders,\n  formatToMime,\n  generateToolCallId,\n  resolveResponse,\n  resolveStrictMode,\n  strictOverrideField,\n} from \"./helpers.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\nimport { DEFAULT_TEST_ID, type Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport type { WebSocketConnection } from \"./ws-framing.js\";\n\n// ─── Gemini Live protocol types ─────────────────────────────────────────────\n\ninterface GeminiLivePart {\n  text?: string;\n  thought?: boolean;\n  functionCall?: { name: string; args: Record<string, unknown> };\n  functionResponse?: { name: string; response: unknown; id?: string };\n  inlineData?: { mimeType: string; data: string };\n}\n\ninterface GeminiLiveTurn {\n  role: string;\n  parts: GeminiLivePart[];\n}\n\ninterface GeminiLiveFunctionDeclaration {\n  name: string;\n  description?: string;\n  parameters?: object;\n}\n\ninterface GeminiLiveToolDef {\n  functionDeclarations?: GeminiLiveFunctionDeclaration[];\n}\n\ninterface GeminiLiveSetup {\n  model?: string;\n  generationConfig?: Record<string, unknown>;\n  tools?: GeminiLiveToolDef[];\n}\n\ninterface GeminiLiveClientContent {\n  turns: GeminiLiveTurn[];\n  turnComplete?: boolean;\n}\n\ninterface GeminiLiveFunctionResponse {\n  id?: string;\n  name: string;\n  response: unknown;\n}\n\ninterface GeminiLiveToolResponse {\n  functionResponses: GeminiLiveFunctionResponse[];\n}\n\ninterface GeminiLiveMessage {\n  setup?: GeminiLiveSetup;\n  config?: GeminiLiveSetup;\n  clientContent?: GeminiLiveClientContent;\n  toolResponse?: GeminiLiveToolResponse;\n}\n\n// ─── Session state ──────────────────────────────────────────────────────────\n\ninterface SessionState {\n  setupDone: boolean;\n  model: string;\n  tools: ToolDefinition[];\n  conversationHistory: ChatMessage[];\n}\n\n// ─── Helpers ────────────────────────────────────────────────────────────────\n\nconst WS_PATH = \"/ws/google.ai.generativelanguage.v1beta.GenerativeService.BidiGenerateContent\";\n\n/**\n * Map HTTP status codes to gRPC error codes.\n * Gemini Live uses gRPC codes, not HTTP status codes.\n */\nfunction httpToGrpc(httpCode: number): number {\n  switch (httpCode) {\n    case 400:\n      return 3; // INVALID_ARGUMENT\n    case 401:\n      return 16; // UNAUTHENTICATED\n    case 403:\n      return 7; // PERMISSION_DENIED\n    case 404:\n      return 5; // NOT_FOUND\n    case 409:\n      return 10; // ABORTED\n    case 429:\n      return 8; // RESOURCE_EXHAUSTED\n    case 501:\n      return 12; // UNIMPLEMENTED\n    case 503:\n      return 14; // UNAVAILABLE\n    default:\n      return 13; // INTERNAL\n  }\n}\n\n/**\n * Convert Gemini Live turns into ChatMessage[] for fixture matching.\n */\nfunction geminiTurnsToMessages(turns: GeminiLiveTurn[]): ChatMessage[] {\n  const messages: ChatMessage[] = [];\n\n  for (const turn of turns) {\n    const role = turn.role ?? \"user\";\n\n    if (role === \"user\") {\n      const funcResponses = turn.parts.filter((p) => p.functionResponse);\n      // inlineData parts (e.g. client audio input) are silently skipped —\n      // only text and functionResponse parts are relevant for fixture matching.\n      const textParts = turn.parts.filter((p) => p.text !== undefined && !p.thought);\n\n      if (funcResponses.length > 0) {\n        for (let i = 0; i < funcResponses.length; i++) {\n          const part = funcResponses[i];\n          const fr = part.functionResponse!;\n          messages.push({\n            role: \"tool\",\n            content: typeof fr.response === \"string\" ? fr.response : JSON.stringify(fr.response),\n            tool_call_id: fr.id ?? `call_gemini_${fr.name}_${i}`,\n          });\n        }\n        if (textParts.length > 0) {\n          messages.push({\n            role: \"user\",\n            content: textParts.map((p) => p.text!).join(\"\"),\n          });\n        }\n      } else {\n        const text = textParts.map((p) => p.text!).join(\"\");\n        messages.push({ role: \"user\", content: text });\n      }\n    } else if (role === \"model\") {\n      const funcCalls = turn.parts.filter((p) => p.functionCall);\n      const textParts = turn.parts.filter((p) => p.text !== undefined && !p.thought);\n\n      if (funcCalls.length > 0) {\n        const text = textParts.map((p) => p.text!).join(\"\");\n        messages.push({\n          role: \"assistant\",\n          content: text || 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  return messages;\n}\n\n/**\n * Convert toolResponse messages into ChatMessage[] for fixture matching.\n */\nfunction toolResponseToMessages(toolResponse: GeminiLiveToolResponse): ChatMessage[] {\n  return toolResponse.functionResponses.map((fr, i) => ({\n    role: \"tool\" as const,\n    content: typeof fr.response === \"string\" ? fr.response : JSON.stringify(fr.response),\n    tool_call_id: fr.id ?? `call_gemini_${fr.name}_${i}`,\n  }));\n}\n\n/**\n * Convert Gemini tool definitions to ChatCompletion ToolDefinition[].\n */\nfunction convertTools(geminiTools?: GeminiLiveToolDef[]): ToolDefinition[] {\n  if (!geminiTools || geminiTools.length === 0) return [];\n  const decls = geminiTools.flatMap((t) => t.functionDeclarations ?? []);\n  return 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// ─── Main handler ───────────────────────────────────────────────────────────\n\nexport function handleWebSocketGeminiLive(\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): void {\n  const { logger } = defaults;\n  const session: SessionState = {\n    setupDone: false,\n    model: defaults.model,\n    tools: [],\n    conversationHistory: [],\n  };\n\n  let pending = Promise.resolve();\n  ws.on(\"message\", (raw: string) => {\n    pending = pending.then(() =>\n      processMessage(raw, ws, fixtures, journal, defaults, session).catch((err: unknown) => {\n        const msg = err instanceof Error ? err.message : \"Internal error\";\n        logger.error(`WebSocket Gemini Live error: ${msg}`);\n        try {\n          ws.send(\n            JSON.stringify({\n              error: { code: 13, message: msg, status: \"INTERNAL\" },\n            }),\n          );\n        } catch (sendErr) {\n          defaults.logger.debug(\n            `Failed to send error to client: ${sendErr instanceof Error ? sendErr.message : \"unknown\"}`,\n          );\n        }\n      }),\n    );\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n  session: SessionState,\n): Promise<void> {\n  let parsed: GeminiLiveMessage;\n  try {\n    parsed = JSON.parse(raw) as GeminiLiveMessage;\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    ws.send(\n      JSON.stringify({\n        error: { code: 3, message: `Malformed JSON: ${detail}`, status: \"INVALID_ARGUMENT\" },\n      }),\n    );\n    return;\n  }\n\n  // Handle setup message (accept both `setup` and `config` as aliases)\n  const setupMsg = parsed.setup ?? parsed.config;\n  if (setupMsg) {\n    session.setupDone = true;\n    session.model = setupMsg.model ?? defaults.model;\n    session.tools = convertTools(setupMsg.tools);\n    ws.send(JSON.stringify({ setupComplete: {} }));\n    return;\n  }\n\n  // Reject messages before setup\n  if (!session.setupDone) {\n    ws.send(\n      JSON.stringify({\n        error: { code: 9, message: \"Setup required\", status: \"FAILED_PRECONDITION\" },\n      }),\n    );\n    return;\n  }\n\n  // Build messages from this interaction\n  let newMessages: ChatMessage[];\n\n  if (parsed.clientContent) {\n    if (!parsed.clientContent.turns || !Array.isArray(parsed.clientContent.turns)) {\n      ws.send(\n        JSON.stringify({\n          error: {\n            code: 3,\n            message: \"Missing 'turns' in clientContent\",\n            status: \"INVALID_ARGUMENT\",\n          },\n        }),\n      );\n      return;\n    }\n    newMessages = geminiTurnsToMessages(parsed.clientContent.turns);\n  } else if (parsed.toolResponse) {\n    if (\n      !parsed.toolResponse.functionResponses ||\n      !Array.isArray(parsed.toolResponse.functionResponses)\n    ) {\n      ws.send(\n        JSON.stringify({\n          error: {\n            code: 3,\n            message: \"Missing 'functionResponses' in toolResponse\",\n            status: \"INVALID_ARGUMENT\",\n          },\n        }),\n      );\n      return;\n    }\n    newMessages = toolResponseToMessages(parsed.toolResponse);\n  } else {\n    ws.send(\n      JSON.stringify({\n        error: {\n          code: 3,\n          message: \"Expected clientContent or toolResponse\",\n          status: \"INVALID_ARGUMENT\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Build completion request for fixture matching (include new messages speculatively)\n  const geminiContextHeader = defaults.upgradeHeaders?.[\"x-aimock-context\"];\n  const geminiContext =\n    typeof geminiContextHeader === \"string\"\n      ? geminiContextHeader\n      : Array.isArray(geminiContextHeader) && geminiContextHeader.length > 0\n        ? geminiContextHeader[0]\n        : undefined;\n\n  const completionReq: ChatCompletionRequest = {\n    model: session.model,\n    messages: [...session.conversationHistory, ...newMessages],\n    stream: true,\n    tools: session.tools.length > 0 ? session.tools : undefined,\n    _endpointType: \"chat\",\n    _context: geminiContext,\n  };\n\n  const testId = defaults.testId ?? DEFAULT_TEST_ID;\n  const fixture = matchFixture(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n  const path = WS_PATH;\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (!fixture) {\n    if (resolveStrictMode(defaults.strict, defaults.upgradeHeaders)) {\n      defaults.logger.warn(`STRICT: No fixture matched for WebSocket message`);\n      journal.add({\n        method: \"WS\",\n        path,\n        headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n        body: completionReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n        },\n      });\n      ws.close(1008, \"Strict mode: no fixture matched\");\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n      },\n    });\n    ws.send(\n      JSON.stringify({\n        error: { code: 5, message: \"No fixture matched\", status: \"NOT_FOUND\" },\n      }),\n    );\n    return;\n  }\n\n  // Commit messages to conversation history only after successful fixture match\n  session.conversationHistory.push(...newMessages);\n\n  const response = await resolveResponse(fixture, completionReq);\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: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    ws.send(\n      JSON.stringify({\n        error: {\n          code: httpToGrpc(status),\n          message: response.error.message,\n          status: response.error.type ?? \"INTERNAL\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Audio response — single frame with inlineData and turnComplete: true\n  if (isAudioResponse(response)) {\n    journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const audioResp = response as AudioResponse;\n    let mimeType: string;\n    let data: string;\n\n    if (typeof audioResp.audio === \"string\") {\n      mimeType = formatToMime(audioResp.format ?? \"mp3\");\n      data = audioResp.audio;\n    } else {\n      mimeType = audioResp.audio.contentType ?? \"audio/mpeg\";\n      data = audioResp.audio.b64Json;\n    }\n\n    ws.send(\n      JSON.stringify({\n        serverContent: {\n          modelTurn: {\n            parts: [{ inlineData: { mimeType, data } }],\n          },\n          turnComplete: true,\n        },\n      }),\n    );\n\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: \"[audio]\",\n    });\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: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const content = response.content;\n    const chunkList: string[] = [];\n    for (let i = 0; i < content.length; i += chunkSize) {\n      chunkList.push(content.slice(i, i + chunkSize));\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n    let interrupted = false;\n\n    // Stream text content chunks (turnComplete omitted — sent as a separate message later)\n    if (content.length === 0) {\n      if (!ws.isClosed) {\n        ws.send(\n          JSON.stringify({\n            serverContent: {\n              modelTurn: { parts: [{ text: \"\" }] },\n            },\n          }),\n        );\n      }\n    } else {\n      for (let i = 0; i < chunkList.length; i++) {\n        if (ws.isClosed) break;\n        const chunkDelay = calculateDelay(i, undefined, latency, recordedTimings, replaySpeed);\n        if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);\n        if (interruption?.signal.aborted) {\n          interrupted = true;\n          break;\n        }\n        if (ws.isClosed) break;\n\n        ws.send(\n          JSON.stringify({\n            serverContent: {\n              modelTurn: { parts: [{ text: chunkList[i] }] },\n            },\n          }),\n        );\n        interruption?.tick();\n        if (interruption?.signal.aborted) {\n          interrupted = true;\n          break;\n        }\n      }\n    }\n\n    if (interrupted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    // Pre-compute tool calls with stable IDs so wire message and history match\n    const resolvedToolCalls = response.toolCalls.map((tc) => ({\n      ...tc,\n      resolvedId: tc.id ?? generateToolCallId(),\n    }));\n\n    // Send tool calls\n    if (!ws.isClosed) {\n      const tcDelay = calculateDelay(\n        chunkList.length,\n        undefined,\n        latency,\n        recordedTimings,\n        replaySpeed,\n      );\n      if (tcDelay > 0) await delay(tcDelay, interruption?.signal);\n      if (interruption?.signal.aborted) {\n        ws.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n        interruption?.cleanup();\n        return;\n      }\n\n      const functionCalls = resolvedToolCalls.map((tc) => {\n        let argsObj: Record<string, unknown>;\n        try {\n          argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n        } catch {\n          defaults.logger.warn(\n            `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n          );\n          argsObj = {};\n        }\n        return {\n          name: tc.name,\n          args: argsObj,\n          id: tc.resolvedId,\n        };\n      });\n\n      ws.send(JSON.stringify({ toolCall: { functionCalls } }));\n      interruption?.tick();\n    }\n\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send turnComplete\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add to conversation history using the same resolved IDs from the wire message\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: content || null,\n      tool_calls: resolvedToolCalls.map((tc) => ({\n        id: tc.resolvedId,\n        type: \"function\" as const,\n        function: {\n          name: tc.name,\n          arguments: tc.arguments,\n        },\n      })),\n    });\n    return;\n  }\n\n  // Text response — stream chunks with serverContent\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const content = response.content;\n\n    if (content.length === 0) {\n      if (ws.isClosed) return;\n      // Empty content: send empty modelTurn, then separate turnComplete\n      ws.send(\n        JSON.stringify({\n          serverContent: {\n            modelTurn: { parts: [{ text: \"\" }] },\n          },\n        }),\n      );\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n      return;\n    }\n\n    // Chunk the content\n    const chunks: string[] = [];\n    for (let i = 0; i < content.length; i += chunkSize) {\n      chunks.push(content.slice(i, i + chunkSize));\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n    let interrupted = false;\n\n    // Stream content chunks without turnComplete (sent separately after)\n    for (let i = 0; i < chunks.length; i++) {\n      if (ws.isClosed) break;\n      const chunkDelay = calculateDelay(i, undefined, latency, recordedTimings, replaySpeed);\n      if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);\n      if (interruption?.signal.aborted) {\n        interrupted = true;\n        break;\n      }\n      if (ws.isClosed) break;\n\n      ws.send(\n        JSON.stringify({\n          serverContent: {\n            modelTurn: { parts: [{ text: chunks[i] }] },\n          },\n        }),\n      );\n      interruption?.tick();\n      if (interruption?.signal.aborted) {\n        interrupted = true;\n        break;\n      }\n    }\n\n    if (interrupted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send separate turnComplete message\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add assistant response to conversation history\n    session.conversationHistory.push({ role: \"assistant\", content });\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n\n    if (ws.isClosed) {\n      interruption?.cleanup();\n      return;\n    }\n    const tcDelay = calculateDelay(0, undefined, latency, recordedTimings, replaySpeed);\n    if (tcDelay > 0) await delay(tcDelay, interruption?.signal);\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n    if (ws.isClosed) {\n      interruption?.cleanup();\n      return;\n    }\n\n    const functionCalls = response.toolCalls.map((tc, i) => {\n      let argsObj: Record<string, unknown>;\n      try {\n        argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n      } catch {\n        defaults.logger.warn(\n          `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n        );\n        argsObj = {};\n      }\n      return {\n        name: tc.name,\n        args: argsObj,\n        id: tc.id ?? `call_gemini_${tc.name}_${i}`,\n      };\n    });\n\n    ws.send(JSON.stringify({ toolCall: { functionCalls } }));\n    interruption?.tick();\n\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send turnComplete after tool call\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add assistant tool_calls to conversation history\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: null,\n      tool_calls: response.toolCalls.map((tc, i) => ({\n        id: tc.id ?? `call_gemini_${tc.name}_${i}`,\n        type: \"function\" as const,\n        function: {\n          name: tc.name,\n          arguments: tc.arguments,\n        },\n      })),\n    });\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: \"WS\",\n    path,\n    headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  ws.send(\n    JSON.stringify({\n      error: {\n        code: 13,\n        message: \"Fixture response did not match any known type\",\n        status: \"INTERNAL\",\n      },\n    }),\n  );\n}\n"],"mappings":";;;;;;;;AAmGA,MAAM,UAAU;;;;;AAMhB,SAAS,WAAW,UAA0B;AAC5C,SAAQ,UAAR;EACE,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,QACE,QAAO;;;;;;AAOb,SAAS,sBAAsB,OAAwC;CACrE,MAAM,WAA0B,EAAE;AAElC,MAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,OAAO,KAAK,QAAQ;AAE1B,MAAI,SAAS,QAAQ;GACnB,MAAM,gBAAgB,KAAK,MAAM,QAAQ,MAAM,EAAE,iBAAiB;GAGlE,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,SAAS,UAAa,CAAC,EAAE,QAAQ;AAE9E,OAAI,cAAc,SAAS,GAAG;AAC5B,SAAK,IAAI,IAAI,GAAG,IAAI,cAAc,QAAQ,KAAK;KAE7C,MAAM,KADO,cAAc,GACX;AAChB,cAAS,KAAK;MACZ,MAAM;MACN,SAAS,OAAO,GAAG,aAAa,WAAW,GAAG,WAAW,KAAK,UAAU,GAAG,SAAS;MACpF,cAAc,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;MAClD,CAAC;;AAEJ,QAAI,UAAU,SAAS,EACrB,UAAS,KAAK;KACZ,MAAM;KACN,SAAS,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;KAChD,CAAC;UAEC;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAQ,SAAS;KAAM,CAAC;;aAEvC,SAAS,SAAS;GAC3B,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,aAAa;GAC1D,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,SAAS,UAAa,CAAC,EAAE,QAAQ;AAE9E,OAAI,UAAU,SAAS,GAAG;IACxB,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KACZ,MAAM;KACN,SAAS,QAAQ;KACjB,YAAY,UAAU,KAAK,GAAG,OAAO;MACnC,IAAI,eAAe,EAAE,aAAc,KAAK,GAAG;MAC3C,MAAM;MACN,UAAU;OACR,MAAM,EAAE,aAAc;OACtB,WAAW,KAAK,UAAU,EAAE,aAAc,QAAQ,EAAE,CAAC;OACtD;MACF,EAAE;KACJ,CAAC;UACG;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAa,SAAS;KAAM,CAAC;;;;AAKzD,QAAO;;;;;AAMT,SAAS,uBAAuB,cAAqD;AACnF,QAAO,aAAa,kBAAkB,KAAK,IAAI,OAAO;EACpD,MAAM;EACN,SAAS,OAAO,GAAG,aAAa,WAAW,GAAG,WAAW,KAAK,UAAU,GAAG,SAAS;EACpF,cAAc,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;EAClD,EAAE;;;;;AAML,SAAS,aAAa,aAAqD;AACzE,KAAI,CAAC,eAAe,YAAY,WAAW,EAAG,QAAO,EAAE;AAEvD,QADc,YAAY,SAAS,MAAM,EAAE,wBAAwB,EAAE,CAAC,CACzD,KAAK,OAAO;EACvB,MAAM;EACN,UAAU;GACR,MAAM,EAAE;GACR,aAAa,EAAE;GACf,YAAY,EAAE;GACf;EACF,EAAE;;AAKL,SAAgB,0BACd,IACA,UACA,SACA,UAWM;CACN,MAAM,EAAE,WAAW;CACnB,MAAM,UAAwB;EAC5B,WAAW;EACX,OAAO,SAAS;EAChB,OAAO,EAAE;EACT,qBAAqB,EAAE;EACxB;CAED,IAAI,UAAU,QAAQ,SAAS;AAC/B,IAAG,GAAG,YAAY,QAAgB;AAChC,YAAU,QAAQ,WAChB,eAAe,KAAK,IAAI,UAAU,SAAS,UAAU,QAAQ,CAAC,OAAO,QAAiB;GACpF,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,UAAO,MAAM,gCAAgC,MAAM;AACnD,OAAI;AACF,OAAG,KACD,KAAK,UAAU,EACb,OAAO;KAAE,MAAM;KAAI,SAAS;KAAK,QAAQ;KAAY,EACtD,CAAC,CACH;YACM,SAAS;AAChB,aAAS,OAAO,MACd,mCAAmC,mBAAmB,QAAQ,QAAQ,UAAU,YACjF;;IAEH,CACH;GACD;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UAWA,SACe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;UACjB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS,mBAAmB;GAAU,QAAQ;GAAoB,EACrF,CAAC,CACH;AACD;;CAIF,MAAM,WAAW,OAAO,SAAS,OAAO;AACxC,KAAI,UAAU;AACZ,UAAQ,YAAY;AACpB,UAAQ,QAAQ,SAAS,SAAS,SAAS;AAC3C,UAAQ,QAAQ,aAAa,SAAS,MAAM;AAC5C,KAAG,KAAK,KAAK,UAAU,EAAE,eAAe,EAAE,EAAE,CAAC,CAAC;AAC9C;;AAIF,KAAI,CAAC,QAAQ,WAAW;AACtB,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS;GAAkB,QAAQ;GAAuB,EAC7E,CAAC,CACH;AACD;;CAIF,IAAI;AAEJ,KAAI,OAAO,eAAe;AACxB,MAAI,CAAC,OAAO,cAAc,SAAS,CAAC,MAAM,QAAQ,OAAO,cAAc,MAAM,EAAE;AAC7E,MAAG,KACD,KAAK,UAAU,EACb,OAAO;IACL,MAAM;IACN,SAAS;IACT,QAAQ;IACT,EACF,CAAC,CACH;AACD;;AAEF,gBAAc,sBAAsB,OAAO,cAAc,MAAM;YACtD,OAAO,cAAc;AAC9B,MACE,CAAC,OAAO,aAAa,qBACrB,CAAC,MAAM,QAAQ,OAAO,aAAa,kBAAkB,EACrD;AACA,MAAG,KACD,KAAK,UAAU,EACb,OAAO;IACL,MAAM;IACN,SAAS;IACT,QAAQ;IACT,EACF,CAAC,CACH;AACD;;AAEF,gBAAc,uBAAuB,OAAO,aAAa;QACpD;AACL,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GACL,MAAM;GACN,SAAS;GACT,QAAQ;GACT,EACF,CAAC,CACH;AACD;;CAIF,MAAM,sBAAsB,SAAS,iBAAiB;CACtD,MAAM,gBACJ,OAAO,wBAAwB,WAC3B,sBACA,MAAM,QAAQ,oBAAoB,IAAI,oBAAoB,SAAS,IACjE,oBAAoB,KACpB;CAER,MAAM,gBAAuC;EAC3C,OAAO,QAAQ;EACf,UAAU,CAAC,GAAG,QAAQ,qBAAqB,GAAG,YAAY;EAC1D,QAAQ;EACR,OAAO,QAAQ,MAAM,SAAS,IAAI,QAAQ,QAAQ;EAClD,eAAe;EACf,UAAU;EACX;CAED,MAAM,SAAS,SAAS,UAAUA;CAClC,MAAM,UAAUC,4BACd,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;CACD,MAAM,OAAO;AAEb,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KAAI,CAAC,SAAS;AACZ,MAAIC,kCAAkB,SAAS,QAAQ,SAAS,eAAe,EAAE;AAC/D,YAAS,OAAO,KAAK,mDAAmD;AACxE,WAAQ,IAAI;IACV,QAAQ;IACR;IACA,SAASC,+BAAe,SAAS,kBAAkB,EAAE,CAAC;IACtD,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAGC,oCAAoB,SAAS,QAAQ,SAAS,eAAe;KACjE;IACF,CAAC;AACF,MAAG,MAAM,MAAM,kCAAkC;AACjD;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAASD,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAGC,oCAAoB,SAAS,QAAQ,SAAS,eAAe;IACjE;GACF,CAAC;AACF,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS;GAAsB,QAAQ;GAAa,EACvE,CAAC,CACH;AACD;;AAIF,SAAQ,oBAAoB,KAAK,GAAG,YAAY;CAEhD,MAAM,WAAW,MAAMC,gCAAgB,SAAS,cAAc;CAC9D,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;AAGtE,KAAIC,gCAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAASH,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GACL,MAAM,WAAW,OAAO;GACxB,SAAS,SAAS,MAAM;GACxB,QAAQ,SAAS,MAAM,QAAQ;GAChC,EACF,CAAC,CACH;AACD;;AAIF,KAAII,gCAAgB,SAAS,EAAE;AAC7B,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAASJ,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,YAAY;EAClB,IAAI;EACJ,IAAI;AAEJ,MAAI,OAAO,UAAU,UAAU,UAAU;AACvC,cAAWK,6BAAa,UAAU,UAAU,MAAM;AAClD,UAAO,UAAU;SACZ;AACL,cAAW,UAAU,MAAM,eAAe;AAC1C,UAAO,UAAU,MAAM;;AAGzB,KAAG,KACD,KAAK,UAAU,EACb,eAAe;GACb,WAAW,EACT,OAAO,CAAC,EAAE,YAAY;IAAE;IAAU;IAAM,EAAE,CAAC,EAC5C;GACD,cAAc;GACf,EACF,CAAC,CACH;AAED,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS;GACV,CAAC;AACF;;AAIF,KAAIC,+CAA+B,SAAS,EAAE;EAC5C,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAASN,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,UAAU,SAAS;EACzB,MAAM,YAAsB,EAAE;AAC9B,OAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,UACvC,WAAU,KAAK,QAAQ,MAAM,GAAG,IAAI,UAAU,CAAC;EAGjD,MAAM,eAAeO,8CAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;EAC5B,IAAI,cAAc;AAGlB,MAAI,QAAQ,WAAW,GACrB;OAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,EAAE,EACrC,EACF,CAAC,CACH;QAGH,MAAK,IAAI,IAAI,GAAG,IAAI,UAAU,QAAQ,KAAK;AACzC,OAAI,GAAG,SAAU;GACjB,MAAM,aAAaC,kCAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACtF,OAAI,aAAa,EAAG,OAAMC,yBAAM,YAAY,cAAc,OAAO;AACjE,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;AAEF,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,UAAU,IAAI,CAAC,EAAE,EAC/C,EACF,CAAC,CACH;AACD,iBAAc,MAAM;AACpB,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;;AAKN,MAAI,aAAa;AACf,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;EAIF,MAAM,oBAAoB,SAAS,UAAU,KAAK,QAAQ;GACxD,GAAG;GACH,YAAY,GAAG,MAAMC,oCAAoB;GAC1C,EAAE;AAGH,MAAI,CAAC,GAAG,UAAU;GAChB,MAAM,UAAUF,kCACd,UAAU,QACV,QACA,SACA,iBACA,YACD;AACD,OAAI,UAAU,EAAG,OAAMC,yBAAM,SAAS,cAAc,OAAO;AAC3D,OAAI,cAAc,OAAO,SAAS;AAChC,OAAG,SAAS;AACZ,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,kBAAc,SAAS;AACvB;;GAGF,MAAM,gBAAgB,kBAAkB,KAAK,OAAO;IAClD,IAAI;AACJ,QAAI;AACF,eAAU,KAAK,MAAM,GAAG,aAAa,KAAK;YACpC;AACN,cAAS,OAAO,KACd,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,eAAU,EAAE;;AAEd,WAAO;KACL,MAAM,GAAG;KACT,MAAM;KACN,IAAI,GAAG;KACR;KACD;AAEF,MAAG,KAAK,KAAK,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;AACxD,iBAAc,MAAM;;AAGtB,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS,WAAW;GACpB,YAAY,kBAAkB,KAAK,QAAQ;IACzC,IAAI,GAAG;IACP,MAAM;IACN,UAAU;KACR,MAAM,GAAG;KACT,WAAW,GAAG;KACf;IACF,EAAE;GACJ,CAAC;AACF;;AAIF,KAAIE,+BAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAASX,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,UAAU,SAAS;AAEzB,MAAI,QAAQ,WAAW,GAAG;AACxB,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,EAAE,EACrC,EACF,CAAC,CACH;AACD,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AACD;;EAIF,MAAM,SAAmB,EAAE;AAC3B,OAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,UACvC,QAAO,KAAK,QAAQ,MAAM,GAAG,IAAI,UAAU,CAAC;EAG9C,MAAM,eAAeO,8CAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;EAC5B,IAAI,cAAc;AAGlB,OAAK,IAAI,IAAI,GAAG,IAAI,OAAO,QAAQ,KAAK;AACtC,OAAI,GAAG,SAAU;GACjB,MAAM,aAAaC,kCAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACtF,OAAI,aAAa,EAAG,OAAMC,yBAAM,YAAY,cAAc,OAAO;AACjE,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;AAEF,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,OAAO,IAAI,CAAC,EAAE,EAC5C,EACF,CAAC,CACH;AACD,iBAAc,MAAM;AACpB,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;;AAIJ,MAAI,aAAa;AACf,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAAE,MAAM;GAAa;GAAS,CAAC;AAChE;;AAIF,KAAIG,mCAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAASZ,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,eAAeO,8CAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;AAE5B,MAAI,GAAG,UAAU;AACf,iBAAc,SAAS;AACvB;;EAEF,MAAM,UAAUC,kCAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACnF,MAAI,UAAU,EAAG,OAAMC,yBAAM,SAAS,cAAc,OAAO;AAC3D,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAEF,MAAI,GAAG,UAAU;AACf,iBAAc,SAAS;AACvB;;EAGF,MAAM,gBAAgB,SAAS,UAAU,KAAK,IAAI,MAAM;GACtD,IAAI;AACJ,OAAI;AACF,cAAU,KAAK,MAAM,GAAG,aAAa,KAAK;WACpC;AACN,aAAS,OAAO,KACd,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,cAAU,EAAE;;AAEd,UAAO;IACL,MAAM,GAAG;IACT,MAAM;IACN,IAAI,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;IACxC;IACD;AAEF,KAAG,KAAK,KAAK,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;AACxD,gBAAc,MAAM;AAEpB,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS;GACT,YAAY,SAAS,UAAU,KAAK,IAAI,OAAO;IAC7C,IAAI,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;IACvC,MAAM;IACN,UAAU;KACR,MAAM,GAAG;KACT,WAAW,GAAG;KACf;IACF,EAAE;GACJ,CAAC;AACF;;AAIF,SAAQ,IAAI;EACV,QAAQ;EACR;EACA,SAAST,+BAAe,SAAS,kBAAkB,EAAE,CAAC;EACtD,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,IAAG,KACD,KAAK,UAAU,EACb,OAAO;EACL,MAAM;EACN,SAAS;EACT,QAAQ;EACT,EACF,CAAC,CACH"}

package/dist/ws-gemini-live.js

@@ -189,12 +189,15 @@ async function processMessage(raw, ws, fixtures, journal, defaults, session) {
 		} }));
 		return;
 	}
+	const geminiContextHeader = defaults.upgradeHeaders?.["x-aimock-context"];
+	const geminiContext = typeof geminiContextHeader === "string" ? geminiContextHeader : Array.isArray(geminiContextHeader) && geminiContextHeader.length > 0 ? geminiContextHeader[0] : void 0;
 	const completionReq = {
 		model: session.model,
 		messages: [...session.conversationHistory, ...newMessages],
 		stream: true,
 		tools: session.tools.length > 0 ? session.tools : void 0,
-		_endpointType: "chat"
+		_endpointType: "chat",
+		_context: geminiContext
 	};
 	const testId = defaults.testId ?? DEFAULT_TEST_ID;
 	const fixture = matchFixture(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);

package/dist/ws-gemini-live.js.map

@@ -1 +1 @@
-{"version":3,"file":"ws-gemini-live.js","names":[],"sources":["../src/ws-gemini-live.ts"],"sourcesContent":["/**\n * WebSocket handler for Gemini Live BidiGenerateContent API.\n *\n * Accepts setup, clientContent, and toolResponse messages over WebSocket\n * and responds with setupComplete, serverContent, toolCall, and error\n * messages in the Gemini Live streaming format.\n */\n\nimport type {\n  Fixture,\n  ChatMessage,\n  ChatCompletionRequest,\n  ToolDefinition,\n  AudioResponse,\n} from \"./types.js\";\nimport { matchFixture } from \"./router.js\";\nimport {\n  isTextResponse,\n  isToolCallResponse,\n  isContentWithToolCallsResponse,\n  isErrorResponse,\n  isAudioResponse,\n  flattenHeaders,\n  formatToMime,\n  generateToolCallId,\n  resolveResponse,\n  resolveStrictMode,\n  strictOverrideField,\n} from \"./helpers.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\nimport { DEFAULT_TEST_ID, type Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport type { WebSocketConnection } from \"./ws-framing.js\";\n\n// ─── Gemini Live protocol types ─────────────────────────────────────────────\n\ninterface GeminiLivePart {\n  text?: string;\n  thought?: boolean;\n  functionCall?: { name: string; args: Record<string, unknown> };\n  functionResponse?: { name: string; response: unknown; id?: string };\n  inlineData?: { mimeType: string; data: string };\n}\n\ninterface GeminiLiveTurn {\n  role: string;\n  parts: GeminiLivePart[];\n}\n\ninterface GeminiLiveFunctionDeclaration {\n  name: string;\n  description?: string;\n  parameters?: object;\n}\n\ninterface GeminiLiveToolDef {\n  functionDeclarations?: GeminiLiveFunctionDeclaration[];\n}\n\ninterface GeminiLiveSetup {\n  model?: string;\n  generationConfig?: Record<string, unknown>;\n  tools?: GeminiLiveToolDef[];\n}\n\ninterface GeminiLiveClientContent {\n  turns: GeminiLiveTurn[];\n  turnComplete?: boolean;\n}\n\ninterface GeminiLiveFunctionResponse {\n  id?: string;\n  name: string;\n  response: unknown;\n}\n\ninterface GeminiLiveToolResponse {\n  functionResponses: GeminiLiveFunctionResponse[];\n}\n\ninterface GeminiLiveMessage {\n  setup?: GeminiLiveSetup;\n  config?: GeminiLiveSetup;\n  clientContent?: GeminiLiveClientContent;\n  toolResponse?: GeminiLiveToolResponse;\n}\n\n// ─── Session state ──────────────────────────────────────────────────────────\n\ninterface SessionState {\n  setupDone: boolean;\n  model: string;\n  tools: ToolDefinition[];\n  conversationHistory: ChatMessage[];\n}\n\n// ─── Helpers ────────────────────────────────────────────────────────────────\n\nconst WS_PATH = \"/ws/google.ai.generativelanguage.v1beta.GenerativeService.BidiGenerateContent\";\n\n/**\n * Map HTTP status codes to gRPC error codes.\n * Gemini Live uses gRPC codes, not HTTP status codes.\n */\nfunction httpToGrpc(httpCode: number): number {\n  switch (httpCode) {\n    case 400:\n      return 3; // INVALID_ARGUMENT\n    case 401:\n      return 16; // UNAUTHENTICATED\n    case 403:\n      return 7; // PERMISSION_DENIED\n    case 404:\n      return 5; // NOT_FOUND\n    case 409:\n      return 10; // ABORTED\n    case 429:\n      return 8; // RESOURCE_EXHAUSTED\n    case 501:\n      return 12; // UNIMPLEMENTED\n    case 503:\n      return 14; // UNAVAILABLE\n    default:\n      return 13; // INTERNAL\n  }\n}\n\n/**\n * Convert Gemini Live turns into ChatMessage[] for fixture matching.\n */\nfunction geminiTurnsToMessages(turns: GeminiLiveTurn[]): ChatMessage[] {\n  const messages: ChatMessage[] = [];\n\n  for (const turn of turns) {\n    const role = turn.role ?? \"user\";\n\n    if (role === \"user\") {\n      const funcResponses = turn.parts.filter((p) => p.functionResponse);\n      // inlineData parts (e.g. client audio input) are silently skipped —\n      // only text and functionResponse parts are relevant for fixture matching.\n      const textParts = turn.parts.filter((p) => p.text !== undefined && !p.thought);\n\n      if (funcResponses.length > 0) {\n        for (let i = 0; i < funcResponses.length; i++) {\n          const part = funcResponses[i];\n          const fr = part.functionResponse!;\n          messages.push({\n            role: \"tool\",\n            content: typeof fr.response === \"string\" ? fr.response : JSON.stringify(fr.response),\n            tool_call_id: fr.id ?? `call_gemini_${fr.name}_${i}`,\n          });\n        }\n        if (textParts.length > 0) {\n          messages.push({\n            role: \"user\",\n            content: textParts.map((p) => p.text!).join(\"\"),\n          });\n        }\n      } else {\n        const text = textParts.map((p) => p.text!).join(\"\");\n        messages.push({ role: \"user\", content: text });\n      }\n    } else if (role === \"model\") {\n      const funcCalls = turn.parts.filter((p) => p.functionCall);\n      const textParts = turn.parts.filter((p) => p.text !== undefined && !p.thought);\n\n      if (funcCalls.length > 0) {\n        const text = textParts.map((p) => p.text!).join(\"\");\n        messages.push({\n          role: \"assistant\",\n          content: text || 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  return messages;\n}\n\n/**\n * Convert toolResponse messages into ChatMessage[] for fixture matching.\n */\nfunction toolResponseToMessages(toolResponse: GeminiLiveToolResponse): ChatMessage[] {\n  return toolResponse.functionResponses.map((fr, i) => ({\n    role: \"tool\" as const,\n    content: typeof fr.response === \"string\" ? fr.response : JSON.stringify(fr.response),\n    tool_call_id: fr.id ?? `call_gemini_${fr.name}_${i}`,\n  }));\n}\n\n/**\n * Convert Gemini tool definitions to ChatCompletion ToolDefinition[].\n */\nfunction convertTools(geminiTools?: GeminiLiveToolDef[]): ToolDefinition[] {\n  if (!geminiTools || geminiTools.length === 0) return [];\n  const decls = geminiTools.flatMap((t) => t.functionDeclarations ?? []);\n  return 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// ─── Main handler ───────────────────────────────────────────────────────────\n\nexport function handleWebSocketGeminiLive(\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): void {\n  const { logger } = defaults;\n  const session: SessionState = {\n    setupDone: false,\n    model: defaults.model,\n    tools: [],\n    conversationHistory: [],\n  };\n\n  let pending = Promise.resolve();\n  ws.on(\"message\", (raw: string) => {\n    pending = pending.then(() =>\n      processMessage(raw, ws, fixtures, journal, defaults, session).catch((err: unknown) => {\n        const msg = err instanceof Error ? err.message : \"Internal error\";\n        logger.error(`WebSocket Gemini Live error: ${msg}`);\n        try {\n          ws.send(\n            JSON.stringify({\n              error: { code: 13, message: msg, status: \"INTERNAL\" },\n            }),\n          );\n        } catch (sendErr) {\n          defaults.logger.debug(\n            `Failed to send error to client: ${sendErr instanceof Error ? sendErr.message : \"unknown\"}`,\n          );\n        }\n      }),\n    );\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n  session: SessionState,\n): Promise<void> {\n  let parsed: GeminiLiveMessage;\n  try {\n    parsed = JSON.parse(raw) as GeminiLiveMessage;\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    ws.send(\n      JSON.stringify({\n        error: { code: 3, message: `Malformed JSON: ${detail}`, status: \"INVALID_ARGUMENT\" },\n      }),\n    );\n    return;\n  }\n\n  // Handle setup message (accept both `setup` and `config` as aliases)\n  const setupMsg = parsed.setup ?? parsed.config;\n  if (setupMsg) {\n    session.setupDone = true;\n    session.model = setupMsg.model ?? defaults.model;\n    session.tools = convertTools(setupMsg.tools);\n    ws.send(JSON.stringify({ setupComplete: {} }));\n    return;\n  }\n\n  // Reject messages before setup\n  if (!session.setupDone) {\n    ws.send(\n      JSON.stringify({\n        error: { code: 9, message: \"Setup required\", status: \"FAILED_PRECONDITION\" },\n      }),\n    );\n    return;\n  }\n\n  // Build messages from this interaction\n  let newMessages: ChatMessage[];\n\n  if (parsed.clientContent) {\n    if (!parsed.clientContent.turns || !Array.isArray(parsed.clientContent.turns)) {\n      ws.send(\n        JSON.stringify({\n          error: {\n            code: 3,\n            message: \"Missing 'turns' in clientContent\",\n            status: \"INVALID_ARGUMENT\",\n          },\n        }),\n      );\n      return;\n    }\n    newMessages = geminiTurnsToMessages(parsed.clientContent.turns);\n  } else if (parsed.toolResponse) {\n    if (\n      !parsed.toolResponse.functionResponses ||\n      !Array.isArray(parsed.toolResponse.functionResponses)\n    ) {\n      ws.send(\n        JSON.stringify({\n          error: {\n            code: 3,\n            message: \"Missing 'functionResponses' in toolResponse\",\n            status: \"INVALID_ARGUMENT\",\n          },\n        }),\n      );\n      return;\n    }\n    newMessages = toolResponseToMessages(parsed.toolResponse);\n  } else {\n    ws.send(\n      JSON.stringify({\n        error: {\n          code: 3,\n          message: \"Expected clientContent or toolResponse\",\n          status: \"INVALID_ARGUMENT\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Build completion request for fixture matching (include new messages speculatively)\n  const completionReq: ChatCompletionRequest = {\n    model: session.model,\n    messages: [...session.conversationHistory, ...newMessages],\n    stream: true,\n    tools: session.tools.length > 0 ? session.tools : undefined,\n    _endpointType: \"chat\",\n  };\n\n  const testId = defaults.testId ?? DEFAULT_TEST_ID;\n  const fixture = matchFixture(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n  const path = WS_PATH;\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (!fixture) {\n    if (resolveStrictMode(defaults.strict, defaults.upgradeHeaders)) {\n      defaults.logger.warn(`STRICT: No fixture matched for WebSocket message`);\n      journal.add({\n        method: \"WS\",\n        path,\n        headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n        body: completionReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n        },\n      });\n      ws.close(1008, \"Strict mode: no fixture matched\");\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n      },\n    });\n    ws.send(\n      JSON.stringify({\n        error: { code: 5, message: \"No fixture matched\", status: \"NOT_FOUND\" },\n      }),\n    );\n    return;\n  }\n\n  // Commit messages to conversation history only after successful fixture match\n  session.conversationHistory.push(...newMessages);\n\n  const response = await resolveResponse(fixture, completionReq);\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: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    ws.send(\n      JSON.stringify({\n        error: {\n          code: httpToGrpc(status),\n          message: response.error.message,\n          status: response.error.type ?? \"INTERNAL\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Audio response — single frame with inlineData and turnComplete: true\n  if (isAudioResponse(response)) {\n    journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const audioResp = response as AudioResponse;\n    let mimeType: string;\n    let data: string;\n\n    if (typeof audioResp.audio === \"string\") {\n      mimeType = formatToMime(audioResp.format ?? \"mp3\");\n      data = audioResp.audio;\n    } else {\n      mimeType = audioResp.audio.contentType ?? \"audio/mpeg\";\n      data = audioResp.audio.b64Json;\n    }\n\n    ws.send(\n      JSON.stringify({\n        serverContent: {\n          modelTurn: {\n            parts: [{ inlineData: { mimeType, data } }],\n          },\n          turnComplete: true,\n        },\n      }),\n    );\n\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: \"[audio]\",\n    });\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: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const content = response.content;\n    const chunkList: string[] = [];\n    for (let i = 0; i < content.length; i += chunkSize) {\n      chunkList.push(content.slice(i, i + chunkSize));\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n    let interrupted = false;\n\n    // Stream text content chunks (turnComplete omitted — sent as a separate message later)\n    if (content.length === 0) {\n      if (!ws.isClosed) {\n        ws.send(\n          JSON.stringify({\n            serverContent: {\n              modelTurn: { parts: [{ text: \"\" }] },\n            },\n          }),\n        );\n      }\n    } else {\n      for (let i = 0; i < chunkList.length; i++) {\n        if (ws.isClosed) break;\n        const chunkDelay = calculateDelay(i, undefined, latency, recordedTimings, replaySpeed);\n        if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);\n        if (interruption?.signal.aborted) {\n          interrupted = true;\n          break;\n        }\n        if (ws.isClosed) break;\n\n        ws.send(\n          JSON.stringify({\n            serverContent: {\n              modelTurn: { parts: [{ text: chunkList[i] }] },\n            },\n          }),\n        );\n        interruption?.tick();\n        if (interruption?.signal.aborted) {\n          interrupted = true;\n          break;\n        }\n      }\n    }\n\n    if (interrupted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    // Pre-compute tool calls with stable IDs so wire message and history match\n    const resolvedToolCalls = response.toolCalls.map((tc) => ({\n      ...tc,\n      resolvedId: tc.id ?? generateToolCallId(),\n    }));\n\n    // Send tool calls\n    if (!ws.isClosed) {\n      const tcDelay = calculateDelay(\n        chunkList.length,\n        undefined,\n        latency,\n        recordedTimings,\n        replaySpeed,\n      );\n      if (tcDelay > 0) await delay(tcDelay, interruption?.signal);\n      if (interruption?.signal.aborted) {\n        ws.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n        interruption?.cleanup();\n        return;\n      }\n\n      const functionCalls = resolvedToolCalls.map((tc) => {\n        let argsObj: Record<string, unknown>;\n        try {\n          argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n        } catch {\n          defaults.logger.warn(\n            `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n          );\n          argsObj = {};\n        }\n        return {\n          name: tc.name,\n          args: argsObj,\n          id: tc.resolvedId,\n        };\n      });\n\n      ws.send(JSON.stringify({ toolCall: { functionCalls } }));\n      interruption?.tick();\n    }\n\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send turnComplete\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add to conversation history using the same resolved IDs from the wire message\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: content || null,\n      tool_calls: resolvedToolCalls.map((tc) => ({\n        id: tc.resolvedId,\n        type: \"function\" as const,\n        function: {\n          name: tc.name,\n          arguments: tc.arguments,\n        },\n      })),\n    });\n    return;\n  }\n\n  // Text response — stream chunks with serverContent\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const content = response.content;\n\n    if (content.length === 0) {\n      if (ws.isClosed) return;\n      // Empty content: send empty modelTurn, then separate turnComplete\n      ws.send(\n        JSON.stringify({\n          serverContent: {\n            modelTurn: { parts: [{ text: \"\" }] },\n          },\n        }),\n      );\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n      return;\n    }\n\n    // Chunk the content\n    const chunks: string[] = [];\n    for (let i = 0; i < content.length; i += chunkSize) {\n      chunks.push(content.slice(i, i + chunkSize));\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n    let interrupted = false;\n\n    // Stream content chunks without turnComplete (sent separately after)\n    for (let i = 0; i < chunks.length; i++) {\n      if (ws.isClosed) break;\n      const chunkDelay = calculateDelay(i, undefined, latency, recordedTimings, replaySpeed);\n      if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);\n      if (interruption?.signal.aborted) {\n        interrupted = true;\n        break;\n      }\n      if (ws.isClosed) break;\n\n      ws.send(\n        JSON.stringify({\n          serverContent: {\n            modelTurn: { parts: [{ text: chunks[i] }] },\n          },\n        }),\n      );\n      interruption?.tick();\n      if (interruption?.signal.aborted) {\n        interrupted = true;\n        break;\n      }\n    }\n\n    if (interrupted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send separate turnComplete message\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add assistant response to conversation history\n    session.conversationHistory.push({ role: \"assistant\", content });\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n\n    if (ws.isClosed) {\n      interruption?.cleanup();\n      return;\n    }\n    const tcDelay = calculateDelay(0, undefined, latency, recordedTimings, replaySpeed);\n    if (tcDelay > 0) await delay(tcDelay, interruption?.signal);\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n    if (ws.isClosed) {\n      interruption?.cleanup();\n      return;\n    }\n\n    const functionCalls = response.toolCalls.map((tc, i) => {\n      let argsObj: Record<string, unknown>;\n      try {\n        argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n      } catch {\n        defaults.logger.warn(\n          `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n        );\n        argsObj = {};\n      }\n      return {\n        name: tc.name,\n        args: argsObj,\n        id: tc.id ?? `call_gemini_${tc.name}_${i}`,\n      };\n    });\n\n    ws.send(JSON.stringify({ toolCall: { functionCalls } }));\n    interruption?.tick();\n\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send turnComplete after tool call\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add assistant tool_calls to conversation history\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: null,\n      tool_calls: response.toolCalls.map((tc, i) => ({\n        id: tc.id ?? `call_gemini_${tc.name}_${i}`,\n        type: \"function\" as const,\n        function: {\n          name: tc.name,\n          arguments: tc.arguments,\n        },\n      })),\n    });\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: \"WS\",\n    path,\n    headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  ws.send(\n    JSON.stringify({\n      error: {\n        code: 13,\n        message: \"Fixture response did not match any known type\",\n        status: \"INTERNAL\",\n      },\n    }),\n  );\n}\n"],"mappings":";;;;;;;;AAmGA,MAAM,UAAU;;;;;AAMhB,SAAS,WAAW,UAA0B;AAC5C,SAAQ,UAAR;EACE,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,QACE,QAAO;;;;;;AAOb,SAAS,sBAAsB,OAAwC;CACrE,MAAM,WAA0B,EAAE;AAElC,MAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,OAAO,KAAK,QAAQ;AAE1B,MAAI,SAAS,QAAQ;GACnB,MAAM,gBAAgB,KAAK,MAAM,QAAQ,MAAM,EAAE,iBAAiB;GAGlE,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,SAAS,UAAa,CAAC,EAAE,QAAQ;AAE9E,OAAI,cAAc,SAAS,GAAG;AAC5B,SAAK,IAAI,IAAI,GAAG,IAAI,cAAc,QAAQ,KAAK;KAE7C,MAAM,KADO,cAAc,GACX;AAChB,cAAS,KAAK;MACZ,MAAM;MACN,SAAS,OAAO,GAAG,aAAa,WAAW,GAAG,WAAW,KAAK,UAAU,GAAG,SAAS;MACpF,cAAc,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;MAClD,CAAC;;AAEJ,QAAI,UAAU,SAAS,EACrB,UAAS,KAAK;KACZ,MAAM;KACN,SAAS,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;KAChD,CAAC;UAEC;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAQ,SAAS;KAAM,CAAC;;aAEvC,SAAS,SAAS;GAC3B,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,aAAa;GAC1D,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,SAAS,UAAa,CAAC,EAAE,QAAQ;AAE9E,OAAI,UAAU,SAAS,GAAG;IACxB,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KACZ,MAAM;KACN,SAAS,QAAQ;KACjB,YAAY,UAAU,KAAK,GAAG,OAAO;MACnC,IAAI,eAAe,EAAE,aAAc,KAAK,GAAG;MAC3C,MAAM;MACN,UAAU;OACR,MAAM,EAAE,aAAc;OACtB,WAAW,KAAK,UAAU,EAAE,aAAc,QAAQ,EAAE,CAAC;OACtD;MACF,EAAE;KACJ,CAAC;UACG;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAa,SAAS;KAAM,CAAC;;;;AAKzD,QAAO;;;;;AAMT,SAAS,uBAAuB,cAAqD;AACnF,QAAO,aAAa,kBAAkB,KAAK,IAAI,OAAO;EACpD,MAAM;EACN,SAAS,OAAO,GAAG,aAAa,WAAW,GAAG,WAAW,KAAK,UAAU,GAAG,SAAS;EACpF,cAAc,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;EAClD,EAAE;;;;;AAML,SAAS,aAAa,aAAqD;AACzE,KAAI,CAAC,eAAe,YAAY,WAAW,EAAG,QAAO,EAAE;AAEvD,QADc,YAAY,SAAS,MAAM,EAAE,wBAAwB,EAAE,CAAC,CACzD,KAAK,OAAO;EACvB,MAAM;EACN,UAAU;GACR,MAAM,EAAE;GACR,aAAa,EAAE;GACf,YAAY,EAAE;GACf;EACF,EAAE;;AAKL,SAAgB,0BACd,IACA,UACA,SACA,UAWM;CACN,MAAM,EAAE,WAAW;CACnB,MAAM,UAAwB;EAC5B,WAAW;EACX,OAAO,SAAS;EAChB,OAAO,EAAE;EACT,qBAAqB,EAAE;EACxB;CAED,IAAI,UAAU,QAAQ,SAAS;AAC/B,IAAG,GAAG,YAAY,QAAgB;AAChC,YAAU,QAAQ,WAChB,eAAe,KAAK,IAAI,UAAU,SAAS,UAAU,QAAQ,CAAC,OAAO,QAAiB;GACpF,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,UAAO,MAAM,gCAAgC,MAAM;AACnD,OAAI;AACF,OAAG,KACD,KAAK,UAAU,EACb,OAAO;KAAE,MAAM;KAAI,SAAS;KAAK,QAAQ;KAAY,EACtD,CAAC,CACH;YACM,SAAS;AAChB,aAAS,OAAO,MACd,mCAAmC,mBAAmB,QAAQ,QAAQ,UAAU,YACjF;;IAEH,CACH;GACD;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UAWA,SACe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;UACjB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS,mBAAmB;GAAU,QAAQ;GAAoB,EACrF,CAAC,CACH;AACD;;CAIF,MAAM,WAAW,OAAO,SAAS,OAAO;AACxC,KAAI,UAAU;AACZ,UAAQ,YAAY;AACpB,UAAQ,QAAQ,SAAS,SAAS,SAAS;AAC3C,UAAQ,QAAQ,aAAa,SAAS,MAAM;AAC5C,KAAG,KAAK,KAAK,UAAU,EAAE,eAAe,EAAE,EAAE,CAAC,CAAC;AAC9C;;AAIF,KAAI,CAAC,QAAQ,WAAW;AACtB,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS;GAAkB,QAAQ;GAAuB,EAC7E,CAAC,CACH;AACD;;CAIF,IAAI;AAEJ,KAAI,OAAO,eAAe;AACxB,MAAI,CAAC,OAAO,cAAc,SAAS,CAAC,MAAM,QAAQ,OAAO,cAAc,MAAM,EAAE;AAC7E,MAAG,KACD,KAAK,UAAU,EACb,OAAO;IACL,MAAM;IACN,SAAS;IACT,QAAQ;IACT,EACF,CAAC,CACH;AACD;;AAEF,gBAAc,sBAAsB,OAAO,cAAc,MAAM;YACtD,OAAO,cAAc;AAC9B,MACE,CAAC,OAAO,aAAa,qBACrB,CAAC,MAAM,QAAQ,OAAO,aAAa,kBAAkB,EACrD;AACA,MAAG,KACD,KAAK,UAAU,EACb,OAAO;IACL,MAAM;IACN,SAAS;IACT,QAAQ;IACT,EACF,CAAC,CACH;AACD;;AAEF,gBAAc,uBAAuB,OAAO,aAAa;QACpD;AACL,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GACL,MAAM;GACN,SAAS;GACT,QAAQ;GACT,EACF,CAAC,CACH;AACD;;CAIF,MAAM,gBAAuC;EAC3C,OAAO,QAAQ;EACf,UAAU,CAAC,GAAG,QAAQ,qBAAqB,GAAG,YAAY;EAC1D,QAAQ;EACR,OAAO,QAAQ,MAAM,SAAS,IAAI,QAAQ,QAAQ;EAClD,eAAe;EAChB;CAED,MAAM,SAAS,SAAS,UAAU;CAClC,MAAM,UAAU,aACd,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;CACD,MAAM,OAAO;AAEb,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KAAI,CAAC,SAAS;AACZ,MAAI,kBAAkB,SAAS,QAAQ,SAAS,eAAe,EAAE;AAC/D,YAAS,OAAO,KAAK,mDAAmD;AACxE,WAAQ,IAAI;IACV,QAAQ;IACR;IACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;IACtD,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAG,oBAAoB,SAAS,QAAQ,SAAS,eAAe;KACjE;IACF,CAAC;AACF,MAAG,MAAM,MAAM,kCAAkC;AACjD;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAG,oBAAoB,SAAS,QAAQ,SAAS,eAAe;IACjE;GACF,CAAC;AACF,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS;GAAsB,QAAQ;GAAa,EACvE,CAAC,CACH;AACD;;AAIF,SAAQ,oBAAoB,KAAK,GAAG,YAAY;CAEhD,MAAM,WAAW,MAAM,gBAAgB,SAAS,cAAc;CAC9D,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;AAGtE,KAAI,gBAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GACL,MAAM,WAAW,OAAO;GACxB,SAAS,SAAS,MAAM;GACxB,QAAQ,SAAS,MAAM,QAAQ;GAChC,EACF,CAAC,CACH;AACD;;AAIF,KAAI,gBAAgB,SAAS,EAAE;AAC7B,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,YAAY;EAClB,IAAI;EACJ,IAAI;AAEJ,MAAI,OAAO,UAAU,UAAU,UAAU;AACvC,cAAW,aAAa,UAAU,UAAU,MAAM;AAClD,UAAO,UAAU;SACZ;AACL,cAAW,UAAU,MAAM,eAAe;AAC1C,UAAO,UAAU,MAAM;;AAGzB,KAAG,KACD,KAAK,UAAU,EACb,eAAe;GACb,WAAW,EACT,OAAO,CAAC,EAAE,YAAY;IAAE;IAAU;IAAM,EAAE,CAAC,EAC5C;GACD,cAAc;GACf,EACF,CAAC,CACH;AAED,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS;GACV,CAAC;AACF;;AAIF,KAAI,+BAA+B,SAAS,EAAE;EAC5C,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,UAAU,SAAS;EACzB,MAAM,YAAsB,EAAE;AAC9B,OAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,UACvC,WAAU,KAAK,QAAQ,MAAM,GAAG,IAAI,UAAU,CAAC;EAGjD,MAAM,eAAe,yBAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;EAC5B,IAAI,cAAc;AAGlB,MAAI,QAAQ,WAAW,GACrB;OAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,EAAE,EACrC,EACF,CAAC,CACH;QAGH,MAAK,IAAI,IAAI,GAAG,IAAI,UAAU,QAAQ,KAAK;AACzC,OAAI,GAAG,SAAU;GACjB,MAAM,aAAa,eAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACtF,OAAI,aAAa,EAAG,OAAM,MAAM,YAAY,cAAc,OAAO;AACjE,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;AAEF,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,UAAU,IAAI,CAAC,EAAE,EAC/C,EACF,CAAC,CACH;AACD,iBAAc,MAAM;AACpB,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;;AAKN,MAAI,aAAa;AACf,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;EAIF,MAAM,oBAAoB,SAAS,UAAU,KAAK,QAAQ;GACxD,GAAG;GACH,YAAY,GAAG,MAAM,oBAAoB;GAC1C,EAAE;AAGH,MAAI,CAAC,GAAG,UAAU;GAChB,MAAM,UAAU,eACd,UAAU,QACV,QACA,SACA,iBACA,YACD;AACD,OAAI,UAAU,EAAG,OAAM,MAAM,SAAS,cAAc,OAAO;AAC3D,OAAI,cAAc,OAAO,SAAS;AAChC,OAAG,SAAS;AACZ,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,kBAAc,SAAS;AACvB;;GAGF,MAAM,gBAAgB,kBAAkB,KAAK,OAAO;IAClD,IAAI;AACJ,QAAI;AACF,eAAU,KAAK,MAAM,GAAG,aAAa,KAAK;YACpC;AACN,cAAS,OAAO,KACd,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,eAAU,EAAE;;AAEd,WAAO;KACL,MAAM,GAAG;KACT,MAAM;KACN,IAAI,GAAG;KACR;KACD;AAEF,MAAG,KAAK,KAAK,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;AACxD,iBAAc,MAAM;;AAGtB,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS,WAAW;GACpB,YAAY,kBAAkB,KAAK,QAAQ;IACzC,IAAI,GAAG;IACP,MAAM;IACN,UAAU;KACR,MAAM,GAAG;KACT,WAAW,GAAG;KACf;IACF,EAAE;GACJ,CAAC;AACF;;AAIF,KAAI,eAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,UAAU,SAAS;AAEzB,MAAI,QAAQ,WAAW,GAAG;AACxB,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,EAAE,EACrC,EACF,CAAC,CACH;AACD,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AACD;;EAIF,MAAM,SAAmB,EAAE;AAC3B,OAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,UACvC,QAAO,KAAK,QAAQ,MAAM,GAAG,IAAI,UAAU,CAAC;EAG9C,MAAM,eAAe,yBAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;EAC5B,IAAI,cAAc;AAGlB,OAAK,IAAI,IAAI,GAAG,IAAI,OAAO,QAAQ,KAAK;AACtC,OAAI,GAAG,SAAU;GACjB,MAAM,aAAa,eAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACtF,OAAI,aAAa,EAAG,OAAM,MAAM,YAAY,cAAc,OAAO;AACjE,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;AAEF,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,OAAO,IAAI,CAAC,EAAE,EAC5C,EACF,CAAC,CACH;AACD,iBAAc,MAAM;AACpB,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;;AAIJ,MAAI,aAAa;AACf,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAAE,MAAM;GAAa;GAAS,CAAC;AAChE;;AAIF,KAAI,mBAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,eAAe,yBAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;AAE5B,MAAI,GAAG,UAAU;AACf,iBAAc,SAAS;AACvB;;EAEF,MAAM,UAAU,eAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACnF,MAAI,UAAU,EAAG,OAAM,MAAM,SAAS,cAAc,OAAO;AAC3D,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAEF,MAAI,GAAG,UAAU;AACf,iBAAc,SAAS;AACvB;;EAGF,MAAM,gBAAgB,SAAS,UAAU,KAAK,IAAI,MAAM;GACtD,IAAI;AACJ,OAAI;AACF,cAAU,KAAK,MAAM,GAAG,aAAa,KAAK;WACpC;AACN,aAAS,OAAO,KACd,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,cAAU,EAAE;;AAEd,UAAO;IACL,MAAM,GAAG;IACT,MAAM;IACN,IAAI,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;IACxC;IACD;AAEF,KAAG,KAAK,KAAK,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;AACxD,gBAAc,MAAM;AAEpB,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS;GACT,YAAY,SAAS,UAAU,KAAK,IAAI,OAAO;IAC7C,IAAI,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;IACvC,MAAM;IACN,UAAU;KACR,MAAM,GAAG;KACT,WAAW,GAAG;KACf;IACF,EAAE;GACJ,CAAC;AACF;;AAIF,SAAQ,IAAI;EACV,QAAQ;EACR;EACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;EACtD,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,IAAG,KACD,KAAK,UAAU,EACb,OAAO;EACL,MAAM;EACN,SAAS;EACT,QAAQ;EACT,EACF,CAAC,CACH"}
+{"version":3,"file":"ws-gemini-live.js","names":[],"sources":["../src/ws-gemini-live.ts"],"sourcesContent":["/**\n * WebSocket handler for Gemini Live BidiGenerateContent API.\n *\n * Accepts setup, clientContent, and toolResponse messages over WebSocket\n * and responds with setupComplete, serverContent, toolCall, and error\n * messages in the Gemini Live streaming format.\n */\n\nimport type {\n  Fixture,\n  ChatMessage,\n  ChatCompletionRequest,\n  ToolDefinition,\n  AudioResponse,\n} from \"./types.js\";\nimport { matchFixture } from \"./router.js\";\nimport {\n  isTextResponse,\n  isToolCallResponse,\n  isContentWithToolCallsResponse,\n  isErrorResponse,\n  isAudioResponse,\n  flattenHeaders,\n  formatToMime,\n  generateToolCallId,\n  resolveResponse,\n  resolveStrictMode,\n  strictOverrideField,\n} from \"./helpers.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\nimport { DEFAULT_TEST_ID, type Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport type { WebSocketConnection } from \"./ws-framing.js\";\n\n// ─── Gemini Live protocol types ─────────────────────────────────────────────\n\ninterface GeminiLivePart {\n  text?: string;\n  thought?: boolean;\n  functionCall?: { name: string; args: Record<string, unknown> };\n  functionResponse?: { name: string; response: unknown; id?: string };\n  inlineData?: { mimeType: string; data: string };\n}\n\ninterface GeminiLiveTurn {\n  role: string;\n  parts: GeminiLivePart[];\n}\n\ninterface GeminiLiveFunctionDeclaration {\n  name: string;\n  description?: string;\n  parameters?: object;\n}\n\ninterface GeminiLiveToolDef {\n  functionDeclarations?: GeminiLiveFunctionDeclaration[];\n}\n\ninterface GeminiLiveSetup {\n  model?: string;\n  generationConfig?: Record<string, unknown>;\n  tools?: GeminiLiveToolDef[];\n}\n\ninterface GeminiLiveClientContent {\n  turns: GeminiLiveTurn[];\n  turnComplete?: boolean;\n}\n\ninterface GeminiLiveFunctionResponse {\n  id?: string;\n  name: string;\n  response: unknown;\n}\n\ninterface GeminiLiveToolResponse {\n  functionResponses: GeminiLiveFunctionResponse[];\n}\n\ninterface GeminiLiveMessage {\n  setup?: GeminiLiveSetup;\n  config?: GeminiLiveSetup;\n  clientContent?: GeminiLiveClientContent;\n  toolResponse?: GeminiLiveToolResponse;\n}\n\n// ─── Session state ──────────────────────────────────────────────────────────\n\ninterface SessionState {\n  setupDone: boolean;\n  model: string;\n  tools: ToolDefinition[];\n  conversationHistory: ChatMessage[];\n}\n\n// ─── Helpers ────────────────────────────────────────────────────────────────\n\nconst WS_PATH = \"/ws/google.ai.generativelanguage.v1beta.GenerativeService.BidiGenerateContent\";\n\n/**\n * Map HTTP status codes to gRPC error codes.\n * Gemini Live uses gRPC codes, not HTTP status codes.\n */\nfunction httpToGrpc(httpCode: number): number {\n  switch (httpCode) {\n    case 400:\n      return 3; // INVALID_ARGUMENT\n    case 401:\n      return 16; // UNAUTHENTICATED\n    case 403:\n      return 7; // PERMISSION_DENIED\n    case 404:\n      return 5; // NOT_FOUND\n    case 409:\n      return 10; // ABORTED\n    case 429:\n      return 8; // RESOURCE_EXHAUSTED\n    case 501:\n      return 12; // UNIMPLEMENTED\n    case 503:\n      return 14; // UNAVAILABLE\n    default:\n      return 13; // INTERNAL\n  }\n}\n\n/**\n * Convert Gemini Live turns into ChatMessage[] for fixture matching.\n */\nfunction geminiTurnsToMessages(turns: GeminiLiveTurn[]): ChatMessage[] {\n  const messages: ChatMessage[] = [];\n\n  for (const turn of turns) {\n    const role = turn.role ?? \"user\";\n\n    if (role === \"user\") {\n      const funcResponses = turn.parts.filter((p) => p.functionResponse);\n      // inlineData parts (e.g. client audio input) are silently skipped —\n      // only text and functionResponse parts are relevant for fixture matching.\n      const textParts = turn.parts.filter((p) => p.text !== undefined && !p.thought);\n\n      if (funcResponses.length > 0) {\n        for (let i = 0; i < funcResponses.length; i++) {\n          const part = funcResponses[i];\n          const fr = part.functionResponse!;\n          messages.push({\n            role: \"tool\",\n            content: typeof fr.response === \"string\" ? fr.response : JSON.stringify(fr.response),\n            tool_call_id: fr.id ?? `call_gemini_${fr.name}_${i}`,\n          });\n        }\n        if (textParts.length > 0) {\n          messages.push({\n            role: \"user\",\n            content: textParts.map((p) => p.text!).join(\"\"),\n          });\n        }\n      } else {\n        const text = textParts.map((p) => p.text!).join(\"\");\n        messages.push({ role: \"user\", content: text });\n      }\n    } else if (role === \"model\") {\n      const funcCalls = turn.parts.filter((p) => p.functionCall);\n      const textParts = turn.parts.filter((p) => p.text !== undefined && !p.thought);\n\n      if (funcCalls.length > 0) {\n        const text = textParts.map((p) => p.text!).join(\"\");\n        messages.push({\n          role: \"assistant\",\n          content: text || 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  return messages;\n}\n\n/**\n * Convert toolResponse messages into ChatMessage[] for fixture matching.\n */\nfunction toolResponseToMessages(toolResponse: GeminiLiveToolResponse): ChatMessage[] {\n  return toolResponse.functionResponses.map((fr, i) => ({\n    role: \"tool\" as const,\n    content: typeof fr.response === \"string\" ? fr.response : JSON.stringify(fr.response),\n    tool_call_id: fr.id ?? `call_gemini_${fr.name}_${i}`,\n  }));\n}\n\n/**\n * Convert Gemini tool definitions to ChatCompletion ToolDefinition[].\n */\nfunction convertTools(geminiTools?: GeminiLiveToolDef[]): ToolDefinition[] {\n  if (!geminiTools || geminiTools.length === 0) return [];\n  const decls = geminiTools.flatMap((t) => t.functionDeclarations ?? []);\n  return 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// ─── Main handler ───────────────────────────────────────────────────────────\n\nexport function handleWebSocketGeminiLive(\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n): void {\n  const { logger } = defaults;\n  const session: SessionState = {\n    setupDone: false,\n    model: defaults.model,\n    tools: [],\n    conversationHistory: [],\n  };\n\n  let pending = Promise.resolve();\n  ws.on(\"message\", (raw: string) => {\n    pending = pending.then(() =>\n      processMessage(raw, ws, fixtures, journal, defaults, session).catch((err: unknown) => {\n        const msg = err instanceof Error ? err.message : \"Internal error\";\n        logger.error(`WebSocket Gemini Live error: ${msg}`);\n        try {\n          ws.send(\n            JSON.stringify({\n              error: { code: 13, message: msg, status: \"INTERNAL\" },\n            }),\n          );\n        } catch (sendErr) {\n          defaults.logger.debug(\n            `Failed to send error to client: ${sendErr instanceof Error ? sendErr.message : \"unknown\"}`,\n          );\n        }\n      }),\n    );\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: {\n    latency: number;\n    chunkSize: number;\n    replaySpeed?: number;\n    model: string;\n    logger: Logger;\n    strict?: boolean;\n    requestTransform?: (req: ChatCompletionRequest) => ChatCompletionRequest;\n    testId?: string;\n    upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  },\n  session: SessionState,\n): Promise<void> {\n  let parsed: GeminiLiveMessage;\n  try {\n    parsed = JSON.parse(raw) as GeminiLiveMessage;\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    ws.send(\n      JSON.stringify({\n        error: { code: 3, message: `Malformed JSON: ${detail}`, status: \"INVALID_ARGUMENT\" },\n      }),\n    );\n    return;\n  }\n\n  // Handle setup message (accept both `setup` and `config` as aliases)\n  const setupMsg = parsed.setup ?? parsed.config;\n  if (setupMsg) {\n    session.setupDone = true;\n    session.model = setupMsg.model ?? defaults.model;\n    session.tools = convertTools(setupMsg.tools);\n    ws.send(JSON.stringify({ setupComplete: {} }));\n    return;\n  }\n\n  // Reject messages before setup\n  if (!session.setupDone) {\n    ws.send(\n      JSON.stringify({\n        error: { code: 9, message: \"Setup required\", status: \"FAILED_PRECONDITION\" },\n      }),\n    );\n    return;\n  }\n\n  // Build messages from this interaction\n  let newMessages: ChatMessage[];\n\n  if (parsed.clientContent) {\n    if (!parsed.clientContent.turns || !Array.isArray(parsed.clientContent.turns)) {\n      ws.send(\n        JSON.stringify({\n          error: {\n            code: 3,\n            message: \"Missing 'turns' in clientContent\",\n            status: \"INVALID_ARGUMENT\",\n          },\n        }),\n      );\n      return;\n    }\n    newMessages = geminiTurnsToMessages(parsed.clientContent.turns);\n  } else if (parsed.toolResponse) {\n    if (\n      !parsed.toolResponse.functionResponses ||\n      !Array.isArray(parsed.toolResponse.functionResponses)\n    ) {\n      ws.send(\n        JSON.stringify({\n          error: {\n            code: 3,\n            message: \"Missing 'functionResponses' in toolResponse\",\n            status: \"INVALID_ARGUMENT\",\n          },\n        }),\n      );\n      return;\n    }\n    newMessages = toolResponseToMessages(parsed.toolResponse);\n  } else {\n    ws.send(\n      JSON.stringify({\n        error: {\n          code: 3,\n          message: \"Expected clientContent or toolResponse\",\n          status: \"INVALID_ARGUMENT\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Build completion request for fixture matching (include new messages speculatively)\n  const geminiContextHeader = defaults.upgradeHeaders?.[\"x-aimock-context\"];\n  const geminiContext =\n    typeof geminiContextHeader === \"string\"\n      ? geminiContextHeader\n      : Array.isArray(geminiContextHeader) && geminiContextHeader.length > 0\n        ? geminiContextHeader[0]\n        : undefined;\n\n  const completionReq: ChatCompletionRequest = {\n    model: session.model,\n    messages: [...session.conversationHistory, ...newMessages],\n    stream: true,\n    tools: session.tools.length > 0 ? session.tools : undefined,\n    _endpointType: \"chat\",\n    _context: geminiContext,\n  };\n\n  const testId = defaults.testId ?? DEFAULT_TEST_ID;\n  const fixture = matchFixture(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n  const path = WS_PATH;\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (!fixture) {\n    if (resolveStrictMode(defaults.strict, defaults.upgradeHeaders)) {\n      defaults.logger.warn(`STRICT: No fixture matched for WebSocket message`);\n      journal.add({\n        method: \"WS\",\n        path,\n        headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n        body: completionReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n        },\n      });\n      ws.close(1008, \"Strict mode: no fixture matched\");\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n      },\n    });\n    ws.send(\n      JSON.stringify({\n        error: { code: 5, message: \"No fixture matched\", status: \"NOT_FOUND\" },\n      }),\n    );\n    return;\n  }\n\n  // Commit messages to conversation history only after successful fixture match\n  session.conversationHistory.push(...newMessages);\n\n  const response = await resolveResponse(fixture, completionReq);\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: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    ws.send(\n      JSON.stringify({\n        error: {\n          code: httpToGrpc(status),\n          message: response.error.message,\n          status: response.error.type ?? \"INTERNAL\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Audio response — single frame with inlineData and turnComplete: true\n  if (isAudioResponse(response)) {\n    journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const audioResp = response as AudioResponse;\n    let mimeType: string;\n    let data: string;\n\n    if (typeof audioResp.audio === \"string\") {\n      mimeType = formatToMime(audioResp.format ?? \"mp3\");\n      data = audioResp.audio;\n    } else {\n      mimeType = audioResp.audio.contentType ?? \"audio/mpeg\";\n      data = audioResp.audio.b64Json;\n    }\n\n    ws.send(\n      JSON.stringify({\n        serverContent: {\n          modelTurn: {\n            parts: [{ inlineData: { mimeType, data } }],\n          },\n          turnComplete: true,\n        },\n      }),\n    );\n\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: \"[audio]\",\n    });\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: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const content = response.content;\n    const chunkList: string[] = [];\n    for (let i = 0; i < content.length; i += chunkSize) {\n      chunkList.push(content.slice(i, i + chunkSize));\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n    let interrupted = false;\n\n    // Stream text content chunks (turnComplete omitted — sent as a separate message later)\n    if (content.length === 0) {\n      if (!ws.isClosed) {\n        ws.send(\n          JSON.stringify({\n            serverContent: {\n              modelTurn: { parts: [{ text: \"\" }] },\n            },\n          }),\n        );\n      }\n    } else {\n      for (let i = 0; i < chunkList.length; i++) {\n        if (ws.isClosed) break;\n        const chunkDelay = calculateDelay(i, undefined, latency, recordedTimings, replaySpeed);\n        if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);\n        if (interruption?.signal.aborted) {\n          interrupted = true;\n          break;\n        }\n        if (ws.isClosed) break;\n\n        ws.send(\n          JSON.stringify({\n            serverContent: {\n              modelTurn: { parts: [{ text: chunkList[i] }] },\n            },\n          }),\n        );\n        interruption?.tick();\n        if (interruption?.signal.aborted) {\n          interrupted = true;\n          break;\n        }\n      }\n    }\n\n    if (interrupted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    // Pre-compute tool calls with stable IDs so wire message and history match\n    const resolvedToolCalls = response.toolCalls.map((tc) => ({\n      ...tc,\n      resolvedId: tc.id ?? generateToolCallId(),\n    }));\n\n    // Send tool calls\n    if (!ws.isClosed) {\n      const tcDelay = calculateDelay(\n        chunkList.length,\n        undefined,\n        latency,\n        recordedTimings,\n        replaySpeed,\n      );\n      if (tcDelay > 0) await delay(tcDelay, interruption?.signal);\n      if (interruption?.signal.aborted) {\n        ws.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n        interruption?.cleanup();\n        return;\n      }\n\n      const functionCalls = resolvedToolCalls.map((tc) => {\n        let argsObj: Record<string, unknown>;\n        try {\n          argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n        } catch {\n          defaults.logger.warn(\n            `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n          );\n          argsObj = {};\n        }\n        return {\n          name: tc.name,\n          args: argsObj,\n          id: tc.resolvedId,\n        };\n      });\n\n      ws.send(JSON.stringify({ toolCall: { functionCalls } }));\n      interruption?.tick();\n    }\n\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send turnComplete\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add to conversation history using the same resolved IDs from the wire message\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: content || null,\n      tool_calls: resolvedToolCalls.map((tc) => ({\n        id: tc.resolvedId,\n        type: \"function\" as const,\n        function: {\n          name: tc.name,\n          arguments: tc.arguments,\n        },\n      })),\n    });\n    return;\n  }\n\n  // Text response — stream chunks with serverContent\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const content = response.content;\n\n    if (content.length === 0) {\n      if (ws.isClosed) return;\n      // Empty content: send empty modelTurn, then separate turnComplete\n      ws.send(\n        JSON.stringify({\n          serverContent: {\n            modelTurn: { parts: [{ text: \"\" }] },\n          },\n        }),\n      );\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n      return;\n    }\n\n    // Chunk the content\n    const chunks: string[] = [];\n    for (let i = 0; i < content.length; i += chunkSize) {\n      chunks.push(content.slice(i, i + chunkSize));\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n    let interrupted = false;\n\n    // Stream content chunks without turnComplete (sent separately after)\n    for (let i = 0; i < chunks.length; i++) {\n      if (ws.isClosed) break;\n      const chunkDelay = calculateDelay(i, undefined, latency, recordedTimings, replaySpeed);\n      if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);\n      if (interruption?.signal.aborted) {\n        interrupted = true;\n        break;\n      }\n      if (ws.isClosed) break;\n\n      ws.send(\n        JSON.stringify({\n          serverContent: {\n            modelTurn: { parts: [{ text: chunks[i] }] },\n          },\n        }),\n      );\n      interruption?.tick();\n      if (interruption?.signal.aborted) {\n        interrupted = true;\n        break;\n      }\n    }\n\n    if (interrupted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send separate turnComplete message\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add assistant response to conversation history\n    session.conversationHistory.push({ role: \"assistant\", content });\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path,\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const interruption = createInterruptionSignal(fixture);\n    const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;\n    const { recordedTimings } = fixture;\n\n    if (ws.isClosed) {\n      interruption?.cleanup();\n      return;\n    }\n    const tcDelay = calculateDelay(0, undefined, latency, recordedTimings, replaySpeed);\n    if (tcDelay > 0) await delay(tcDelay, interruption?.signal);\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n    if (ws.isClosed) {\n      interruption?.cleanup();\n      return;\n    }\n\n    const functionCalls = response.toolCalls.map((tc, i) => {\n      let argsObj: Record<string, unknown>;\n      try {\n        argsObj = JSON.parse(tc.arguments || \"{}\") as Record<string, unknown>;\n      } catch {\n        defaults.logger.warn(\n          `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n        );\n        argsObj = {};\n      }\n      return {\n        name: tc.name,\n        args: argsObj,\n        id: tc.id ?? `call_gemini_${tc.name}_${i}`,\n      };\n    });\n\n    ws.send(JSON.stringify({ toolCall: { functionCalls } }));\n    interruption?.tick();\n\n    if (interruption?.signal.aborted) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n      interruption?.cleanup();\n      return;\n    }\n\n    interruption?.cleanup();\n\n    // Send turnComplete after tool call\n    if (!ws.isClosed) {\n      ws.send(\n        JSON.stringify({\n          serverContent: { turnComplete: true },\n        }),\n      );\n    }\n\n    // Add assistant tool_calls to conversation history\n    session.conversationHistory.push({\n      role: \"assistant\",\n      content: null,\n      tool_calls: response.toolCalls.map((tc, i) => ({\n        id: tc.id ?? `call_gemini_${tc.name}_${i}`,\n        type: \"function\" as const,\n        function: {\n          name: tc.name,\n          arguments: tc.arguments,\n        },\n      })),\n    });\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: \"WS\",\n    path,\n    headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  ws.send(\n    JSON.stringify({\n      error: {\n        code: 13,\n        message: \"Fixture response did not match any known type\",\n        status: \"INTERNAL\",\n      },\n    }),\n  );\n}\n"],"mappings":";;;;;;;;AAmGA,MAAM,UAAU;;;;;AAMhB,SAAS,WAAW,UAA0B;AAC5C,SAAQ,UAAR;EACE,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,KAAK,IACH,QAAO;EACT,QACE,QAAO;;;;;;AAOb,SAAS,sBAAsB,OAAwC;CACrE,MAAM,WAA0B,EAAE;AAElC,MAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,OAAO,KAAK,QAAQ;AAE1B,MAAI,SAAS,QAAQ;GACnB,MAAM,gBAAgB,KAAK,MAAM,QAAQ,MAAM,EAAE,iBAAiB;GAGlE,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,SAAS,UAAa,CAAC,EAAE,QAAQ;AAE9E,OAAI,cAAc,SAAS,GAAG;AAC5B,SAAK,IAAI,IAAI,GAAG,IAAI,cAAc,QAAQ,KAAK;KAE7C,MAAM,KADO,cAAc,GACX;AAChB,cAAS,KAAK;MACZ,MAAM;MACN,SAAS,OAAO,GAAG,aAAa,WAAW,GAAG,WAAW,KAAK,UAAU,GAAG,SAAS;MACpF,cAAc,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;MAClD,CAAC;;AAEJ,QAAI,UAAU,SAAS,EACrB,UAAS,KAAK;KACZ,MAAM;KACN,SAAS,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;KAChD,CAAC;UAEC;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAQ,SAAS;KAAM,CAAC;;aAEvC,SAAS,SAAS;GAC3B,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,aAAa;GAC1D,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,SAAS,UAAa,CAAC,EAAE,QAAQ;AAE9E,OAAI,UAAU,SAAS,GAAG;IACxB,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KACZ,MAAM;KACN,SAAS,QAAQ;KACjB,YAAY,UAAU,KAAK,GAAG,OAAO;MACnC,IAAI,eAAe,EAAE,aAAc,KAAK,GAAG;MAC3C,MAAM;MACN,UAAU;OACR,MAAM,EAAE,aAAc;OACtB,WAAW,KAAK,UAAU,EAAE,aAAc,QAAQ,EAAE,CAAC;OACtD;MACF,EAAE;KACJ,CAAC;UACG;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAa,SAAS;KAAM,CAAC;;;;AAKzD,QAAO;;;;;AAMT,SAAS,uBAAuB,cAAqD;AACnF,QAAO,aAAa,kBAAkB,KAAK,IAAI,OAAO;EACpD,MAAM;EACN,SAAS,OAAO,GAAG,aAAa,WAAW,GAAG,WAAW,KAAK,UAAU,GAAG,SAAS;EACpF,cAAc,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;EAClD,EAAE;;;;;AAML,SAAS,aAAa,aAAqD;AACzE,KAAI,CAAC,eAAe,YAAY,WAAW,EAAG,QAAO,EAAE;AAEvD,QADc,YAAY,SAAS,MAAM,EAAE,wBAAwB,EAAE,CAAC,CACzD,KAAK,OAAO;EACvB,MAAM;EACN,UAAU;GACR,MAAM,EAAE;GACR,aAAa,EAAE;GACf,YAAY,EAAE;GACf;EACF,EAAE;;AAKL,SAAgB,0BACd,IACA,UACA,SACA,UAWM;CACN,MAAM,EAAE,WAAW;CACnB,MAAM,UAAwB;EAC5B,WAAW;EACX,OAAO,SAAS;EAChB,OAAO,EAAE;EACT,qBAAqB,EAAE;EACxB;CAED,IAAI,UAAU,QAAQ,SAAS;AAC/B,IAAG,GAAG,YAAY,QAAgB;AAChC,YAAU,QAAQ,WAChB,eAAe,KAAK,IAAI,UAAU,SAAS,UAAU,QAAQ,CAAC,OAAO,QAAiB;GACpF,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,UAAO,MAAM,gCAAgC,MAAM;AACnD,OAAI;AACF,OAAG,KACD,KAAK,UAAU,EACb,OAAO;KAAE,MAAM;KAAI,SAAS;KAAK,QAAQ;KAAY,EACtD,CAAC,CACH;YACM,SAAS;AAChB,aAAS,OAAO,MACd,mCAAmC,mBAAmB,QAAQ,QAAQ,UAAU,YACjF;;IAEH,CACH;GACD;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UAWA,SACe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;UACjB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS,mBAAmB;GAAU,QAAQ;GAAoB,EACrF,CAAC,CACH;AACD;;CAIF,MAAM,WAAW,OAAO,SAAS,OAAO;AACxC,KAAI,UAAU;AACZ,UAAQ,YAAY;AACpB,UAAQ,QAAQ,SAAS,SAAS,SAAS;AAC3C,UAAQ,QAAQ,aAAa,SAAS,MAAM;AAC5C,KAAG,KAAK,KAAK,UAAU,EAAE,eAAe,EAAE,EAAE,CAAC,CAAC;AAC9C;;AAIF,KAAI,CAAC,QAAQ,WAAW;AACtB,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS;GAAkB,QAAQ;GAAuB,EAC7E,CAAC,CACH;AACD;;CAIF,IAAI;AAEJ,KAAI,OAAO,eAAe;AACxB,MAAI,CAAC,OAAO,cAAc,SAAS,CAAC,MAAM,QAAQ,OAAO,cAAc,MAAM,EAAE;AAC7E,MAAG,KACD,KAAK,UAAU,EACb,OAAO;IACL,MAAM;IACN,SAAS;IACT,QAAQ;IACT,EACF,CAAC,CACH;AACD;;AAEF,gBAAc,sBAAsB,OAAO,cAAc,MAAM;YACtD,OAAO,cAAc;AAC9B,MACE,CAAC,OAAO,aAAa,qBACrB,CAAC,MAAM,QAAQ,OAAO,aAAa,kBAAkB,EACrD;AACA,MAAG,KACD,KAAK,UAAU,EACb,OAAO;IACL,MAAM;IACN,SAAS;IACT,QAAQ;IACT,EACF,CAAC,CACH;AACD;;AAEF,gBAAc,uBAAuB,OAAO,aAAa;QACpD;AACL,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GACL,MAAM;GACN,SAAS;GACT,QAAQ;GACT,EACF,CAAC,CACH;AACD;;CAIF,MAAM,sBAAsB,SAAS,iBAAiB;CACtD,MAAM,gBACJ,OAAO,wBAAwB,WAC3B,sBACA,MAAM,QAAQ,oBAAoB,IAAI,oBAAoB,SAAS,IACjE,oBAAoB,KACpB;CAER,MAAM,gBAAuC;EAC3C,OAAO,QAAQ;EACf,UAAU,CAAC,GAAG,QAAQ,qBAAqB,GAAG,YAAY;EAC1D,QAAQ;EACR,OAAO,QAAQ,MAAM,SAAS,IAAI,QAAQ,QAAQ;EAClD,eAAe;EACf,UAAU;EACX;CAED,MAAM,SAAS,SAAS,UAAU;CAClC,MAAM,UAAU,aACd,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;CACD,MAAM,OAAO;AAEb,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KAAI,CAAC,SAAS;AACZ,MAAI,kBAAkB,SAAS,QAAQ,SAAS,eAAe,EAAE;AAC/D,YAAS,OAAO,KAAK,mDAAmD;AACxE,WAAQ,IAAI;IACV,QAAQ;IACR;IACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;IACtD,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAG,oBAAoB,SAAS,QAAQ,SAAS,eAAe;KACjE;IACF,CAAC;AACF,MAAG,MAAM,MAAM,kCAAkC;AACjD;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAG,oBAAoB,SAAS,QAAQ,SAAS,eAAe;IACjE;GACF,CAAC;AACF,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAG,SAAS;GAAsB,QAAQ;GAAa,EACvE,CAAC,CACH;AACD;;AAIF,SAAQ,oBAAoB,KAAK,GAAG,YAAY;CAEhD,MAAM,WAAW,MAAM,gBAAgB,SAAS,cAAc;CAC9D,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;AAGtE,KAAI,gBAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GACL,MAAM,WAAW,OAAO;GACxB,SAAS,SAAS,MAAM;GACxB,QAAQ,SAAS,MAAM,QAAQ;GAChC,EACF,CAAC,CACH;AACD;;AAIF,KAAI,gBAAgB,SAAS,EAAE;AAC7B,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,YAAY;EAClB,IAAI;EACJ,IAAI;AAEJ,MAAI,OAAO,UAAU,UAAU,UAAU;AACvC,cAAW,aAAa,UAAU,UAAU,MAAM;AAClD,UAAO,UAAU;SACZ;AACL,cAAW,UAAU,MAAM,eAAe;AAC1C,UAAO,UAAU,MAAM;;AAGzB,KAAG,KACD,KAAK,UAAU,EACb,eAAe;GACb,WAAW,EACT,OAAO,CAAC,EAAE,YAAY;IAAE;IAAU;IAAM,EAAE,CAAC,EAC5C;GACD,cAAc;GACf,EACF,CAAC,CACH;AAED,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS;GACV,CAAC;AACF;;AAIF,KAAI,+BAA+B,SAAS,EAAE;EAC5C,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,UAAU,SAAS;EACzB,MAAM,YAAsB,EAAE;AAC9B,OAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,UACvC,WAAU,KAAK,QAAQ,MAAM,GAAG,IAAI,UAAU,CAAC;EAGjD,MAAM,eAAe,yBAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;EAC5B,IAAI,cAAc;AAGlB,MAAI,QAAQ,WAAW,GACrB;OAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,EAAE,EACrC,EACF,CAAC,CACH;QAGH,MAAK,IAAI,IAAI,GAAG,IAAI,UAAU,QAAQ,KAAK;AACzC,OAAI,GAAG,SAAU;GACjB,MAAM,aAAa,eAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACtF,OAAI,aAAa,EAAG,OAAM,MAAM,YAAY,cAAc,OAAO;AACjE,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;AAEF,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,UAAU,IAAI,CAAC,EAAE,EAC/C,EACF,CAAC,CACH;AACD,iBAAc,MAAM;AACpB,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;;AAKN,MAAI,aAAa;AACf,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;EAIF,MAAM,oBAAoB,SAAS,UAAU,KAAK,QAAQ;GACxD,GAAG;GACH,YAAY,GAAG,MAAM,oBAAoB;GAC1C,EAAE;AAGH,MAAI,CAAC,GAAG,UAAU;GAChB,MAAM,UAAU,eACd,UAAU,QACV,QACA,SACA,iBACA,YACD;AACD,OAAI,UAAU,EAAG,OAAM,MAAM,SAAS,cAAc,OAAO;AAC3D,OAAI,cAAc,OAAO,SAAS;AAChC,OAAG,SAAS;AACZ,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,kBAAc,SAAS;AACvB;;GAGF,MAAM,gBAAgB,kBAAkB,KAAK,OAAO;IAClD,IAAI;AACJ,QAAI;AACF,eAAU,KAAK,MAAM,GAAG,aAAa,KAAK;YACpC;AACN,cAAS,OAAO,KACd,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,eAAU,EAAE;;AAEd,WAAO;KACL,MAAM,GAAG;KACT,MAAM;KACN,IAAI,GAAG;KACR;KACD;AAEF,MAAG,KAAK,KAAK,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;AACxD,iBAAc,MAAM;;AAGtB,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS,WAAW;GACpB,YAAY,kBAAkB,KAAK,QAAQ;IACzC,IAAI,GAAG;IACP,MAAM;IACN,UAAU;KACR,MAAM,GAAG;KACT,WAAW,GAAG;KACf;IACF,EAAE;GACJ,CAAC;AACF;;AAIF,KAAI,eAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,UAAU,SAAS;AAEzB,MAAI,QAAQ,WAAW,GAAG;AACxB,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,EAAE,EACrC,EACF,CAAC,CACH;AACD,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AACD;;EAIF,MAAM,SAAmB,EAAE;AAC3B,OAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,UACvC,QAAO,KAAK,QAAQ,MAAM,GAAG,IAAI,UAAU,CAAC;EAG9C,MAAM,eAAe,yBAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;EAC5B,IAAI,cAAc;AAGlB,OAAK,IAAI,IAAI,GAAG,IAAI,OAAO,QAAQ,KAAK;AACtC,OAAI,GAAG,SAAU;GACjB,MAAM,aAAa,eAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACtF,OAAI,aAAa,EAAG,OAAM,MAAM,YAAY,cAAc,OAAO;AACjE,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;AAEF,OAAI,GAAG,SAAU;AAEjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe,EACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,OAAO,IAAI,CAAC,EAAE,EAC5C,EACF,CAAC,CACH;AACD,iBAAc,MAAM;AACpB,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;;AAIJ,MAAI,aAAa;AACf,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAAE,MAAM;GAAa;GAAS,CAAC;AAChE;;AAIF,KAAI,mBAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,eAAe,yBAAyB,QAAQ;EACtD,MAAM,cAAc,QAAQ,eAAe,SAAS;EACpD,MAAM,EAAE,oBAAoB;AAE5B,MAAI,GAAG,UAAU;AACf,iBAAc,SAAS;AACvB;;EAEF,MAAM,UAAU,eAAe,GAAG,QAAW,SAAS,iBAAiB,YAAY;AACnF,MAAI,UAAU,EAAG,OAAM,MAAM,SAAS,cAAc,OAAO;AAC3D,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAEF,MAAI,GAAG,UAAU;AACf,iBAAc,SAAS;AACvB;;EAGF,MAAM,gBAAgB,SAAS,UAAU,KAAK,IAAI,MAAM;GACtD,IAAI;AACJ,OAAI;AACF,cAAU,KAAK,MAAM,GAAG,aAAa,KAAK;WACpC;AACN,aAAS,OAAO,KACd,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,cAAU,EAAE;;AAEd,UAAO;IACL,MAAM,GAAG;IACT,MAAM;IACN,IAAI,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;IACxC;IACD;AAEF,KAAG,KAAK,KAAK,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;AACxD,gBAAc,MAAM;AAEpB,MAAI,cAAc,OAAO,SAAS;AAChC,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;AAC9D,iBAAc,SAAS;AACvB;;AAGF,gBAAc,SAAS;AAGvB,MAAI,CAAC,GAAG,SACN,IAAG,KACD,KAAK,UAAU,EACb,eAAe,EAAE,cAAc,MAAM,EACtC,CAAC,CACH;AAIH,UAAQ,oBAAoB,KAAK;GAC/B,MAAM;GACN,SAAS;GACT,YAAY,SAAS,UAAU,KAAK,IAAI,OAAO;IAC7C,IAAI,GAAG,MAAM,eAAe,GAAG,KAAK,GAAG;IACvC,MAAM;IACN,UAAU;KACR,MAAM,GAAG;KACT,WAAW,GAAG;KACf;IACF,EAAE;GACJ,CAAC;AACF;;AAIF,SAAQ,IAAI;EACV,QAAQ;EACR;EACA,SAAS,eAAe,SAAS,kBAAkB,EAAE,CAAC;EACtD,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,IAAG,KACD,KAAK,UAAU,EACb,OAAO;EACL,MAAM;EACN,SAAS;EACT,QAAQ;EACT,EACF,CAAC,CACH"}

package/dist/ws-realtime.cjs

@@ -409,10 +409,13 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 		transcription: "realtime-transcription",
 		translation: "realtime-translation"
 	}[session.type] ?? "realtime";
+	const realtimeContextHeader = defaults.upgradeHeaders?.["x-aimock-context"];
+	const realtimeContext = typeof realtimeContextHeader === "string" ? realtimeContextHeader : Array.isArray(realtimeContextHeader) && realtimeContextHeader.length > 0 ? realtimeContextHeader[0] : void 0;
 	const completionReq = {
 		model: session.model,
 		messages,
-		_endpointType: endpointType
+		_endpointType: endpointType,
+		_context: realtimeContext
 	};
 	const testId = defaults.testId ?? require_constants.DEFAULT_TEST_ID;
 	const fixture = require_router.matchFixture(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);