@copilotkit/aimock 1.7.0

package/dist/ws-gemini-live.cjs

@@ -0,0 +1,364 @@
+const require_helpers = require('./helpers.cjs');
+const require_router = require('./router.cjs');
+const require_sse_writer = require('./sse-writer.cjs');
+const require_interruption = require('./interruption.cjs');
+
+//#region src/ws-gemini-live.ts
+const WS_PATH = "/ws/google.ai.generativelanguage.v1beta.GenerativeService.BidiGenerateContent";
+/**
+* Convert Gemini Live turns into ChatMessage[] for fixture matching.
+*/
+function geminiTurnsToMessages(turns) {
+	const messages = [];
+	for (const turn of turns) {
+		const role = turn.role ?? "user";
+		if (role === "user") {
+			const funcResponses = turn.parts.filter((p) => p.functionResponse);
+			const textParts = turn.parts.filter((p) => p.text !== void 0);
+			if (funcResponses.length > 0) {
+				for (let i = 0; i < funcResponses.length; i++) {
+					const fr = funcResponses[i].functionResponse;
+					messages.push({
+						role: "tool",
+						content: typeof fr.response === "string" ? fr.response : JSON.stringify(fr.response),
+						tool_call_id: fr.id ?? `call_gemini_${fr.name}_${i}`
+					});
+				}
+				if (textParts.length > 0) messages.push({
+					role: "user",
+					content: textParts.map((p) => p.text).join("")
+				});
+			} else {
+				const text = textParts.map((p) => p.text).join("");
+				messages.push({
+					role: "user",
+					content: text
+				});
+			}
+		} else if (role === "model") {
+			const funcCalls = turn.parts.filter((p) => p.functionCall);
+			const textParts = turn.parts.filter((p) => p.text !== void 0);
+			if (funcCalls.length > 0) messages.push({
+				role: "assistant",
+				content: null,
+				tool_calls: funcCalls.map((p, i) => ({
+					id: `call_gemini_${p.functionCall.name}_${i}`,
+					type: "function",
+					function: {
+						name: p.functionCall.name,
+						arguments: JSON.stringify(p.functionCall.args)
+					}
+				}))
+			});
+			else {
+				const text = textParts.map((p) => p.text).join("");
+				messages.push({
+					role: "assistant",
+					content: text
+				});
+			}
+		}
+	}
+	return messages;
+}
+/**
+* Convert toolResponse messages into ChatMessage[] for fixture matching.
+*/
+function toolResponseToMessages(toolResponse) {
+	return toolResponse.functionResponses.map((fr, i) => ({
+		role: "tool",
+		content: typeof fr.response === "string" ? fr.response : JSON.stringify(fr.response),
+		tool_call_id: fr.id ?? `call_gemini_${fr.name}_${i}`
+	}));
+}
+/**
+* Convert Gemini tool definitions to ChatCompletion ToolDefinition[].
+*/
+function convertTools(geminiTools) {
+	if (!geminiTools || geminiTools.length === 0) return [];
+	return geminiTools.flatMap((t) => t.functionDeclarations ?? []).map((d) => ({
+		type: "function",
+		function: {
+			name: d.name,
+			description: d.description,
+			parameters: d.parameters
+		}
+	}));
+}
+function handleWebSocketGeminiLive(ws, fixtures, journal, defaults) {
+	const { logger } = defaults;
+	const session = {
+		setupDone: false,
+		model: defaults.model,
+		tools: [],
+		conversationHistory: []
+	};
+	let pending = Promise.resolve();
+	ws.on("message", (raw) => {
+		pending = pending.then(() => processMessage(raw, ws, fixtures, journal, defaults, session).catch((err) => {
+			const msg = err instanceof Error ? err.message : "Internal error";
+			logger.error(`WebSocket Gemini Live error: ${msg}`);
+			try {
+				ws.send(JSON.stringify({ error: {
+					code: 500,
+					message: msg,
+					status: "INTERNAL"
+				} }));
+			} catch {}
+		}));
+	});
+}
+async function processMessage(raw, ws, fixtures, journal, defaults, session) {
+	let parsed;
+	try {
+		parsed = JSON.parse(raw);
+	} catch {
+		ws.send(JSON.stringify({ error: {
+			code: 400,
+			message: "Malformed JSON",
+			status: "INVALID_ARGUMENT"
+		} }));
+		return;
+	}
+	if (parsed.setup) {
+		session.setupDone = true;
+		session.model = parsed.setup.model ?? defaults.model;
+		session.tools = convertTools(parsed.setup.tools);
+		ws.send(JSON.stringify({ setupComplete: {} }));
+		return;
+	}
+	if (!session.setupDone) {
+		ws.send(JSON.stringify({ error: {
+			code: 400,
+			message: "Setup required",
+			status: "FAILED_PRECONDITION"
+		} }));
+		return;
+	}
+	let newMessages;
+	if (parsed.clientContent) {
+		if (!parsed.clientContent.turns || !Array.isArray(parsed.clientContent.turns)) {
+			ws.send(JSON.stringify({ error: {
+				code: 400,
+				message: "Missing 'turns' in clientContent",
+				status: "INVALID_ARGUMENT"
+			} }));
+			return;
+		}
+		newMessages = geminiTurnsToMessages(parsed.clientContent.turns);
+	} else if (parsed.toolResponse) {
+		if (!parsed.toolResponse.functionResponses || !Array.isArray(parsed.toolResponse.functionResponses)) {
+			ws.send(JSON.stringify({ error: {
+				code: 400,
+				message: "Missing 'functionResponses' in toolResponse",
+				status: "INVALID_ARGUMENT"
+			} }));
+			return;
+		}
+		newMessages = toolResponseToMessages(parsed.toolResponse);
+	} else {
+		ws.send(JSON.stringify({ error: {
+			code: 400,
+			message: "Expected clientContent or toolResponse",
+			status: "INVALID_ARGUMENT"
+		} }));
+		return;
+	}
+	const completionReq = {
+		model: session.model,
+		messages: [...session.conversationHistory, ...newMessages],
+		stream: true,
+		tools: session.tools.length > 0 ? session.tools : void 0
+	};
+	const fixture = require_router.matchFixture(fixtures, completionReq, journal.fixtureMatchCounts);
+	const path = WS_PATH;
+	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures);
+	if (!fixture) {
+		if (defaults.strict) {
+			defaults.logger.warn(`STRICT: No fixture matched for WebSocket message`);
+			ws.close(1008, "Strict mode: no fixture matched");
+			return;
+		}
+		journal.add({
+			method: "WS",
+			path,
+			headers: {},
+			body: completionReq,
+			response: {
+				status: 404,
+				fixture: null
+			}
+		});
+		ws.send(JSON.stringify({ error: {
+			code: 404,
+			message: "No fixture matched",
+			status: "NOT_FOUND"
+		} }));
+		return;
+	}
+	session.conversationHistory.push(...newMessages);
+	const response = fixture.response;
+	const latency = fixture.latency ?? defaults.latency;
+	const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);
+	if (require_helpers.isErrorResponse(response)) {
+		const status = response.status ?? 500;
+		journal.add({
+			method: "WS",
+			path,
+			headers: {},
+			body: completionReq,
+			response: {
+				status,
+				fixture
+			}
+		});
+		ws.send(JSON.stringify({ error: {
+			code: status,
+			message: response.error.message,
+			status: "ERROR"
+		} }));
+		return;
+	}
+	if (require_helpers.isTextResponse(response)) {
+		const journalEntry = journal.add({
+			method: "WS",
+			path,
+			headers: {},
+			body: completionReq,
+			response: {
+				status: 200,
+				fixture
+			}
+		});
+		const content = response.content;
+		if (content.length === 0) {
+			if (ws.isClosed) return;
+			ws.send(JSON.stringify({ serverContent: {
+				modelTurn: { parts: [{ text: "" }] },
+				turnComplete: true
+			} }));
+			return;
+		}
+		const chunks = [];
+		for (let i = 0; i < content.length; i += chunkSize) chunks.push(content.slice(i, i + chunkSize));
+		const interruption = require_interruption.createInterruptionSignal(fixture);
+		let interrupted = false;
+		for (let i = 0; i < chunks.length; i++) {
+			if (ws.isClosed) break;
+			if (latency > 0) await require_sse_writer.delay(latency, interruption?.signal);
+			if (interruption?.signal.aborted) {
+				interrupted = true;
+				break;
+			}
+			if (ws.isClosed) break;
+			const isLast = i === chunks.length - 1;
+			ws.send(JSON.stringify({ serverContent: {
+				modelTurn: { parts: [{ text: chunks[i] }] },
+				turnComplete: isLast
+			} }));
+			interruption?.tick();
+			if (interruption?.signal.aborted) {
+				interrupted = true;
+				break;
+			}
+		}
+		if (interrupted) {
+			ws.destroy();
+			journalEntry.response.interrupted = true;
+			journalEntry.response.interruptReason = interruption?.reason();
+			interruption?.cleanup();
+			return;
+		}
+		interruption?.cleanup();
+		session.conversationHistory.push({
+			role: "assistant",
+			content
+		});
+		return;
+	}
+	if (require_helpers.isToolCallResponse(response)) {
+		const journalEntry = journal.add({
+			method: "WS",
+			path,
+			headers: {},
+			body: completionReq,
+			response: {
+				status: 200,
+				fixture
+			}
+		});
+		const interruption = require_interruption.createInterruptionSignal(fixture);
+		if (ws.isClosed) {
+			interruption?.cleanup();
+			return;
+		}
+		if (latency > 0) await require_sse_writer.delay(latency, interruption?.signal);
+		if (interruption?.signal.aborted) {
+			ws.destroy();
+			journalEntry.response.interrupted = true;
+			journalEntry.response.interruptReason = interruption?.reason();
+			interruption?.cleanup();
+			return;
+		}
+		if (ws.isClosed) {
+			interruption?.cleanup();
+			return;
+		}
+		const functionCalls = response.toolCalls.map((tc, i) => {
+			let argsObj;
+			try {
+				argsObj = JSON.parse(tc.arguments || "{}");
+			} catch {
+				defaults.logger.warn(`Malformed JSON in fixture tool call arguments for "${tc.name}": ${tc.arguments}`);
+				argsObj = {};
+			}
+			return {
+				name: tc.name,
+				args: argsObj,
+				id: tc.id ?? `call_gemini_${tc.name}_${i}`
+			};
+		});
+		ws.send(JSON.stringify({ toolCall: { functionCalls } }));
+		interruption?.tick();
+		if (interruption?.signal.aborted) {
+			ws.destroy();
+			journalEntry.response.interrupted = true;
+			journalEntry.response.interruptReason = interruption?.reason();
+			interruption?.cleanup();
+			return;
+		}
+		interruption?.cleanup();
+		session.conversationHistory.push({
+			role: "assistant",
+			content: null,
+			tool_calls: response.toolCalls.map((tc, i) => ({
+				id: tc.id ?? `call_gemini_${tc.name}_${i}`,
+				type: "function",
+				function: {
+					name: tc.name,
+					arguments: tc.arguments
+				}
+			}))
+		});
+		return;
+	}
+	journal.add({
+		method: "WS",
+		path,
+		headers: {},
+		body: completionReq,
+		response: {
+			status: 500,
+			fixture
+		}
+	});
+	ws.send(JSON.stringify({ error: {
+		code: 500,
+		message: "Fixture response did not match any known type",
+		status: "INTERNAL"
+	} }));
+}
+
+//#endregion
+exports.handleWebSocketGeminiLive = handleWebSocketGeminiLive;
+//# sourceMappingURL=ws-gemini-live.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ws-gemini-live.cjs","names":["matchFixture","isErrorResponse","isTextResponse","createInterruptionSignal","delay","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 { Fixture, ChatMessage, ChatCompletionRequest, ToolDefinition } from \"./types.js\";\nimport { matchFixture } from \"./router.js\";\nimport { isTextResponse, isToolCallResponse, isErrorResponse } from \"./helpers.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport { delay } from \"./sse-writer.js\";\nimport 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  functionCall?: { name: string; args: Record<string, unknown> };\n  functionResponse?: { name: string; response: unknown; id?: 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  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 * 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      const textParts = turn.parts.filter((p) => p.text !== undefined);\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);\n\n      if (funcCalls.length > 0) {\n        messages.push({\n          role: \"assistant\",\n          content: null,\n          tool_calls: funcCalls.map((p, i) => ({\n            id: `call_gemini_${p.functionCall!.name}_${i}`,\n            type: \"function\" as const,\n            function: {\n              name: p.functionCall!.name,\n              arguments: JSON.stringify(p.functionCall!.args),\n            },\n          })),\n        });\n      } else {\n        const text = textParts.map((p) => p.text!).join(\"\");\n        messages.push({ role: \"assistant\", content: text });\n      }\n    }\n  }\n\n  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: { latency: number; chunkSize: number; model: string; logger: Logger; strict?: boolean },\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: 500, message: msg, status: \"INTERNAL\" },\n            }),\n          );\n        } catch {\n          // Connection already gone — original error already logged above\n        }\n      }),\n    );\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: { latency: number; chunkSize: number; model: string; logger: Logger; strict?: boolean },\n  session: SessionState,\n): Promise<void> {\n  let parsed: GeminiLiveMessage;\n  try {\n    parsed = JSON.parse(raw) as GeminiLiveMessage;\n  } catch {\n    ws.send(\n      JSON.stringify({\n        error: { code: 400, message: \"Malformed JSON\", status: \"INVALID_ARGUMENT\" },\n      }),\n    );\n    return;\n  }\n\n  // Handle setup message\n  if (parsed.setup) {\n    session.setupDone = true;\n    session.model = parsed.setup.model ?? defaults.model;\n    session.tools = convertTools(parsed.setup.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: 400, 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: 400,\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: 400,\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: 400,\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  };\n\n  const fixture = matchFixture(fixtures, completionReq, journal.fixtureMatchCounts);\n  const path = WS_PATH;\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures);\n  }\n\n  if (!fixture) {\n    if (defaults.strict) {\n      defaults.logger.warn(`STRICT: No fixture matched for WebSocket message`);\n      ws.close(1008, \"Strict mode: no fixture matched\");\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path,\n      headers: {},\n      body: completionReq,\n      response: { status: 404, fixture: null },\n    });\n    ws.send(\n      JSON.stringify({\n        error: { code: 404, 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 = fixture.response;\n  const latency = fixture.latency ?? defaults.latency;\n  const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);\n\n  // Error response\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method: \"WS\",\n      path,\n      headers: {},\n      body: completionReq,\n      response: { status, fixture },\n    });\n    ws.send(\n      JSON.stringify({\n        error: { code: status, message: response.error.message, status: \"ERROR\" },\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: {},\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      ws.send(\n        JSON.stringify({\n          serverContent: {\n            modelTurn: { parts: [{ text: \"\" }] },\n            turnComplete: true,\n          },\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    let interrupted = false;\n\n    for (let i = 0; i < chunks.length; i++) {\n      if (ws.isClosed) break;\n      if (latency > 0) await delay(latency, interruption?.signal);\n      if (interruption?.signal.aborted) {\n        interrupted = true;\n        break;\n      }\n      if (ws.isClosed) break;\n\n      const isLast = i === chunks.length - 1;\n      ws.send(\n        JSON.stringify({\n          serverContent: {\n            modelTurn: { parts: [{ text: chunks[i] }] },\n            turnComplete: isLast,\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    // 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: {},\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n\n    const interruption = createInterruptionSignal(fixture);\n\n    if (ws.isClosed) {\n      interruption?.cleanup();\n      return;\n    }\n    if (latency > 0) await delay(latency, 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    // 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: {},\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  ws.send(\n    JSON.stringify({\n      error: {\n        code: 500,\n        message: \"Fixture response did not match any known type\",\n        status: \"INTERNAL\",\n      },\n    }),\n  );\n}\n"],"mappings":";;;;;;AA8EA,MAAM,UAAU;;;;AAKhB,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;GAClE,MAAM,YAAY,KAAK,MAAM,QAAQ,MAAM,EAAE,SAAS,OAAU;AAEhE,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,OAAU;AAEhE,OAAI,UAAU,SAAS,EACrB,UAAS,KAAK;IACZ,MAAM;IACN,SAAS;IACT,YAAY,UAAU,KAAK,GAAG,OAAO;KACnC,IAAI,eAAe,EAAE,aAAc,KAAK,GAAG;KAC3C,MAAM;KACN,UAAU;MACR,MAAM,EAAE,aAAc;MACtB,WAAW,KAAK,UAAU,EAAE,aAAc,KAAK;MAChD;KACF,EAAE;IACJ,CAAC;QACG;IACL,MAAM,OAAO,UAAU,KAAK,MAAM,EAAE,KAAM,CAAC,KAAK,GAAG;AACnD,aAAS,KAAK;KAAE,MAAM;KAAa,SAAS;KAAM,CAAC;;;;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,UACM;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;KAAK,SAAS;KAAK,QAAQ;KAAY,EACvD,CAAC,CACH;WACK;IAGR,CACH;GACD;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UACA,SACe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;SAClB;AACN,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAK,SAAS;GAAkB,QAAQ;GAAoB,EAC5E,CAAC,CACH;AACD;;AAIF,KAAI,OAAO,OAAO;AAChB,UAAQ,YAAY;AACpB,UAAQ,QAAQ,OAAO,MAAM,SAAS,SAAS;AAC/C,UAAQ,QAAQ,aAAa,OAAO,MAAM,MAAM;AAChD,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;GAAK,SAAS;GAAkB,QAAQ;GAAuB,EAC/E,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;EACnD;CAED,MAAM,UAAUA,4BAAa,UAAU,eAAe,QAAQ,mBAAmB;CACjF,MAAM,OAAO;AAEb,KAAI,QACF,SAAQ,2BAA2B,SAAS,SAAS;AAGvD,KAAI,CAAC,SAAS;AACZ,MAAI,SAAS,QAAQ;AACnB,YAAS,OAAO,KAAK,mDAAmD;AACxE,MAAG,MAAM,MAAM,kCAAkC;AACjD;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR;GACA,SAAS,EAAE;GACX,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAK,SAAS;GAAsB,QAAQ;GAAa,EACzE,CAAC,CACH;AACD;;AAIF,SAAQ,oBAAoB,KAAK,GAAG,YAAY;CAEhD,MAAM,WAAW,QAAQ;CACzB,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,SAAS,EAAE;GACX,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,KAAG,KACD,KAAK,UAAU,EACb,OAAO;GAAE,MAAM;GAAQ,SAAS,SAAS,MAAM;GAAS,QAAQ;GAAS,EAC1E,CAAC,CACH;AACD;;AAIF,KAAIC,+BAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAAS,EAAE;GACX,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,UAAU,SAAS;AAEzB,MAAI,QAAQ,WAAW,GAAG;AACxB,OAAI,GAAG,SAAU;AACjB,MAAG,KACD,KAAK,UAAU,EACb,eAAe;IACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,EAAE;IACpC,cAAc;IACf,EACF,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,eAAeC,8CAAyB,QAAQ;EACtD,IAAI,cAAc;AAElB,OAAK,IAAI,IAAI,GAAG,IAAI,OAAO,QAAQ,KAAK;AACtC,OAAI,GAAG,SAAU;AACjB,OAAI,UAAU,EAAG,OAAMC,yBAAM,SAAS,cAAc,OAAO;AAC3D,OAAI,cAAc,OAAO,SAAS;AAChC,kBAAc;AACd;;AAEF,OAAI,GAAG,SAAU;GAEjB,MAAM,SAAS,MAAM,OAAO,SAAS;AACrC,MAAG,KACD,KAAK,UAAU,EACb,eAAe;IACb,WAAW,EAAE,OAAO,CAAC,EAAE,MAAM,OAAO,IAAI,CAAC,EAAE;IAC3C,cAAc;IACf,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,UAAQ,oBAAoB,KAAK;GAAE,MAAM;GAAa;GAAS,CAAC;AAChE;;AAIF,KAAIC,mCAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR;GACA,SAAS,EAAE;GACX,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EAEF,MAAM,eAAeF,8CAAyB,QAAQ;AAEtD,MAAI,GAAG,UAAU;AACf,iBAAc,SAAS;AACvB;;AAEF,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,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;EACX,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.d.cts

