@copilotkit/aimock 1.22.0 → 1.22.1

package/dist/server.js

@@ -1,4 +1,4 @@
-import { buildContentWithToolCallsChunks, buildContentWithToolCallsCompletion, buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, extractOverrides, flattenHeaders, getTestId, isAudioResponse, isContentWithToolCallsResponse, isErrorResponse, isTextResponse, isToolCallResponse, resolveResponse, resolveStrictMode, strictOverrideField } from "./helpers.js";
+import { buildContentWithToolCallsChunks, buildContentWithToolCallsCompletion, buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, extractOverrides, flattenHeaders, getTestId, isAudioResponse, isContentWithToolCallsResponse, isErrorResponse, isTextResponse, isToolCallResponse, readBody, resolveResponse, resolveStrictMode, serializeErrorResponse, strictOverrideField } from "./helpers.js";
 import { Logger } from "./logger.js";
 import { Journal } from "./journal.js";
 import { matchFixture } from "./router.js";
@@ -12,7 +12,7 @@ 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 { handleGeminiInteractions, resetEventIdCounter, resetInteractionCounter } from "./gemini-interactions.js";
 import { handleEmbeddings } from "./embeddings.js";
 import { handleImages } from "./images.js";
 import { handleSpeech } from "./speech.js";
