@copilotkit/aimock 1.21.0 → 1.22.0

package/dist/server.js

@@ -1,4 +1,4 @@
-import { buildContentWithToolCallsChunks, buildContentWithToolCallsCompletion, buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, extractOverrides, flattenHeaders, getTestId, isAudioResponse, isContentWithToolCallsResponse, isErrorResponse, isTextResponse, isToolCallResponse, resolveResponse } from "./helpers.js";
+import { buildContentWithToolCallsChunks, buildContentWithToolCallsCompletion, buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, extractOverrides, flattenHeaders, getTestId, isAudioResponse, isContentWithToolCallsResponse, isErrorResponse, isTextResponse, isToolCallResponse, resolveResponse, resolveStrictMode, strictOverrideField } from "./helpers.js";
 import { Logger } from "./logger.js";
 import { Journal } from "./journal.js";
 import { matchFixture } from "./router.js";
@@ -252,12 +252,17 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 			}
 		};
 		fixtures.unshift(errorFixture);
+		let consumed = false;
 		const original = errorFixture.match.predicate;
 		errorFixture.match.predicate = (req) => {
+			if (consumed) return false;
 			const result = original(req);
 			if (result) {
-				const idx = fixtures.indexOf(errorFixture);
-				if (idx !== -1) fixtures.splice(idx, 1);
+				consumed = true;
+				queueMicrotask(() => {
+					const idx = fixtures.indexOf(errorFixture);
+					if (idx !== -1) fixtures.splice(idx, 1);
+				});
 			}
 			return result;
 		};
@@ -295,7 +300,8 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 	try {
 		body = JSON.parse(raw);
 		if (modelFallback && !body.model) body.model = modelFallback;
-	} catch {
+	} catch (parseErr) {
+		const detail = parseErr instanceof Error ? parseErr.message : "unknown parse error";
 		journal.add({
 			method: req.method ?? "POST",
 			path: req.url ?? COMPLETIONS_PATH,
@@ -307,7 +313,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 			}
 		});
 		writeErrorResponse(res, 400, JSON.stringify({ error: {
-			message: "Malformed JSON",
+			message: `Malformed JSON: ${detail}`,
 			type: "invalid_request_error",
 			param: null,
 			code: "invalid_json"
@@ -370,7 +376,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 					return true;
 				},
 				onHookBypassed: (reason) => {
-					defaults.logger.warn(`[chaos] malformed bypassed on proxy: upstream returned SSE (${reason})`);
+					defaults.logger.warn(`[chaos] malformed bypassed on proxy: upstream returned streaming response (${reason})`);
 					defaults.registry?.incrementCounter("aimock_chaos_bypassed_total", {
 						action: "malformed",
 						source: "proxy",
@@ -395,9 +401,10 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 				return;
 			}
 		}
-		const strictStatus = defaults.strict ? 503 : 404;
-		const strictMessage = defaults.strict ? "Strict mode: no fixture matched" : "No fixture matched";
-		if (defaults.strict) defaults.logger.error(`STRICT: No fixture matched for ${req.method ?? "POST"} ${req.url ?? COMPLETIONS_PATH}`);
+		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,
@@ -405,7 +412,8 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 			body,
 			response: {
 				status: strictStatus,
-				fixture: null
+				fixture: null,
+				...strictOverrideField(defaults.strict, req.headers)
 			}
 		});
 		writeErrorResponse(res, strictStatus, JSON.stringify({ error: {
@@ -634,7 +642,7 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 					message: msg,
 					type: "server_error"
 				} }));
-			}
+			} else if (!res.writableEnded) res.end();
 		});
 	});
 	async function handleHttpRequest(req, res) {
@@ -775,13 +783,11 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 					message: msg,
 					type: "server_error"
 				} }));
-				else if (!res.writableEnded) {
-					try {
-						res.write(`event: error\ndata: ${JSON.stringify({ error: { message: msg } })}\n\n`);
-					} catch (writeErr) {
-						logger.debug("Failed to write error recovery response:", writeErr);
-					}
+				else if (!res.writableEnded) try {
+					res.write(`event: error\ndata: ${JSON.stringify({ error: { message: msg } })}\n\n`);
 					res.end();
+				} catch (writeErr) {
+					logger.debug("Failed to write error recovery response:", writeErr);
 				}
 			}
 			return;
@@ -795,13 +801,11 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 					message: msg,
 					type: "server_error"
 				} }));
-				else if (!res.writableEnded) {
-					try {
-						res.write(`event: error\ndata: ${JSON.stringify({ error: { message: msg } })}\n\n`);
-					} catch (writeErr) {
-						logger.debug("Failed to write error recovery response:", writeErr);
-					}
+				else if (!res.writableEnded) try {
+					res.write(`event: error\ndata: ${JSON.stringify({ error: { message: msg } })}\n\n`);
 					res.end();
+				} catch (writeErr) {
+					logger.debug("Failed to write error recovery response:", writeErr);
 				}
 			}
 			return;
@@ -815,13 +819,11 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 					message: msg,
 					type: "server_error"
 				} }));
-				else if (!res.writableEnded) {
-					try {
-						res.write(`event: error\ndata: ${JSON.stringify({ error: { message: msg } })}\n\n`);
-					} catch (writeErr) {
-						logger.debug("Failed to write error recovery response:", writeErr);
-					}
+				else if (!res.writableEnded) try {
+					res.write(`event: error\ndata: ${JSON.stringify({ error: { message: msg } })}\n\n`);
 					res.end();
+				} catch (writeErr) {
+					logger.debug("Failed to write error recovery response:", writeErr);
 				}
 			}
 			return;
@@ -837,7 +839,9 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 						parsed.model = deploymentId;
 						raw = JSON.stringify(parsed);
 					}
