@copilotkit/aimock 1.34.0 → 1.35.1

package/dist/server.d.cts

@@ -5,6 +5,8 @@ import { RerankFixture } from "./rerank.cjs";
 import { ModerationFixture } from "./moderation.cjs";
 import { VideoStateMap } from "./video.cjs";
 import { OpenRouterVideoJobMap } from "./openrouter-video.cjs";
+import { VeoVideoJobMap } from "./veo-video.cjs";
+import { GrokVideoJobMap } from "./grok-video.cjs";
 import * as http$1 from "node:http";
 
 //#region src/server.d.ts
@@ -15,6 +17,8 @@ interface ServerInstance {
   defaults: HandlerDefaults;
   videoStates: VideoStateMap;
   openRouterVideoJobs: OpenRouterVideoJobMap;
+  veoVideoJobs: VeoVideoJobMap;
+  grokVideoJobs: GrokVideoJobMap;
 }
 interface ServiceFixtures {
   search: SearchFixture[];

package/dist/server.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.cts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;UA0FiB,cAAA;UACP,MAAA,CAAK;EADE,OAAA,EAEN,OAFM;EAAc,GAAA,EAAA,MAAA;UACrB,EAGE,eAHG;aACJ,EAGI,aAHJ;qBAEC,EAEW,qBAFX;;AAEW,UA68BN,eAAA,CA78BM;EAAqB,MAAA,EA88BlC,aA98BkC,EAAA;EA68B3B,MAAA,EAEP,aAFsB,EAAA;EAAA,UAAA,EAGlB,iBAHkB,EAAA;;AAEtB,iBAOY,YAAA,CAPZ,QAAA,EAQE,OARF,EAAA,EAAA,OAAA,CAAA,EASE,iBATF,EAAA,MAOV,CAPU,EAUC,KAVD,CAAA;MACI,EAAA,MAAA;EAAiB,OAAA,EASW,SATX;AAM/B,CAAA,CAAA,EAAsB,eAAY,CAAA,EAId,eAJc,CAAA,EAK/B,OAL+B,CAKvB,cALuB,CAAA"}
+{"version":3,"file":"server.d.cts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;UA+FiB,cAAA;UACP,MAAA,CAAK;EADE,OAAA,EAEN,OAFM;EAAc,GAAA,EAAA,MAAA;UACrB,EAGE,eAHG;aACJ,EAGI,aAHJ;qBAEC,EAEW,qBAFX;cACG,EAEC,cAFD;eACQ,EAEN,eAFM;;AAEN,UAq+BA,eAAA,CAr+BA;EAAe,MAAA,EAs+BtB,aAt+BsB,EAAA;EAq+Bf,MAAA,EAEP,aAFsB,EAAA;EAAA,UAAA,EAGlB,iBAHkB,EAAA;;AAEtB,iBAOY,YAAA,CAPZ,QAAA,EAQE,OARF,EAAA,EAAA,OAAA,CAAA,EASE,iBATF,EAAA,MAOV,CAPU,EAUC,KAVD,CAAA;MACI,EAAA,MAAA;EAAiB,OAAA,EASW,SATX;AAM/B,CAAA,CAAA,EAAsB,eAAY,CAAA,EAId,eAJc,CAAA,EAK/B,OAL+B,CAKvB,cALuB,CAAA"}

package/dist/server.d.ts

@@ -5,6 +5,8 @@ import { RerankFixture } from "./rerank.js";
 import { ModerationFixture } from "./moderation.js";
 import { VideoStateMap } from "./video.js";
 import { OpenRouterVideoJobMap } from "./openrouter-video.js";
+import { VeoVideoJobMap } from "./veo-video.js";
+import { GrokVideoJobMap } from "./grok-video.js";
 import * as http$1 from "node:http";
 
 //#region src/server.d.ts
@@ -15,6 +17,8 @@ interface ServerInstance {
   defaults: HandlerDefaults;
   videoStates: VideoStateMap;
   openRouterVideoJobs: OpenRouterVideoJobMap;
+  veoVideoJobs: VeoVideoJobMap;
+  grokVideoJobs: GrokVideoJobMap;
 }
 interface ServiceFixtures {
   search: SearchFixture[];

package/dist/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;UA0FiB,cAAA;UACP,MAAA,CAAK;EADE,OAAA,EAEN,OAFM;EAAc,GAAA,EAAA,MAAA;UACrB,EAGE,eAHG;aACJ,EAGI,aAHJ;qBAEC,EAEW,qBAFX;;AAEW,UA68BN,eAAA,CA78BM;EAAqB,MAAA,EA88BlC,aA98BkC,EAAA;EA68B3B,MAAA,EAEP,aAFsB,EAAA;EAAA,UAAA,EAGlB,iBAHkB,EAAA;;AAEtB,iBAOY,YAAA,CAPZ,QAAA,EAQE,OARF,EAAA,EAAA,OAAA,CAAA,EASE,iBATF,EAAA,MAOV,CAPU,EAUC,KAVD,CAAA;MACI,EAAA,MAAA;EAAiB,OAAA,EASW,SATX;AAM/B,CAAA,CAAA,EAAsB,eAAY,CAAA,EAId,eAJc,CAAA,EAK/B,OAL+B,CAKvB,cALuB,CAAA"}
+{"version":3,"file":"server.d.ts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;UA+FiB,cAAA;UACP,MAAA,CAAK;EADE,OAAA,EAEN,OAFM;EAAc,GAAA,EAAA,MAAA;UACrB,EAGE,eAHG;aACJ,EAGI,aAHJ;qBAEC,EAEW,qBAFX;cACG,EAEC,cAFD;eACQ,EAEN,eAFM;;AAEN,UAq+BA,eAAA,CAr+BA;EAAe,MAAA,EAs+BtB,aAt+BsB,EAAA;EAq+Bf,MAAA,EAEP,aAFsB,EAAA;EAAA,UAAA,EAGlB,iBAHkB,EAAA;;AAEtB,iBAOY,YAAA,CAPZ,QAAA,EAQE,OARF,EAAA,EAAA,OAAA,CAAA,EASE,iBATF,EAAA,MAOV,CAPU,EAUC,KAVD,CAAA;MACI,EAAA,MAAA;EAAiB,OAAA,EASW,SATX;AAM/B,CAAA,CAAA,EAAsB,eAAY,CAAA,EAId,eAJc,CAAA,EAK/B,OAL+B,CAKvB,cALuB,CAAA"}

package/dist/server.js

@@ -18,10 +18,12 @@ import { handleEmbeddings } from "./embeddings.js";
 import { handleTranscription } from "./transcription.js";
 import { handleImageEdit, handleImageVariations, handleImages } from "./images.js";
 import { handleSpeech } from "./speech.js";
-import { VideoStateMap, handleVideoCreate, handleVideoStatus } from "./video.js";
+import { VideoStateMap, handleVideoCreate } from "./video.js";
 import { falJobs, handleFalQueue } from "./fal-audio.js";
 import { falQueueStates, handleFal } from "./fal.js";
 import { OPENROUTER_VIDEO_DEFAULT_MAX_CONTENT_BYTES, OpenRouterVideoJobMap, handleOpenRouterVideoContent, handleOpenRouterVideoCreate, handleOpenRouterVideoModels, handleOpenRouterVideoStatus } from "./openrouter-video.js";
+import { VeoVideoJobMap, handleVeoVideoCreate, handleVeoVideoStatus } from "./veo-video.js";
+import { GrokVideoJobMap, handleGrokVideoCreate, handleGrokVideoStatus } from "./grok-video.js";
 import { handleElevenLabsAudio, handleElevenLabsTTS } from "./elevenlabs-audio.js";
 import { handleOllama, handleOllamaEmbeddings, handleOllamaGenerate } from "./ollama.js";
 import { handleCohere, handleCohereEmbed } from "./cohere.js";
@@ -32,7 +34,7 @@ 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 { OPENAI_VIDEO_STATUS_RE, OPENROUTER_VIDEO_CONTENT_RE, OPENROUTER_VIDEO_STATUS_RE, createMetricsRegistry, normalizePathLabel } from "./metrics.js";
+import { GROK_VIDEO_STATUS_RE, GROK_VIDEO_SUBMIT_PATH, OPENROUTER_VIDEO_CONTENT_RE, OPENROUTER_VIDEO_STATUS_RE, VEO_OPERATION_RE, VEO_PREDICT_LRO_RE, createMetricsRegistry, normalizePathLabel } from "./metrics.js";
 import * as http$1 from "node:http";
 
 //#region src/server.ts
@@ -144,11 +146,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();
 	falJobs.clear();
 	falQueueStates.clear();
 	resetInteractionCounter();
@@ -159,7 +163,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);
@@ -223,7 +227,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;
@@ -235,7 +239,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, {
@@ -528,12 +532,12 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 			}
 		});
 		if (body.stream !== true) {
-			const completion = buildContentWithToolCallsCompletion(response.content, response.toolCalls, body.model, effReasoning, overrides, body.messages);
+			const completion = 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 = 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 = 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 ? 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,
@@ -702,6 +706,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) {
@@ -721,13 +731,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];
@@ -746,6 +767,8 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 	});
 	const videoStates = new VideoStateMap();
 	const openRouterVideoJobs = new OpenRouterVideoJobMap();