@@ -0,0 +1,18 @@
+import { Journal } from "./journal.cjs";
+import { Logger } from "./logger.cjs";
+import { Fixture } from "./types.cjs";
+import { WebSocketConnection } from "./ws-framing.cjs";
+
+//#region src/ws-gemini-live.d.ts
+
+declare function handleWebSocketGeminiLive(ws: WebSocketConnection, fixtures: Fixture[], journal: Journal, defaults: {
+  latency: number;
+  chunkSize: number;
+  model: string;
+  logger: Logger;
+  strict?: boolean;
+}): void;
+//# sourceMappingURL=ws-gemini-live.d.ts.map
+//#endregion
+export { handleWebSocketGeminiLive };
+//# sourceMappingURL=ws-gemini-live.d.cts.map

package/dist/ws-gemini-live.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-gemini-live.d.cts","names":[],"sources":["../src/ws-gemini-live.ts"],"sourcesContent":[],"mappings":";;;;;;;AA6KyE,iBAJzD,yBAAA,CAIyD,EAAA,EAHnE,mBAGmE,EAAA,QAAA,EAF7D,OAE6D,EAAA,EAAA,OAAA,EAD9D,OAC8D,EAAA,QAAA,EAAA;EAAM,OAAA,EAAA,MAAA;;;UAAN"}