-				} catch {}
+				} catch (err) {
+					if (!(err instanceof SyntaxError)) defaults.logger.error(`Unexpected error in Azure model injection: ${err instanceof Error ? err.message : String(err)}`);
+				}
 				await handleEmbeddings(req, res, raw, fixtures, journal, defaults, setCorsHeaders, embeddingsProvider);
 			} catch (err) {
 				const msg = err instanceof Error ? err.message : "Internal error";
@@ -931,13 +935,11 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 					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);
-					}
+				else if (!res.writableEnded) try {
+					res.write(`data: ${JSON.stringify({ error: { message: msg } })}\n\n`);
 					res.end();
+				} catch (writeErr) {
+					logger.debug("Failed to write error recovery response:", writeErr);
 				}
 			}
 			return;
@@ -954,13 +956,11 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 					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);
-					}
+				else if (!res.writableEnded) try {
+					res.write(`data: ${JSON.stringify({ error: { message: msg } })}\n\n`);
 					res.end();
+				} catch (writeErr) {
+					logger.debug("Failed to write error recovery response:", writeErr);
 				}
 			}
 			return;
@@ -977,13 +977,11 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 					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);
-					}
+				else if (!res.writableEnded) try {
+					res.write(`data: ${JSON.stringify({ error: { message: msg } })}\n\n`);
 					res.end();
+				} catch (writeErr) {
+					logger.debug("Failed to write error recovery response:", writeErr);
 				}
 			}
 			return;
@@ -1133,9 +1131,23 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			setCorsHeaders(res);
 			try {
 				const raw = await readBody(req);
-				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				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, null, journal, {
+					applyChaosAction(chaosAction, res, elSoundFixture, journal, {
 						method: req.method ?? "POST",
 						path: pathname,
 						headers: flattenHeaders(req.headers),
@@ -1163,9 +1175,24 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			const musicSubType = musicMatch[1] ?? "music";
 			try {
 				const raw = await readBody(req);
-				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				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, null, journal, {
+					applyChaosAction(chaosAction, res, elMusicFixture, journal, {
 						method: req.method ?? "POST",
 						path: pathname,
 						headers: flattenHeaders(req.headers),
@@ -1187,10 +1214,12 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 			return;
 		}
+		let falBody;
 		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) : "";
+				falBody = req.method === "POST" || req.method === "PUT" ? await readBody(req) : "";
+				const raw = falBody;
 				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
 				if (chaosAction) {
 					applyChaosAction(chaosAction, res, null, journal, {
@@ -1215,13 +1244,29 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 				return;
 			}
 		}
-		if (pathname.match(FAL_QUEUE_SUBMIT_RE) && req.method === "POST") {
+		const falQueueSubmitMatch = pathname.match(FAL_QUEUE_SUBMIT_RE);
+		if (falQueueSubmitMatch && req.method === "POST") {
 			setCorsHeaders(res);
 			try {
-				const raw = await readBody(req);
-				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				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, null, journal, {
+					applyChaosAction(chaosAction, res, falSubmitFixture, journal, {
 						method: req.method ?? "POST",
 						path: pathname,
 						headers: flattenHeaders(req.headers),
@@ -1246,7 +1291,7 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 		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 raw = req.method === "POST" || req.method === "PUT" ? falBody ?? await readBody(req) : "{}";
 				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
 				if (chaosAction) {
 					applyChaosAction(chaosAction, res, null, journal, {
@@ -1271,13 +1316,29 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 			return;
 		}
-		if (pathname.match(FAL_RUN_RE) && req.method === "POST") {
+		const falRunMatch = pathname.match(FAL_RUN_RE);
+		if (falRunMatch && req.method === "POST") {
 			setCorsHeaders(res);
 			try {
-				const raw = await readBody(req);
-				const chaosAction = evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				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, null, journal, {
+					applyChaosAction(chaosAction, res, falRunFixture, journal, {
 						method: req.method ?? "POST",
 						path: pathname,
 						headers: flattenHeaders(req.headers),
@@ -1316,16 +1377,14 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 				message: msg,
 				type: "server_error"
 			} }));
-			else if (!res.writableEnded) {
-				try {
-					res.write(`data: ${JSON.stringify({ error: {
-						message: msg,
-						type: "server_error"
-					} })}\n\n`);
-				} catch (writeErr) {
-					logger.debug("Failed to write error recovery response:", writeErr);
-				}
+			else if (!res.writableEnded) try {
+				res.write(`data: ${JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} })}\n\n`);
 				res.end();
+			} catch (writeErr) {
+				logger.debug("Failed to write error recovery response:", writeErr);
 			}
 		}
 	}
@@ -1374,19 +1433,22 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 		if (pathname === RESPONSES_PATH) handleWebSocketResponses(ws, fixtures, journal, {
 			...defaults,
 			model: "gpt-4",
-			testId: wsTestId
+			testId: wsTestId,
+			upgradeHeaders: req.headers
 		});
 		else if (pathname === REALTIME_PATH) {
 			const model = parsedUrl.searchParams.get("model") ?? "gpt-4o-realtime";
 			handleWebSocketRealtime(ws, fixtures, journal, {
 				...defaults,
 				model,
-				testId: wsTestId
+				testId: wsTestId,
+				upgradeHeaders: req.headers
 			});
 		} else if (pathname === GEMINI_LIVE_PATH) handleWebSocketGeminiLive(ws, fixtures, journal, {
 			...defaults,
 			model: "gemini-2.0-flash",
-			testId: wsTestId
+			testId: wsTestId,
+			upgradeHeaders: req.headers
 		});
 	}
 	const originalClose = server.close.bind(server);