@@ -111,21 +111,6 @@ const CORS_HEADERS = {
 function setCorsHeaders(res) {
 	for (const [key, value] of Object.entries(CORS_HEADERS)) res.setHeader(key, value);
 }
-const DEFAULT_MAX_BODY_BYTES = 10 * 1024 * 1024;
-async function readBody(req, maxBytes = DEFAULT_MAX_BODY_BYTES) {
-	const buffers = [];
-	let totalBytes = 0;
-	for await (const chunk of req) {
-		const buf = chunk;
-		totalBytes += buf.length;
-		if (totalBytes > maxBytes) {
-			req.destroy();
-			throw new Error(`Request body exceeded size limit of ${maxBytes} bytes`);
-		}
-		buffers.push(buf);
-	}
-	return Buffer.concat(buffers).toString();
-}
 function handleOptions(res) {
 	setCorsHeaders(res);
 	res.writeHead(204);
@@ -212,6 +197,8 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		videoStates.clear();
 		falJobs.clear();
 		falQueueStates.clear();
+		resetInteractionCounter();
+		resetEventIdCounter();
 		if (defaults.registry) defaults.registry.setGauge("aimock_fixtures_loaded", {}, fixtures.length);
 		res.writeHead(200, { "Content-Type": "application/json" });
 		res.end(JSON.stringify({ reset: true }));
@@ -369,6 +356,29 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 		return;
 	}
 	if (!fixture) {
+		if (resolveStrictMode(defaults.strict, req.headers)) {
+			const strictStatus = 503;
+			const strictMessage = "Strict mode: no fixture matched";
+			defaults.logger.error(`STRICT: No fixture matched for ${req.method ?? "POST"} ${req.url ?? COMPLETIONS_PATH}`);
+			journal.add({
+				method: req.method ?? "POST",
+				path: req.url ?? COMPLETIONS_PATH,
+				headers: flattenHeaders(req.headers),
+				body,
+				response: {
+					status: strictStatus,
+					fixture: null,
+					...strictOverrideField(defaults.strict, req.headers)
+				}
+			});
+			writeErrorResponse(res, strictStatus, JSON.stringify({ error: {
+				message: strictMessage,
+				type: "invalid_request_error",
+				param: null,
+				code: "no_fixture_match"
+			} }));
+			return;
+		}
 		if (defaults.record && providerKey) {
 			const hookOptions = chaosAction === "malformed" ? {
 				beforeWriteResponse: () => {
@@ -386,7 +396,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 			} : 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") {
+			if (outcome !== "not_configured") {
 				journal.add({
 					method: req.method ?? "POST",
 					path: req.url ?? COMPLETIONS_PATH,
@@ -401,23 +411,19 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 				return;
 			}
 		}
-		const effectiveStrict = resolveStrictMode(defaults.strict, req.headers);
-		const strictStatus = effectiveStrict ? 503 : 404;
-		const strictMessage = effectiveStrict ? "Strict mode: no fixture matched" : "No fixture matched";
-		if (effectiveStrict) defaults.logger.error(`STRICT: No fixture matched for ${req.method ?? "POST"} ${req.url ?? COMPLETIONS_PATH}`);
 		journal.add({
 			method: req.method ?? "POST",
 			path: req.url ?? COMPLETIONS_PATH,
 			headers: flattenHeaders(req.headers),
 			body,
 			response: {
-				status: strictStatus,
+				status: 404,
 				fixture: null,
 				...strictOverrideField(defaults.strict, req.headers)
 			}
 		});
-		writeErrorResponse(res, strictStatus, JSON.stringify({ error: {
-			message: strictMessage,
+		writeErrorResponse(res, 404, JSON.stringify({ error: {
+			message: "No fixture matched",
 			type: "invalid_request_error",
 			param: null,
 			code: "no_fixture_match"
@@ -439,13 +445,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 				fixture
 			}
 		});
-		const errorBody = { error: {
-			message: response.error.message,
-			type: response.error.type ?? "server_error",
-			param: null,
-			code: response.error.code ?? null
-		} };
-		writeErrorResponse(res, status, JSON.stringify(errorBody));
+		writeErrorResponse(res, status, serializeErrorResponse(response));
 		return;
 	}
 	if (isAudioResponse(response)) {
@@ -534,6 +534,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 		return;
 	}
 	if (isToolCallResponse(response)) {
+		if (response.webSearches?.length) defaults.logger.warn("webSearches in fixture response are not supported for Chat Completions API — ignoring");
 		const overrides = extractOverrides(response);
 		const journalEntry = journal.add({
 			method: req.method ?? "POST",
@@ -1130,35 +1131,7 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 		if (pathname === ELEVENLABS_SOUND_GENERATION_PATH && req.method === "POST") {
 			setCorsHeaders(res);
 			try {
-				const raw = await readBody(req);
-				let elSoundFixture = null;
-				try {
-					const parsed = JSON.parse(raw);
-					const syntheticReq = {
-						model: parsed.model_id ?? "eleven_text_to_sound_v2",
-						messages: [{
-							role: "user",
-							content: parsed.text ?? ""
-						}],
-						_endpointType: "audio-gen"
-					};
-					const testId = getTestId(req);
-					elSoundFixture = matchFixture(fixtures, syntheticReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
-				} catch {}
-				const chaosAction = evaluateChaos(elSoundFixture, defaults.chaos, req.headers, defaults.logger);
-				if (chaosAction) {
-					applyChaosAction(chaosAction, res, elSoundFixture, 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");
+				await handleElevenLabsAudio(req, res, await readBody(req), 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: {
@@ -1174,36 +1147,7 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			setCorsHeaders(res);
 			const musicSubType = musicMatch[1] ?? "music";
 			try {
-				const raw = await readBody(req);
-				let elMusicFixture = null;
-				try {
-					const parsed = JSON.parse(raw);
-					const prompt = (typeof parsed.prompt === "string" ? parsed.prompt : null) ?? (parsed.composition_plan != null ? typeof parsed.composition_plan === "string" ? parsed.composition_plan : JSON.stringify(parsed.composition_plan) : "");
-					const syntheticReq = {
-						model: parsed.model_id ?? "music_v1",
-						messages: [{
-							role: "user",
-							content: prompt
-						}],
-						_endpointType: "audio-gen"
-					};
-					const testId = getTestId(req);
-					elMusicFixture = matchFixture(fixtures, syntheticReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
-				} catch {}
-				const chaosAction = evaluateChaos(elMusicFixture, defaults.chaos, req.headers, defaults.logger);
-				if (chaosAction) {
-					applyChaosAction(chaosAction, res, elMusicFixture, 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);
+				await handleElevenLabsAudio(req, res, await readBody(req), fixtures, defaults, journal, musicSubType);
 			} catch (err) {
 				const msg = err instanceof Error ? err.message : "Internal error";
 				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
@@ -1244,40 +1188,10 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 				return;
 			}
 		}
-		const falQueueSubmitMatch = pathname.match(FAL_QUEUE_SUBMIT_RE);
-		if (falQueueSubmitMatch && req.method === "POST") {
+		if (pathname.match(FAL_QUEUE_SUBMIT_RE) && req.method === "POST") {
 			setCorsHeaders(res);
 			try {
-				const raw = falBody ?? await readBody(req);
-				let falSubmitFixture = null;
-				try {
-					const parsed = raw.trim() ? JSON.parse(raw) : {};
-					const prompt = (typeof parsed.prompt === "string" ? parsed.prompt : null) ?? (typeof parsed.text === "string" ? parsed.text : null) ?? "";
-					const syntheticReq = {
-						model: falQueueSubmitMatch[1],
-						messages: [{
-							role: "user",
-							content: prompt
-						}],
-						_endpointType: "fal-audio"
-					};
-					const testId = getTestId(req);
-					falSubmitFixture = matchFixture(fixtures, syntheticReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
-				} catch {}
-				const chaosAction = evaluateChaos(falSubmitFixture, defaults.chaos, req.headers, defaults.logger);
-				if (chaosAction) {
-					applyChaosAction(chaosAction, res, falSubmitFixture, 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);
+				await handleFalQueue(req, res, falBody ?? await readBody(req), pathname, fixtures, defaults, journal);
 			} catch (err) {
 				const msg = err instanceof Error ? err.message : "Internal error";
 				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
@@ -1316,40 +1230,10 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 			return;
 		}
-		const falRunMatch = pathname.match(FAL_RUN_RE);
-		if (falRunMatch && req.method === "POST") {
+		if (pathname.match(FAL_RUN_RE) && req.method === "POST") {
 			setCorsHeaders(res);
 			try {
-				const raw = falBody ?? await readBody(req);
-				let falRunFixture = null;
-				try {
-					const parsed = raw.trim() ? JSON.parse(raw) : {};
-					const prompt = (typeof parsed.prompt === "string" ? parsed.prompt : null) ?? (typeof parsed.text === "string" ? parsed.text : null) ?? "";
-					const syntheticReq = {
-						model: falRunMatch[1],
-						messages: [{
-							role: "user",
-							content: prompt
-						}],
-						_endpointType: "fal-audio"
-					};
-					const testId = getTestId(req);
-					falRunFixture = matchFixture(fixtures, syntheticReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
-				} catch {}
-				const chaosAction = evaluateChaos(falRunFixture, defaults.chaos, req.headers, defaults.logger);
-				if (chaosAction) {
-					applyChaosAction(chaosAction, res, falRunFixture, 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);
+				await handleFalQueue(req, res, falBody ?? await readBody(req), pathname, fixtures, defaults, journal);
 			} catch (err) {
 				const msg = err instanceof Error ? err.message : "Internal error";
 				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {