@copilotkit/aimock 1.16.4 → 1.18.0

package/dist/server.js

@@ -1,21 +1,26 @@
-import { buildContentWithToolCallsChunks, buildContentWithToolCallsCompletion, buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, extractOverrides, flattenHeaders, getTestId, isContentWithToolCallsResponse, isErrorResponse, isTextResponse, isToolCallResponse } from "./helpers.js";
+import { buildContentWithToolCallsChunks, buildContentWithToolCallsCompletion, buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, extractOverrides, flattenHeaders, getTestId, isAudioResponse, isContentWithToolCallsResponse, isErrorResponse, isTextResponse, isToolCallResponse } from "./helpers.js";
+import { Logger } from "./logger.js";
 import { Journal } from "./journal.js";
 import { matchFixture } from "./router.js";
 import { entryToFixture, validateFixtures } from "./fixture-loader.js";
 import { writeErrorResponse, writeSSEStream } from "./sse-writer.js";
 import { createInterruptionSignal } from "./interruption.js";
-import { applyChaos } from "./chaos.js";
+import { applyChaosAction, evaluateChaos } from "./chaos.js";
 import { proxyAndRecord } from "./recorder.js";
 import { handleResponses } from "./responses.js";
 import { handleMessages } from "./messages.js";
 import { handleGemini } from "./gemini.js";
 import { handleBedrock, handleBedrockStream } from "./bedrock.js";
 import { handleConverse, handleConverseStream } from "./bedrock-converse.js";
+import { handleGeminiInteractions } from "./gemini-interactions.js";
 import { handleEmbeddings } from "./embeddings.js";
 import { handleImages } from "./images.js";
 import { handleSpeech } from "./speech.js";
 import { handleTranscription } from "./transcription.js";
 import { VideoStateMap, handleVideoCreate, handleVideoStatus } from "./video.js";
+import { handleElevenLabsAudio } from "./elevenlabs-audio.js";
+import { falJobs, handleFalQueue } from "./fal-audio.js";
+import { falQueueStates, handleFal } from "./fal.js";
 import { handleOllama, handleOllamaGenerate } from "./ollama.js";
 import { handleCohere } from "./cohere.js";
 import { handleSearch } from "./search.js";
@@ -25,9 +30,8 @@ import { upgradeToWebSocket } from "./ws-framing.js";
 import { handleWebSocketResponses } from "./ws-responses.js";
 import { handleWebSocketRealtime } from "./ws-realtime.js";
 import { handleWebSocketGeminiLive } from "./ws-gemini-live.js";
-import { Logger } from "./logger.js";
 import { createMetricsRegistry, normalizePathLabel } from "./metrics.js";
-import * as http from "node:http";
+import * as http$1 from "node:http";
 
 //#region src/server.ts
 const COMPLETIONS_PATH = "/v1/chat/completions";
@@ -46,6 +50,12 @@ const TRANSCRIPTIONS_PATH = "/v1/audio/transcriptions";
 const VIDEOS_PATH = "/v1/videos";
 const VIDEOS_STATUS_RE = /^\/v1\/videos\/([^/]+)$/;
 const GEMINI_PREDICT_RE = /^\/v1beta\/models\/([^:]+):predict$/;
+const ELEVENLABS_SOUND_GENERATION_PATH = "/v1/sound-generation";
+const ELEVENLABS_MUSIC_RE = /^\/v1\/music(?:\/(.+))?$/;
+const FAL_QUEUE_SUBMIT_RE = /^\/fal\/queue\/submit\/(.+)$/;
+const FAL_QUEUE_REQUESTS_RE = /^\/fal\/queue\/requests\/(.+)$/;
+const FAL_RUN_RE = /^\/fal\/run\/(.+)$/;
+const FAL_PREFIX_RE = /^\/fal(?:\/.*)?$/;
 const DEFAULT_CHUNK_SIZE = 20;
 const COMPAT_SUFFIXES = [
 	"/chat/completions",
@@ -71,6 +81,7 @@ function normalizeCompatPath(pathname, logger) {
 	}
 	return pathname;
 }
+const GEMINI_INTERACTIONS_PATH = "/v1beta/interactions";
 const GEMINI_PATH_RE = /^\/v1beta\/models\/([^:]+):(generateContent|streamGenerateContent)$/;
 const AZURE_DEPLOYMENT_RE = /^\/openai\/deployments\/([^/]+)\/(chat\/completions|embeddings)$/;
 const BEDROCK_INVOKE_RE = /^\/model\/([^/]+)\/invoke$/;
@@ -199,6 +210,8 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		fixtures.length = 0;
 		journal.clear();
 		videoStates.clear();
+		falJobs.clear();
+		falQueueStates.clear();
 		if (defaults.registry) defaults.registry.setGauge("aimock_fixtures_loaded", {}, fixtures.length);
 		res.writeHead(200, { "Content-Type": "application/json" });
 		res.end(JSON.stringify({ reset: true }));
@@ -317,6 +330,9 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 		} }));
 		return;
 	}
+	const method = req.method ?? "POST";
+	const path = req.url ?? COMPLETIONS_PATH;
+	const flatHeaders = flattenHeaders(req.headers);
 	body._endpointType = "chat";
 	const testId = getTestId(req);
 	const fixture = matchFixture(fixtures, body, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
@@ -328,18 +344,40 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 		const snippet = typeof lastUserMsg?.content === "string" ? lastUserMsg.content.slice(0, 80) : "";
 		defaults.logger.debug(`No fixture matched for request (model=${body.model ?? "?"}, msg="${snippet}")`);
 	}
