@copilotkit/aimock 1.17.0 → 1.19.0

package/dist/server.cjs

@@ -21,6 +21,7 @@ const require_transcription = require('./transcription.cjs');
 const require_video = require('./video.cjs');
 const require_elevenlabs_audio = require('./elevenlabs-audio.cjs');
 const require_fal_audio = require('./fal-audio.cjs');
+const require_fal = require('./fal.cjs');
 const require_ollama = require('./ollama.cjs');
 const require_cohere = require('./cohere.cjs');
 const require_search = require('./search.cjs');
@@ -56,6 +57,7 @@ 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",
@@ -106,7 +108,7 @@ const DEFAULT_MODELS = [
 const CORS_HEADERS = {
 	"Access-Control-Allow-Origin": "*",
 	"Access-Control-Allow-Methods": "GET, POST, DELETE, OPTIONS",
-	"Access-Control-Allow-Headers": "Content-Type, Authorization"
+	"Access-Control-Allow-Headers": "*"
 };
 function setCorsHeaders(res) {
 	for (const [key, value] of Object.entries(CORS_HEADERS)) res.setHeader(key, value);
@@ -211,6 +213,7 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		journal.clear();
 		videoStates.clear();
 		require_fal_audio.falJobs.clear();
+		require_fal.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 }));
@@ -329,6 +332,9 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 		} }));
 		return;
 	}
