@copilotkit/aimock 1.39.0 → 1.41.0

package/dist/server.d.cts

@@ -7,6 +7,7 @@ import { VideoStateMap } from "./video.cjs";
 import { OpenRouterVideoJobMap } from "./openrouter-video.cjs";
 import { VeoVideoJobMap } from "./veo-video.cjs";
 import { GrokVideoJobMap } from "./grok-video.cjs";
+import { BytePlusVideoJobMap } from "./byteplus-video.cjs";
 import * as http$1 from "node:http";
 
 //#region src/server.d.ts
@@ -19,6 +20,7 @@ interface ServerInstance {
   openRouterVideoJobs: OpenRouterVideoJobMap;
   veoVideoJobs: VeoVideoJobMap;
   grokVideoJobs: GrokVideoJobMap;
+  bytePlusVideoJobs: BytePlusVideoJobMap;
 }
 /**
  * The per-server state a full reset clears. `ServerInstance` structurally

package/dist/server.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.cts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;UAyHiB,cAAA;EAAA,MAAA,EACP,MAAA,CAAK,MADgB;EAAA,OAAA,EAEpB,OAFoB;KACrB,EAAA,MAAK;UACJ,EAEC,eAFD;aAEC,EACG,aADH;qBACG,EACQ,qBADR;cACQ,EACP,cADO;eACP,EACC,eADD;;;AAorChB;;;;;UAAiB,eAAA;UACP;UACA;cACI;;iBAMQ,YAAA,WACV,qBACA,4BACD;;WAA+B;sBACtB,kBACjB,QAAQ"}
+{"version":3,"file":"server.d.cts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;UAgIiB,cAAA;EAAA,MAAA,EACP,MAAA,CAAK,MADgB;EAAA,OAAA,EAEpB,OAFoB;KACrB,EAAA,MAAK;UACJ,EAEC,eAFD;aAEC,EACG,aADH;qBACG,EACQ,qBADR;cACQ,EACP,cADO;eACP,EACC,eADD;mBACC,EACI,mBADJ;;;AA+rCjB;;;;;UAAiB,eAAA;UACP;UACA;cACI;;iBAMQ,YAAA,WACV,qBACA,4BACD;;WAA+B;sBACtB,kBACjB,QAAQ"}

package/dist/server.d.ts

@@ -7,6 +7,7 @@ import { VideoStateMap } from "./video.js";
 import { OpenRouterVideoJobMap } from "./openrouter-video.js";
 import { VeoVideoJobMap } from "./veo-video.js";
 import { GrokVideoJobMap } from "./grok-video.js";
+import { BytePlusVideoJobMap } from "./byteplus-video.js";
 import * as http$1 from "node:http";
 
 //#region src/server.d.ts
@@ -19,6 +20,7 @@ interface ServerInstance {
   openRouterVideoJobs: OpenRouterVideoJobMap;
   veoVideoJobs: VeoVideoJobMap;
   grokVideoJobs: GrokVideoJobMap;
+  bytePlusVideoJobs: BytePlusVideoJobMap;
 }
 /**
  * The per-server state a full reset clears. `ServerInstance` structurally

package/dist/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;UAyHiB,cAAA;EAAA,MAAA,EACP,MAAA,CAAK,MADgB;EAAA,OAAA,EAEpB,OAFoB;KACrB,EAAA,MAAK;UACJ,EAEC,eAFD;aAEC,EACG,aADH;qBACG,EACQ,qBADR;cACQ,EACP,cADO;eACP,EACC,eADD;;;AAorChB;;;;;UAAiB,eAAA;UACP;UACA;cACI;;iBAMQ,YAAA,WACV,qBACA,4BACD;;WAA+B;sBACtB,kBACjB,QAAQ"}
+{"version":3,"file":"server.d.ts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;UAgIiB,cAAA;EAAA,MAAA,EACP,MAAA,CAAK,MADgB;EAAA,OAAA,EAEpB,OAFoB;KACrB,EAAA,MAAK;UACJ,EAEC,eAFD;aAEC,EACG,aADH;qBACG,EACQ,qBADR;cACQ,EACP,cADO;eACP,EACC,eADD;mBACC,EACI,mBADJ;;;AA+rCjB;;;;;UAAiB,eAAA;UACP;UACA;cACI;;iBAMQ,YAAA,WACV,qBACA,4BACD;;WAA+B;sBACtB,kBACjB,QAAQ"}

package/dist/server.js

@@ -26,6 +26,7 @@ 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 { BytePlusVideoJobMap, handleBytePlusVideoCreate, handleBytePlusVideoStatus } from "./byteplus-video.js";
 import { handleElevenLabsAudio, handleElevenLabsTTS } from "./elevenlabs-audio.js";
 import { handleOllama, handleOllamaEmbeddings, handleOllamaGenerate } from "./ollama.js";
 import { handleCohere, handleCohereEmbed } from "./cohere.js";
@@ -36,7 +37,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 { 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 { BYTEPLUS_VIDEO_STATUS_RE, BYTEPLUS_VIDEO_SUBMIT_RE, 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
@@ -167,13 +168,14 @@ function performFullReset(fixtures, targets) {
 	targets.openRouterVideoJobs.clear();
 	targets.veoVideoJobs.clear();
 	targets.grokVideoJobs.clear();
+	targets.bytePlusVideoJobs.clear();
 	if (targets.defaults.registry) targets.defaults.registry.setGauge("aimock_fixtures_loaded", {}, fixtures.length);
 }
 /**
 * 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, veoVideoJobs, grokVideoJobs, defaults) {
+async function handleControlAPI(req, res, pathname, fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, bytePlusVideoJobs, defaults) {
 	if (!pathname.startsWith(CONTROL_PREFIX)) return false;
 	const subPath = pathname.slice(9);
 	setCorsHeaders(res);
@@ -187,6 +189,11 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		res.end(JSON.stringify(journal.getAll()));
 		return true;
 	}
+	if (subPath === "/fixtures" && req.method === "GET") {
+		res.writeHead(200, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ count: fixtures.length }));
+		return true;
+	}
 	if (subPath === "/fixtures" && req.method === "POST") {
 		let raw;
 		try {
@@ -242,6 +249,7 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		openRouterVideoJobs,
 		veoVideoJobs,
 		grokVideoJobs,
+		bytePlusVideoJobs,
 		defaults
 	});
 	if (subPath === "/reset" && req.method === "POST") {
@@ -791,6 +799,9 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		},
 		get grokVideo() {
 			return serverOptions.grokVideo;
+		},
+		get bytePlusVideo() {
+			return serverOptions.bytePlusVideo;
 		}
 	};
 	if (options?.chaos) {
@@ -826,6 +837,10 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		{
 			name: "grokVideo",
 			config: options?.grokVideo
+		},
+		{
+			name: "bytePlusVideo",
+			config: options?.bytePlusVideo
 		}
 	]) {
 		if (!config) continue;
@@ -840,6 +855,16 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		const cap = options?.record?.openRouterVideo?.maxContentBytes;
 		if (cap !== void 0 && (!Number.isInteger(cap) || cap < 0)) logger.warn(`record.openRouterVideo.maxContentBytes (${cap}) is not a non-negative integer — using the default cap (${OPENROUTER_VIDEO_DEFAULT_MAX_CONTENT_BYTES})`);
 	}
+	{
+		const bytePlusBase = options?.record?.providers?.byteplus;
+		if (bytePlusBase !== void 0) {
+			let basePath;
+			try {
+				basePath = new URL(bytePlusBase).pathname;
+			} catch {}
+			if (basePath !== void 0 && /(^|\/)api\/v3\/?$/.test(basePath)) throw new Error(`record.providers.byteplus (${bytePlusBase}) ends in /api/v3, but aimock appends the /api/v3 prefix itself — every BytePlus request would go to /api/v3/api/v3/... and fail upstream. Configure the ORIGIN only (e.g. https://ark.ap-southeast.bytepluses.com).`);
+		}
+	}
 	const journal = new Journal({
 		maxEntries: options?.journalMaxEntries ?? 1e3,
 		fixtureCountsMaxTestIds: options?.fixtureCountsMaxTestIds ?? 500
@@ -848,6 +873,7 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 	const openRouterVideoJobs = new OpenRouterVideoJobMap();
 	const veoVideoJobs = new VeoVideoJobMap();
 	const grokVideoJobs = new GrokVideoJobMap();
+	const bytePlusVideoJobs = new BytePlusVideoJobMap();
 	if (mounts) for (const { handler } of mounts) {
 		if (handler.setJournal) handler.setJournal(journal);
 		if (registry && handler.setRegistry) handler.setRegistry(registry);
@@ -873,7 +899,9 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		const startTime = registry ? process.hrtime.bigint() : 0n;
 		const parsedUrl = new URL(req.url ?? "/", `http://${req.headers.host ?? "localhost"}`);
 		let pathname = parsedUrl.pathname;
-		const isOpenRouter = isOpenRouterPath(pathname);
+		const originalPathname = pathname;
+		const isOpenRouter = isOpenRouterPath(originalPathname);
+		const isBytePlusArk = defaults.record?.providers.byteplus !== void 0 && originalPathname.startsWith("/api/v3/");
 		if (registry) res.on("finish", () => {
 			try {
 				const normalizedPath = normalizePathLabel(pathname);
@@ -905,7 +933,7 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 			markAuthenticatedRequest(req, resolvedAuth.policy);
 		}
 		if (pathname.startsWith(CONTROL_PREFIX)) {
-			await handleControlAPI(req, res, pathname, fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults);
+			await handleControlAPI(req, res, pathname, fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, bytePlusVideoJobs, defaults);
 			return;
 		}
 		if (mounts) {
@@ -1088,6 +1116,66 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 			}
 			return;
 		}
+		const bytePlusVideoStatusMatch = pathname.match(BYTEPLUS_VIDEO_STATUS_RE);
+		if (bytePlusVideoStatusMatch && req.method === "GET") {
+			try {
+				await handleBytePlusVideoStatus(req, res, bytePlusVideoStatusMatch[1], fixtures, journal, defaults, setCorsHeaders, bytePlusVideoJobs);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				defaults.logger.error(`byteplus-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(`byteplus-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;
+		}
+		if (BYTEPLUS_VIDEO_SUBMIT_RE.test(pathname) && req.method === "POST") {
+			setCorsHeaders(res);
+			try {
+				await handleBytePlusVideoCreate(req, res, await readBody(req), fixtures, journal, defaults, setCorsHeaders, bytePlusVideoJobs);
+			} catch (err) {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				defaults.logger.error(`byteplus-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(`byteplus-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 === "/api/v1/models" && req.method === "GET") {
 			handleOpenRouterModels(req, res, fixtures, journal, defaults, setCorsHeaders);
 			return;
@@ -1286,7 +1374,7 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		}
 		if (pathname === IMAGES_PATH && req.method === "POST") {
 			try {
-				await handleImages(req, res, await readBody(req), fixtures, journal, defaults, setCorsHeaders);
+				await handleImages(req, res, await readBody(req), fixtures, journal, defaults, setCorsHeaders, "openai", void 0, isBytePlusArk ? "byteplus" : void 0);
 			} catch (err) {
 				const msg = err instanceof Error ? err.message : "Internal error";
 				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
@@ -1825,7 +1913,7 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 			handleNotFound(res, "Not found");
 			return;
 		}
-		const completionsProvider = azureDeploymentId ? "azure" : isOpenRouter ? "openrouter" : "openai";
+		const completionsProvider = azureDeploymentId ? "azure" : isOpenRouter ? "openrouter" : isBytePlusArk ? "byteplus" : "openai";
 		try {
 			await handleCompletions(req, res, fixtures, journal, defaults, azureDeploymentId, completionsProvider, isOpenRouter);
 		} catch (err) {
@@ -1853,7 +1941,48 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 			if (!socket.destroyed) socket.destroy();
 		});
 	});
+	/**
+	* Body bytes the platform already parsed onto `req` for an upgrade request.
+	*
+	* Node >= 26 delivers an upgrade request's body here and leaves `head`
+	* empty; Node <= 24 leaves the body in `head` and ends `req` with nothing.
+	* The stream has already ended in both cases, so this resolves on the next
+	* tick either way — it never waits on the network.
+	*/
+	function drainParsedRequestBody(req) {
+		return new Promise((resolve) => {
+			const chunks = [];
+			let settled = false;
+			const done = () => {
+				if (settled) return;
+				settled = true;
+				resolve(Buffer.concat(chunks));
+			};
+			req.on("data", (c) => chunks.push(c));
+			req.on("end", done);
+			req.on("error", done);
+			req.on("close", done);
+			req.resume();
+		});
+	}
 	async function handleUpgradeRequest(req, socket, head) {
+		if ((req.headers.upgrade ?? "").toLowerCase() !== "websocket") {
+			const parsedBody = await drainParsedRequestBody(req);
+			const requestLine = `${req.method} ${req.url} HTTP/${req.httpVersion}\r\n`;
+			const headerLines = [];
+			for (let i = 0; i < req.rawHeaders.length; i += 2) {
+				if (/^(?:upgrade|connection)$/i.test(req.rawHeaders[i])) continue;
+				headerLines.push(`${req.rawHeaders[i]}: ${req.rawHeaders[i + 1]}`);
+			}
+			const rebuilt = Buffer.from(requestLine + headerLines.join("\r\n") + "\r\n\r\n");
+			socket.unshift(Buffer.concat([
+				rebuilt,
+				parsedBody,
+				head
+			]));
+			server.emit("connection", socket);
+			return;
+		}
 		const parsedUrl = new URL(req.url ?? "/", `http://${req.headers.host ?? "localhost"}`);
 		let pathname = parsedUrl.pathname;
 		if (!validateRequestApiKey(req, resolvedAuth.policy).ok) {
@@ -1922,6 +2051,7 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		openRouterVideoJobs.clear();
 		veoVideoJobs.clear();
 		grokVideoJobs.clear();
+		bytePlusVideoJobs.clear();
 		originalClose(callback);
 		return this;
 	};
@@ -1945,7 +2075,8 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 				videoStates,
 				openRouterVideoJobs,
 				veoVideoJobs,
-				grokVideoJobs
+				grokVideoJobs,
+				bytePlusVideoJobs
 			});
 		});
 	});