package/dist/ws-gemini-live.d.ts

@@ -0,0 +1,18 @@
+import { Journal } from "./journal.js";
+import { Logger } from "./logger.js";
+import { Fixture } from "./types.js";
+import { WebSocketConnection } from "./ws-framing.js";
+
+//#region src/ws-gemini-live.d.ts
+
+declare function handleWebSocketGeminiLive(ws: WebSocketConnection, fixtures: Fixture[], journal: Journal, defaults: {
+  latency: number;
+  chunkSize: number;
+  model: string;
+  logger: Logger;
+  strict?: boolean;
+}): void;
+//# sourceMappingURL=ws-gemini-live.d.ts.map
+//#endregion
+export { handleWebSocketGeminiLive };
+//# sourceMappingURL=ws-gemini-live.d.ts.map

package/dist/ws-gemini-live.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-gemini-live.d.ts","names":[],"sources":["../src/ws-gemini-live.ts"],"sourcesContent":[],"mappings":";;;;;;;AA6KyE,iBAJzD,yBAAA,CAIyD,EAAA,EAHnE,mBAGmE,EAAA,QAAA,EAF7D,OAE6D,EAAA,EAAA,OAAA,EAD9D,OAC8D,EAAA,QAAA,EAAA;EAAM,OAAA,EAAA,MAAA;;;UAAN"}