@copilotkit/aimock 1.33.0 → 1.35.0

package/dist/server.cjs

@@ -23,6 +23,8 @@ const require_video = require('./video.cjs');
 const require_fal_audio = require('./fal-audio.cjs');
 const require_fal = require('./fal.cjs');
 const require_openrouter_video = require('./openrouter-video.cjs');
+const require_veo_video = require('./veo-video.cjs');
+const require_grok_video = require('./grok-video.cjs');
 const require_elevenlabs_audio = require('./elevenlabs-audio.cjs');
 const require_ollama = require('./ollama.cjs');
 const require_cohere = require('./cohere.cjs');
@@ -146,11 +148,13 @@ const CONTROL_PREFIX = "/__aimock";
 * `aimock_fixtures_loaded` gauge. Shared by `/reset/fixtures` and the
 * deprecated `/reset` alias.
 */
-function performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, defaults) {
+function performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults) {
 	fixtures.length = 0;
 	journal.clear();
 	videoStates.clear();
 	openRouterVideoJobs.clear();
+	veoVideoJobs.clear();
+	grokVideoJobs.clear();
 	require_fal_audio.falJobs.clear();
 	require_fal.falQueueStates.clear();
 	require_gemini_interactions.resetInteractionCounter();
@@ -161,7 +165,7 @@ function performFixturesReset(fixtures, journal, videoStates, openRouterVideoJob
 * Handle requests under `/__aimock/`. Returns `true` if the request was
 * handled, `false` if the path doesn't match the control prefix.
 */
-async function handleControlAPI(req, res, pathname, fixtures, journal, videoStates, openRouterVideoJobs, defaults) {
+async function handleControlAPI(req, res, pathname, fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults) {
 	if (!pathname.startsWith(CONTROL_PREFIX)) return false;
 	const subPath = pathname.slice(9);
 	setCorsHeaders(res);
@@ -225,7 +229,7 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		return true;
 	}
 	if (subPath === "/reset/fixtures" && req.method === "POST") {
-		performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, defaults);
+		performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults);
 		res.writeHead(200, { "Content-Type": "application/json" });
 		res.end(JSON.stringify({ reset: true }));
 		return true;
@@ -237,7 +241,7 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		return true;
 	}
 	if (subPath === "/reset" && req.method === "POST") {
-		performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, defaults);
+		performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults);
 		const deprecation = "POST /__aimock/reset is deprecated; use POST /__aimock/reset/fixtures (full reset) or POST /__aimock/reset/journal (journal only)";
 		defaults.logger.warn("POST /__aimock/reset is deprecated; use /__aimock/reset/fixtures or /__aimock/reset/journal");
 		res.writeHead(200, {
@@ -379,7 +383,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 	body._endpointType = "chat";
 	body._context = require_helpers.getContext(req);
 	const testId = require_helpers.getTestId(req);
-	const { fixture, skippedBySequenceOrTurn } = require_router.matchFixtureDiagnostic(fixtures, body, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
+	const { fixture, skippedBySequenceOrTurn } = require_router.matchFixtureDiagnostic(fixtures, body, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform, require_router.recordMatchOptions(!!(defaults.record && providerKey), defaults.logger));
 	if (fixture) {
 		journal.incrementFixtureMatchCount(fixture, fixtures, testId);
 		defaults.logger.debug(`Fixture matched: ${JSON.stringify(fixture.match).slice(0, 120)}`);
@@ -530,12 +534,12 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 			}
 		});
 		if (body.stream !== true) {
-			const completion = require_helpers.buildContentWithToolCallsCompletion(response.content, response.toolCalls, body.model, effReasoning, overrides, body.messages);
+			const completion = require_helpers.buildContentWithToolCallsCompletion(response.content ?? "", response.toolCalls ?? [], body.model, effReasoning, overrides, body.messages);
 			res.writeHead(200, { "Content-Type": "application/json" });
 			res.end(JSON.stringify(completion));
 		} else {
-			const chunks = require_helpers.buildContentWithToolCallsChunks(response.content, response.toolCalls, body.model, chunkSize, effReasoning, overrides);
-			const completionText = response.content + response.toolCalls.map((tc) => tc.name + tc.arguments).join("");
+			const chunks = require_helpers.buildContentWithToolCallsChunks(response.content ?? "", response.toolCalls ?? [], body.model, chunkSize, effReasoning, overrides, response.blocks);
+			const completionText = (response.content ?? "") + (response.toolCalls ?? []).map((tc) => tc.name + tc.arguments).join("");
 			const usageChunk = includeUsage ? require_helpers.buildUsageChunk(chunks[0]?.id ?? "chatcmpl-unknown", overrides?.model ?? body.model, chunks[0]?.created ?? Math.floor(Date.now() / 1e3), overrides?.usage ? {
 				prompt_tokens: overrides.usage.prompt_tokens ?? 0,
 				completion_tokens: overrides.usage.completion_tokens ?? 0,
@@ -704,6 +708,12 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 		},
 		get openRouterVideo() {
 			return serverOptions.openRouterVideo;
+		},
+		get veoVideo() {
+			return serverOptions.veoVideo;
+		},
+		get grokVideo() {
+			return serverOptions.grokVideo;
 		}
 	};
 	if (options?.chaos) {
@@ -723,13 +733,24 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 		];
 		for (const { name, value } of chaosRates) if (value !== void 0 && (value < 0 || value > 1)) logger.warn(`Chaos ${name} (${value}) is outside 0-1 range — will be clamped at runtime`);
 	}
-	for (const { name, config } of [{
-		name: "falQueue",
-		config: options?.falQueue
-	}, {
-		name: "openRouterVideo",
-		config: options?.openRouterVideo
-	}]) {
+	for (const { name, config } of [
+		{
+			name: "falQueue",
+			config: options?.falQueue
+		},
+		{
+			name: "openRouterVideo",
+			config: options?.openRouterVideo
+		},
+		{
+			name: "veoVideo",
+			config: options?.veoVideo
+		},
+		{
+			name: "grokVideo",
+			config: options?.grokVideo
+		}
+	]) {
 		if (!config) continue;
 		for (const field of ["pollsBeforeInProgress", "pollsBeforeCompleted"]) {
 			const value = config[field];
@@ -748,6 +769,8 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 	});
 	const videoStates = new require_video.VideoStateMap();
 	const openRouterVideoJobs = new require_openrouter_video.OpenRouterVideoJobMap();
+	const veoVideoJobs = new require_veo_video.VeoVideoJobMap();
+	const grokVideoJobs = new require_grok_video.GrokVideoJobMap();
 	if (mounts) for (const { handler } of mounts) {
 		if (handler.setJournal) handler.setJournal(journal);
 		if (registry && handler.setRegistry) handler.setRegistry(registry);
@@ -794,7 +817,7 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 		});
 		if (pathname.startsWith(CONTROL_PREFIX)) {
-			await handleControlAPI(req, res, pathname, fixtures, journal, videoStates, openRouterVideoJobs, defaults);
+			await handleControlAPI(req, res, pathname, fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults);
 			return;
 		}
 		if (mounts) {
@@ -1239,6 +1262,36 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 			return;
 		}
+		if (pathname === require_metrics.GROK_VIDEO_SUBMIT_PATH && req.method === "POST") {
+			setCorsHeaders(res);
+			try {
+				await require_grok_video.handleGrokVideoCreate(req, res, await require_helpers.readBody(req), fixtures, journal, defaults, setCorsHeaders, grokVideoJobs);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				defaults.logger.error(`grok-video submit: ${msg}`);
+				if (!res.headersSent) {
+					try {
+						journal.add({
+							method: req.method ?? "POST",
+							path: req.url ?? pathname,
+							headers: require_helpers.flattenHeaders(req.headers),
+							body: null,
+							response: {
+								status: 500,
+								fixture: null
+							}
+						});
+					} catch (jErr) {
+						defaults.logger.warn(`grok-video submit: journal write failed after handler error: ${jErr instanceof Error ? jErr.message : String(jErr)}`);
+					}
+					require_sse_writer.writeErrorResponse(res, 500, JSON.stringify({ error: {
+						message: msg,
+						type: "server_error"
+					} }));
+				} else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
 		if (pathname === VIDEOS_PATH && req.method === "POST") {
 			try {
 				await require_video.handleVideoCreate(req, res, await require_helpers.readBody(req), fixtures, journal, defaults, setCorsHeaders, videoStates);
@@ -1252,10 +1305,95 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 			return;
 		}
-		const videoStatusMatch = pathname.match(require_metrics.OPENAI_VIDEO_STATUS_RE);
-		if (videoStatusMatch && req.method === "GET") {
-			const videoId = videoStatusMatch[1];
-			require_video.handleVideoStatus(req, res, videoId, journal, defaults, setCorsHeaders, videoStates);
+		const grokVideoStatusMatch = pathname.match(require_metrics.GROK_VIDEO_STATUS_RE);
+		if (grokVideoStatusMatch && grokVideoStatusMatch[1] !== "generations" && req.method === "GET") {
+			try {
+				await require_grok_video.handleGrokVideoStatus(req, res, grokVideoStatusMatch[1], fixtures, journal, defaults, setCorsHeaders, grokVideoJobs, videoStates);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				defaults.logger.error(`grok-video status: ${msg}`);
+				if (!res.headersSent) {
+					try {
+						journal.add({
+							method: req.method ?? "GET",
+							path: req.url ?? pathname,
+							headers: require_helpers.flattenHeaders(req.headers),
+							body: null,
+							response: {
+								status: 500,
+								fixture: null
+							}
+						});
+					} catch (jErr) {
+						defaults.logger.warn(`grok-video status: journal write failed after handler error: ${jErr instanceof Error ? jErr.message : String(jErr)}`);
+					}
+					require_sse_writer.writeErrorResponse(res, 500, JSON.stringify({ error: {
+						message: msg,
+						type: "server_error"
+					} }));
+				} else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
+		const veoPredictMatch = pathname.match(require_metrics.VEO_PREDICT_LRO_RE);
+		if (veoPredictMatch && req.method === "POST") {
+			setCorsHeaders(res);
+			try {
+				await require_veo_video.handleVeoVideoCreate(req, res, await require_helpers.readBody(req), veoPredictMatch[1], fixtures, journal, defaults, setCorsHeaders, veoVideoJobs);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				defaults.logger.error(`veo-video submit: ${msg}`);
+				if (!res.headersSent) {
+					try {
+						journal.add({
+							method: req.method ?? "POST",
+							path: req.url ?? pathname,
+							headers: require_helpers.flattenHeaders(req.headers),
+							body: null,
+							response: {
+								status: 500,
+								fixture: null
+							}
+						});
+					} catch (jErr) {
+						defaults.logger.warn(`veo-video submit: journal write failed after handler error: ${jErr instanceof Error ? jErr.message : String(jErr)}`);
+					}
+					require_sse_writer.writeErrorResponse(res, 500, JSON.stringify({ error: {
+						message: msg,
+						type: "server_error"
+					} }));
+				} else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
+		const veoOperationMatch = pathname.match(require_metrics.VEO_OPERATION_RE);
+		if (veoOperationMatch && req.method === "GET") {
+			try {
+				await require_veo_video.handleVeoVideoStatus(req, res, veoOperationMatch[1], fixtures, journal, defaults, setCorsHeaders, veoVideoJobs);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				defaults.logger.error(`veo-video status: ${msg}`);
+				if (!res.headersSent) {
+					try {
+						journal.add({
+							method: req.method ?? "GET",
+							path: req.url ?? pathname,
+							headers: require_helpers.flattenHeaders(req.headers),
+							body: null,
+							response: {
+								status: 500,
+								fixture: null
+							}
+						});
+					} catch (jErr) {
+						defaults.logger.warn(`veo-video status: journal write failed after handler error: ${jErr instanceof Error ? jErr.message : String(jErr)}`);
+					}
+					require_sse_writer.writeErrorResponse(res, 500, JSON.stringify({ error: {
+						message: msg,
+						type: "server_error"
+					} }));
+				} else if (!res.writableEnded) res.destroy();
+			}
 			return;
 		}
 		const geminiPredictMatch = pathname.match(GEMINI_PREDICT_RE);
@@ -1676,6 +1814,8 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 		activeConnections.clear();
 		videoStates.clear();
 		openRouterVideoJobs.clear();
+		veoVideoJobs.clear();
+		grokVideoJobs.clear();
 		originalClose(callback);
 		return this;
 	};
@@ -1697,7 +1837,9 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 				url,
 				defaults,
 				videoStates,
-				openRouterVideoJobs
+				openRouterVideoJobs,
+				veoVideoJobs,
+				grokVideoJobs
 			});
 		});
 	});