+	const veoVideoJobs = new VeoVideoJobMap();
+	const grokVideoJobs = new GrokVideoJobMap();
 	if (mounts) for (const { handler } of mounts) {
 		if (handler.setJournal) handler.setJournal(journal);
 		if (registry && handler.setRegistry) handler.setRegistry(registry);
@@ -792,7 +815,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) {
@@ -1237,6 +1260,36 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 			return;
 		}
+		if (pathname === GROK_VIDEO_SUBMIT_PATH && req.method === "POST") {
+			setCorsHeaders(res);
+			try {
+				await handleGrokVideoCreate(req, res, await 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: 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)}`);
+					}
+					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 handleVideoCreate(req, res, await readBody(req), fixtures, journal, defaults, setCorsHeaders, videoStates);
@@ -1250,10 +1303,95 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 			}
 			return;
 		}
-		const videoStatusMatch = pathname.match(OPENAI_VIDEO_STATUS_RE);
-		if (videoStatusMatch && req.method === "GET") {
-			const videoId = videoStatusMatch[1];
-			handleVideoStatus(req, res, videoId, journal, defaults, setCorsHeaders, videoStates);
+		const grokVideoStatusMatch = pathname.match(GROK_VIDEO_STATUS_RE);
+		if (grokVideoStatusMatch && grokVideoStatusMatch[1] !== "generations" && req.method === "GET") {
+			try {
+				await 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: 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)}`);
+					}
+					writeErrorResponse(res, 500, JSON.stringify({ error: {
+						message: msg,
+						type: "server_error"
+					} }));
+				} else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
+		const veoPredictMatch = pathname.match(VEO_PREDICT_LRO_RE);
+		if (veoPredictMatch && req.method === "POST") {
+			setCorsHeaders(res);
+			try {
+				await handleVeoVideoCreate(req, res, await 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: 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)}`);
+					}
+					writeErrorResponse(res, 500, JSON.stringify({ error: {
+						message: msg,
+						type: "server_error"
+					} }));
+				} else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
+		const veoOperationMatch = pathname.match(VEO_OPERATION_RE);
+		if (veoOperationMatch && req.method === "GET") {
+			try {
+				await 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: 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)}`);
+					}
+					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);
@@ -1674,6 +1812,8 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 		activeConnections.clear();
 		videoStates.clear();
 		openRouterVideoJobs.clear();
+		veoVideoJobs.clear();
+		grokVideoJobs.clear();
 		originalClose(callback);
 		return this;
 	};
@@ -1695,7 +1835,9 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 				url,
 				defaults,
 				videoStates,
-				openRouterVideoJobs
+				openRouterVideoJobs,
+				veoVideoJobs,
+				grokVideoJobs
 			});
 		});
 	});