@copilotkit/aimock 1.40.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);
@@ -249,6 +251,7 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		openRouterVideoJobs,
 		veoVideoJobs,
 		grokVideoJobs,
+		bytePlusVideoJobs,
 		defaults
 	});
 	if (subPath === "/reset" && req.method === "POST") {
@@ -798,6 +801,9 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		},
 		get grokVideo() {
 			return serverOptions.grokVideo;
+		},
+		get bytePlusVideo() {
+			return serverOptions.bytePlusVideo;
 		}
 	};
 	if (options?.chaos) {
@@ -833,6 +839,10 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		{
 			name: "grokVideo",
 			config: options?.grokVideo
+		},
+		{
+			name: "bytePlusVideo",
+			config: options?.bytePlusVideo
 		}
 	]) {
 		if (!config) continue;
@@ -847,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
@@ -855,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);
@@ -880,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);
@@ -912,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) {
@@ -1095,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;
@@ -1293,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: {
@@ -1832,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) {
@@ -1970,6 +2053,7 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 		openRouterVideoJobs.clear();
 		veoVideoJobs.clear();
 		grokVideoJobs.clear();
+		bytePlusVideoJobs.clear();
 		originalClose(callback);
 		return this;
 	};
@@ -1993,7 +2077,8 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 				videoStates,
 				openRouterVideoJobs,
 				veoVideoJobs,
-				grokVideoJobs
+				grokVideoJobs,
+				bytePlusVideoJobs
 			});
 		});
 	});