@copilotkit/aimock 1.39.0 → 1.41.0

package/dist/server.cjs

@@ -27,6 +27,7 @@ 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_byteplus_video = require('./byteplus-video.cjs');
 const require_elevenlabs_audio = require('./elevenlabs-audio.cjs');
 const require_ollama = require('./ollama.cjs');
 const require_cohere = require('./cohere.cjs');
@@ -169,13 +170,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);
@@ -189,6 +191,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 {
@@ -244,6 +251,7 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		openRouterVideoJobs,
 		veoVideoJobs,
 		grokVideoJobs,
+		bytePlusVideoJobs,
 		defaults
 	});
 	if (subPath === "/reset" && req.method === "POST") {
@@ -793,6 +801,9 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		},
 		get grokVideo() {
 			return serverOptions.grokVideo;
+		},
+		get bytePlusVideo() {
+			return serverOptions.bytePlusVideo;
 		}
 	};
 	if (options?.chaos) {
@@ -828,6 +839,10 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		{
 			name: "grokVideo",
 			config: options?.grokVideo
+		},
+		{
+			name: "bytePlusVideo",
+			config: options?.bytePlusVideo
 		}
 	]) {
 		if (!config) continue;
@@ -842,6 +857,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 (${require_openrouter_video.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 require_journal.Journal({
 		maxEntries: options?.journalMaxEntries ?? 1e3,
 		fixtureCountsMaxTestIds: options?.fixtureCountsMaxTestIds ?? 500
@@ -850,6 +875,7 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 	const openRouterVideoJobs = new require_openrouter_video.OpenRouterVideoJobMap();
 	const veoVideoJobs = new require_veo_video.VeoVideoJobMap();
 	const grokVideoJobs = new require_grok_video.GrokVideoJobMap();
+	const bytePlusVideoJobs = new require_byteplus_video.BytePlusVideoJobMap();
 	if (mounts) for (const { handler } of mounts) {
 		if (handler.setJournal) handler.setJournal(journal);
 		if (registry && handler.setRegistry) handler.setRegistry(registry);
@@ -875,7 +901,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 = require_openrouter_chat.isOpenRouterPath(pathname);
+		const originalPathname = pathname;
+		const isOpenRouter = require_openrouter_chat.isOpenRouterPath(originalPathname);
+		const isBytePlusArk = defaults.record?.providers.byteplus !== void 0 && originalPathname.startsWith("/api/v3/");
 		if (registry) res.on("finish", () => {
 			try {
 				const normalizedPath = require_metrics.normalizePathLabel(pathname);
@@ -907,7 +935,7 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 			require_api_key_auth.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) {
@@ -1090,6 +1118,66 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 			}
 			return;
 		}
+		const bytePlusVideoStatusMatch = pathname.match(require_metrics.BYTEPLUS_VIDEO_STATUS_RE);
+		if (bytePlusVideoStatusMatch && req.method === "GET") {
+			try {
+				await require_byteplus_video.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: require_helpers.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)}`);
+					}
+					require_sse_writer.writeErrorResponse(res, 500, JSON.stringify({ error: {
+						message: msg,
+						type: "server_error"
+					} }));
+				} else if (!res.writableEnded) res.destroy();
+			}
+			return;
+		}
+		if (require_metrics.BYTEPLUS_VIDEO_SUBMIT_RE.test(pathname) && req.method === "POST") {
+			setCorsHeaders(res);
+			try {
+				await require_byteplus_video.handleBytePlusVideoCreate(req, res, await require_helpers.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: require_helpers.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)}`);
+					}
+					require_sse_writer.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") {
 			require_openrouter_chat.handleOpenRouterModels(req, res, fixtures, journal, defaults, setCorsHeaders);
 			return;
@@ -1288,7 +1376,7 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		}
 		if (pathname === IMAGES_PATH && req.method === "POST") {
 			try {
-				await require_images.handleImages(req, res, await require_helpers.readBody(req), fixtures, journal, defaults, setCorsHeaders);
+				await require_images.handleImages(req, res, await require_helpers.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) require_sse_writer.writeErrorResponse(res, 500, JSON.stringify({ error: {
@@ -1827,7 +1915,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) {
@@ -1855,7 +1943,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 (!require_api_key_auth.validateRequestApiKey(req, resolvedAuth.policy).ok) {
@@ -1924,6 +2053,7 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		openRouterVideoJobs.clear();
 		veoVideoJobs.clear();
 		grokVideoJobs.clear();
+		bytePlusVideoJobs.clear();
 		originalClose(callback);
 		return this;
 	};
@@ -1947,7 +2077,8 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 				videoStates,
 				openRouterVideoJobs,
 				veoVideoJobs,
-				grokVideoJobs
+				grokVideoJobs,
+				bytePlusVideoJobs
 			});
 		});
 	});