@copilotkit/aimock 1.16.4 → 1.17.0

package/dist/server.js

@@ -1,4 +1,5 @@
-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";
@@ -11,11 +12,14 @@ 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 { handleOllama, handleOllamaGenerate } from "./ollama.js";
 import { handleCohere } from "./cohere.js";
 import { handleSearch } from "./search.js";
@@ -25,9 +29,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 +49,11 @@ 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 DEFAULT_CHUNK_SIZE = 20;
 const COMPAT_SUFFIXES = [
 	"/chat/completions",
@@ -71,6 +79,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 +208,7 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		fixtures.length = 0;
 		journal.clear();
 		videoStates.clear();
+		falJobs.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 }));
@@ -392,6 +402,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 +586,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}`);
@@ -857,6 +884,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 +1091,128 @@ 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);
+				if (applyChaos(res, null, defaults.chaos, req.headers, journal, {
+					method: req.method ?? "POST",
+					path: pathname,
+					headers: flattenHeaders(req.headers),
+					body: {
+						model: "",
+						messages: []
+					}
+				}, defaults.registry, defaults.logger)) 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);
+				if (applyChaos(res, null, defaults.chaos, req.headers, journal, {
+					method: req.method ?? "POST",
+					path: pathname,
+					headers: flattenHeaders(req.headers),
+					body: {
+						model: "",
+						messages: []
+					}
+				}, defaults.registry, defaults.logger)) 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 (pathname.match(FAL_QUEUE_SUBMIT_RE) && req.method === "POST") {
+			setCorsHeaders(res);
+			try {
+				const raw = await readBody(req);
+				if (applyChaos(res, null, defaults.chaos, req.headers, journal, {
+					method: req.method ?? "POST",
+					path: pathname,
+					headers: flattenHeaders(req.headers),
+					body: {
+						model: "",
+						messages: []
+					}
+				}, defaults.registry, defaults.logger)) 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) : "{}";
+				if (applyChaos(res, null, defaults.chaos, req.headers, journal, {
+					method: req.method ?? "GET",
+					path: pathname,
+					headers: flattenHeaders(req.headers),
+					body: {
+						model: "",
+						messages: []
+					}
+				}, defaults.registry, defaults.logger)) 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);
+				if (applyChaos(res, null, defaults.chaos, req.headers, journal, {
+					method: req.method ?? "POST",
+					path: pathname,
+					headers: flattenHeaders(req.headers),
+					body: {
+						model: "",
+						messages: []
+					}
+				}, defaults.registry, defaults.logger)) 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;