-	const method = req.method ?? "POST";
-	const path = req.url ?? COMPLETIONS_PATH;
-	const flatHeaders = flattenHeaders(req.headers);
-	if (applyChaos(res, fixture, defaults.chaos, req.headers, journal, {
+	const chaosAction = evaluateChaos(fixture, defaults.chaos, req.headers, defaults.logger);
+	const chaosContext = {
 		method,
 		path,
 		headers: flatHeaders,
 		body
-	}, defaults.registry, defaults.logger)) return;
+	};
+	if (chaosAction === "drop" || chaosAction === "disconnect") {
+		applyChaosAction(chaosAction, res, fixture, journal, chaosContext, fixture ? "fixture" : "proxy", defaults.registry);
+		return;
+	}
+	if (fixture && chaosAction === "malformed") {
+		applyChaosAction(chaosAction, res, fixture, journal, chaosContext, "fixture", defaults.registry);
+		return;
+	}
 	if (!fixture) {
 		if (defaults.record && providerKey) {
-			if (await proxyAndRecord(req, res, body, providerKey, req.url ?? COMPLETIONS_PATH, fixtures, defaults, raw)) {
+			const hookOptions = chaosAction === "malformed" ? {
+				beforeWriteResponse: () => {
+					applyChaosAction(chaosAction, res, null, journal, chaosContext, "proxy", defaults.registry);
+					return true;
+				},
+				onHookBypassed: (reason) => {
+					defaults.logger.warn(`[chaos] malformed bypassed on proxy: upstream returned SSE (${reason})`);
+					defaults.registry?.incrementCounter("aimock_chaos_bypassed_total", {
+						action: "malformed",
+						source: "proxy",
+						reason
+					});
+				}
+			} : void 0;
+			const outcome = await proxyAndRecord(req, res, body, providerKey, req.url ?? COMPLETIONS_PATH, fixtures, defaults, raw, hookOptions);
+			if (outcome === "handled_by_hook") return;
+			if (outcome === "relayed") {
 				journal.add({
 					method: req.method ?? "POST",
 					path: req.url ?? COMPLETIONS_PATH,
@@ -392,6 +430,23 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 		writeErrorResponse(res, status, JSON.stringify(response));
 		return;
 	}
+	if (isAudioResponse(response)) {
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? COMPLETIONS_PATH,
+			headers: flattenHeaders(req.headers),
+			body,
+			response: {
+				status: 422,
+				fixture
+			}
+		});
+		writeErrorResponse(res, 422, JSON.stringify({ error: {
+			message: "Audio responses are not supported on the chat completions endpoint. Use Gemini generateContent or a dedicated audio endpoint.",
+			type: "invalid_request_error"
+		} }));
+		return;
+	}
 	if (isContentWithToolCallsResponse(response)) {
 		if (response.webSearches?.length) defaults.logger.warn("webSearches in fixture response are not supported for Chat Completions API — ignoring");
 		const overrides = extractOverrides(response);
@@ -559,7 +614,7 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 		if (registry && handler.setRegistry) handler.setRegistry(registry);
 	}
 	if (registry) registry.setGauge("aimock_fixtures_loaded", {}, fixtures.length);
-	const server = http.createServer((req, res) => {
+	const server = http$1.createServer((req, res) => {
 		handleHttpRequest(req, res).catch((err) => {
 			const msg = err instanceof Error ? err.message : "Internal error";
 			defaults.logger.warn(`Unhandled request error: ${msg}`);
@@ -839,7 +894,7 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 		const videoStatusMatch = pathname.match(VIDEOS_STATUS_RE);
 		if (videoStatusMatch && req.method === "GET") {
 			const videoId = videoStatusMatch[1];
-			handleVideoStatus(req, res, videoId, journal, setCorsHeaders, videoStates);
+			handleVideoStatus(req, res, videoId, journal, defaults, setCorsHeaders, videoStates);
 			return;
 		}
 		const geminiPredictMatch = pathname.match(GEMINI_PREDICT_RE);
@@ -857,6 +912,26 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 			return;
 		}
+		if (pathname === GEMINI_INTERACTIONS_PATH && req.method === "POST") {
+			try {
+				await handleGeminiInteractions(req, res, await readBody(req), fixtures, journal, defaults, setCorsHeaders);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) {
+					try {
+						res.write(`data: ${JSON.stringify({ error: { message: msg } })}\n\n`);
+					} catch (writeErr) {
+						logger.debug("Failed to write error recovery response:", writeErr);
+					}
+					res.end();
+				}
+			}
+			return;
+		}
 		const geminiMatch = pathname.match(GEMINI_PATH_RE);
 		if (geminiMatch && req.method === "POST") {
 			const geminiModel = geminiMatch[1];
@@ -1044,6 +1119,176 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 			return;
 		}
+		if (pathname === ELEVENLABS_SOUND_GENERATION_PATH && req.method === "POST") {
+			setCorsHeaders(res);
+			try {
+				const raw = await readBody(req);
+				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "POST",
+						path: pathname,
+						headers: flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
+				await handleElevenLabsAudio(req, res, raw, fixtures, defaults, journal, "sound-generation");
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
+		const musicMatch = pathname.match(ELEVENLABS_MUSIC_RE);
+		if (musicMatch && req.method === "POST") {
+			setCorsHeaders(res);
+			const musicSubType = musicMatch[1] ?? "music";
+			try {
+				const raw = await readBody(req);
+				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "POST",
+						path: pathname,
+						headers: flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
+				await handleElevenLabsAudio(req, res, raw, fixtures, defaults, journal, musicSubType);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
+		if (FAL_PREFIX_RE.test(pathname) && req.headers["x-fal-target-host"]) {
+			setCorsHeaders(res);
+			try {
+				const raw = req.method === "POST" || req.method === "PUT" ? await readBody(req) : "";
+				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "GET",
+						path: pathname,
+						headers: flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
+				if (await handleFal(req, res, raw, pathname, fixtures, defaults, journal) === "handled") return;
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+				return;
+			}
+		}
+		if (pathname.match(FAL_QUEUE_SUBMIT_RE) && req.method === "POST") {
+			setCorsHeaders(res);
+			try {
+				const raw = await readBody(req);
+				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "POST",
+						path: pathname,
+						headers: flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
+				await handleFalQueue(req, res, raw, pathname, fixtures, defaults, journal);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
+		if (pathname.match(FAL_QUEUE_REQUESTS_RE) && (req.method === "GET" || req.method === "POST" || req.method === "PUT")) {
+			setCorsHeaders(res);
+			try {
+				const raw = req.method === "POST" ? await readBody(req) : "{}";
+				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "GET",
+						path: pathname,
+						headers: flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
+				await handleFalQueue(req, res, raw, pathname, fixtures, defaults, journal);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
+		if (pathname.match(FAL_RUN_RE) && req.method === "POST") {
+			setCorsHeaders(res);
+			try {
+				const raw = await readBody(req);
+				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "POST",
+						path: pathname,
+						headers: flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
+				await handleFalQueue(req, res, raw, pathname, fixtures, defaults, journal);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
 		if (pathname !== COMPLETIONS_PATH) {
 			handleNotFound(res, "Not found");
 			return;