+	const method = req.method ?? "POST";
+	const path = req.url ?? COMPLETIONS_PATH;
+	const flatHeaders = require_helpers.flattenHeaders(req.headers);
 	body._endpointType = "chat";
 	const testId = require_helpers.getTestId(req);
 	const fixture = require_router.matchFixture(fixtures, body, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
@@ -340,18 +346,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 = require_helpers.flattenHeaders(req.headers);
-	if (require_chaos.applyChaos(res, fixture, defaults.chaos, req.headers, journal, {
+	const chaosAction = require_chaos.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") {
+		require_chaos.applyChaosAction(chaosAction, res, fixture, journal, chaosContext, fixture ? "fixture" : "proxy", defaults.registry);
+		return;
+	}
+	if (fixture && chaosAction === "malformed") {
+		require_chaos.applyChaosAction(chaosAction, res, fixture, journal, chaosContext, "fixture", defaults.registry);
+		return;
+	}
 	if (!fixture) {
 		if (defaults.record && providerKey) {
-			if (await require_recorder.proxyAndRecord(req, res, body, providerKey, req.url ?? COMPLETIONS_PATH, fixtures, defaults, raw)) {
+			const hookOptions = chaosAction === "malformed" ? {
+				beforeWriteResponse: () => {
+					require_chaos.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 require_recorder.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,
@@ -386,7 +414,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 		} }));
 		return;
 	}
-	const response = fixture.response;
+	const response = await require_helpers.resolveResponse(fixture, body);
 	const latency = fixture.latency ?? defaults.latency;
 	const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);
 	if (require_helpers.isErrorResponse(response)) {
@@ -868,7 +896,7 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 		const videoStatusMatch = pathname.match(VIDEOS_STATUS_RE);
 		if (videoStatusMatch && req.method === "GET") {
 			const videoId = videoStatusMatch[1];
-			require_video.handleVideoStatus(req, res, videoId, journal, setCorsHeaders, videoStates);
+			require_video.handleVideoStatus(req, res, videoId, journal, defaults, setCorsHeaders, videoStates);
 			return;
 		}
 		const geminiPredictMatch = pathname.match(GEMINI_PREDICT_RE);
@@ -1097,15 +1125,19 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			setCorsHeaders(res);
 			try {
 				const raw = await readBody(req);
-				if (require_chaos.applyChaos(res, null, defaults.chaos, req.headers, journal, {
-					method: req.method ?? "POST",
-					path: pathname,
-					headers: require_helpers.flattenHeaders(req.headers),
-					body: {
-						model: "",
-						messages: []
-					}
-				}, defaults.registry, defaults.logger)) return;
+				const chaosAction = require_chaos.evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					require_chaos.applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "POST",
+						path: pathname,
+						headers: require_helpers.flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
 				await require_elevenlabs_audio.handleElevenLabsAudio(req, res, raw, fixtures, defaults, journal, "sound-generation");
 			} catch (err) {
 				const msg = err instanceof Error ? err.message : "Internal error";
@@ -1123,15 +1155,19 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			const musicSubType = musicMatch[1] ?? "music";
 			try {
 				const raw = await readBody(req);
-				if (require_chaos.applyChaos(res, null, defaults.chaos, req.headers, journal, {
-					method: req.method ?? "POST",
-					path: pathname,
-					headers: require_helpers.flattenHeaders(req.headers),
-					body: {
-						model: "",
-						messages: []
-					}
-				}, defaults.registry, defaults.logger)) return;
+				const chaosAction = require_chaos.evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					require_chaos.applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "POST",
+						path: pathname,
+						headers: require_helpers.flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
 				await require_elevenlabs_audio.handleElevenLabsAudio(req, res, raw, fixtures, defaults, journal, musicSubType);
 			} catch (err) {
 				const msg = err instanceof Error ? err.message : "Internal error";
@@ -1143,19 +1179,51 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 			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 = require_chaos.evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					require_chaos.applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "GET",
+						path: pathname,
+						headers: require_helpers.flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
+				if (await require_fal.handleFal(req, res, raw, pathname, fixtures, defaults, journal) === "handled") return;
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) require_sse_writer.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 (require_chaos.applyChaos(res, null, defaults.chaos, req.headers, journal, {
-					method: req.method ?? "POST",
-					path: pathname,
-					headers: require_helpers.flattenHeaders(req.headers),
-					body: {
-						model: "",
-						messages: []
-					}
-				}, defaults.registry, defaults.logger)) return;
+				const chaosAction = require_chaos.evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					require_chaos.applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "POST",
+						path: pathname,
+						headers: require_helpers.flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
 				await require_fal_audio.handleFalQueue(req, res, raw, pathname, fixtures, defaults, journal);
 			} catch (err) {
 				const msg = err instanceof Error ? err.message : "Internal error";
@@ -1171,15 +1239,19 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			setCorsHeaders(res);
 			try {
 				const raw = req.method === "POST" ? await readBody(req) : "{}";
-				if (require_chaos.applyChaos(res, null, defaults.chaos, req.headers, journal, {
-					method: req.method ?? "GET",
-					path: pathname,
-					headers: require_helpers.flattenHeaders(req.headers),
-					body: {
-						model: "",
-						messages: []
-					}
-				}, defaults.registry, defaults.logger)) return;
+				const chaosAction = require_chaos.evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					require_chaos.applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "GET",
+						path: pathname,
+						headers: require_helpers.flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
 				await require_fal_audio.handleFalQueue(req, res, raw, pathname, fixtures, defaults, journal);
 			} catch (err) {
 				const msg = err instanceof Error ? err.message : "Internal error";
@@ -1195,15 +1267,19 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			setCorsHeaders(res);
 			try {
 				const raw = await readBody(req);
-				if (require_chaos.applyChaos(res, null, defaults.chaos, req.headers, journal, {
-					method: req.method ?? "POST",
-					path: pathname,
-					headers: require_helpers.flattenHeaders(req.headers),
-					body: {
-						model: "",
-						messages: []
-					}
-				}, defaults.registry, defaults.logger)) return;
+				const chaosAction = require_chaos.evaluateChaos(null, defaults.chaos, req.headers, defaults.logger);
+				if (chaosAction) {
+					require_chaos.applyChaosAction(chaosAction, res, null, journal, {
+						method: req.method ?? "POST",
+						path: pathname,
+						headers: require_helpers.flattenHeaders(req.headers),
+						body: {
+							model: "",
+							messages: []
+						}
+					}, "fixture", defaults.registry);
+					return;
+				}
 				await require_fal_audio.handleFalQueue(req, res, raw, pathname, fixtures, defaults, journal);
 			} catch (err) {
 				const msg = err instanceof Error ? err.message : "Internal error";