@copilotkit/aimock 1.38.0 → 1.40.0

package/dist/server.d.cts

@@ -20,6 +20,12 @@ interface ServerInstance {
   veoVideoJobs: VeoVideoJobMap;
   grokVideoJobs: GrokVideoJobMap;
 }
+/**
+ * The per-server state a full reset clears. `ServerInstance` structurally
+ * satisfies this, so `LLMock.reset()` and the control-API full-reset route
+ * share a single definition of "everything" instead of two lists that drift.
+ */
+
 interface ServiceFixtures {
   search: SearchFixture[];
   rerank: RerankFixture[];
@@ -30,7 +36,6 @@ declare function createServer(fixtures: Fixture[], options?: MockServerOptions,
   handler: Mountable;
 }>, serviceFixtures?: ServiceFixtures): Promise<ServerInstance>;
 /** @internal Config startup uses this to avoid parsing a selected auth source twice. */
-
 //#endregion
 export { ServerInstance, ServiceFixtures, createServer };
 //# sourceMappingURL=server.d.cts.map

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;;AACgB,UAyqCf,eAAA,CAzqCe;EAyqCf,MAAA,EACP,aADsB,EAAA;EAAA,MAAA,EAEtB,aAFsB,EAAA;YACtB,EAEI,iBAFJ,EAAA;;AAEI,iBAMQ,YAAA,CANR,QAAA,EAOF,OAPE,EAAA,EAAA,OAAA,CAAA,EAQF,iBARE,EAAA,MAMoB,CANpB,EASH,KATG,CAAA;EAAiB,IAAA,EAAA,MAAA;EAMT,OAAA,EAGoB,SAHR;CAAA,CAAA,EAAA,eAAA,CAAA,EAId,eAJc,CAAA,EAK/B,OAL+B,CAKvB,cALuB,CAAA"}
+{"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;;;AA2rChB;;;;;UAAiB,eAAA;UACP;UACA;cACI;;iBAMQ,YAAA,WACV,qBACA,4BACD;;WAA+B;sBACtB,kBACjB,QAAQ"}

package/dist/server.d.ts

@@ -20,6 +20,12 @@ interface ServerInstance {
   veoVideoJobs: VeoVideoJobMap;
   grokVideoJobs: GrokVideoJobMap;
 }
+/**
+ * The per-server state a full reset clears. `ServerInstance` structurally
+ * satisfies this, so `LLMock.reset()` and the control-API full-reset route
+ * share a single definition of "everything" instead of two lists that drift.
+ */
+
 interface ServiceFixtures {
   search: SearchFixture[];
   rerank: RerankFixture[];
@@ -30,7 +36,6 @@ declare function createServer(fixtures: Fixture[], options?: MockServerOptions,
   handler: Mountable;
 }>, serviceFixtures?: ServiceFixtures): Promise<ServerInstance>;
 /** @internal Config startup uses this to avoid parsing a selected auth source twice. */
-
 //#endregion
 export { ServerInstance, ServiceFixtures, createServer };
 //# sourceMappingURL=server.d.ts.map

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;;AACgB,UAyqCf,eAAA,CAzqCe;EAyqCf,MAAA,EACP,aADsB,EAAA;EAAA,MAAA,EAEtB,aAFsB,EAAA;YACtB,EAEI,iBAFJ,EAAA;;AAEI,iBAMQ,YAAA,CANR,QAAA,EAOF,OAPE,EAAA,EAAA,OAAA,CAAA,EAQF,iBARE,EAAA,MAMoB,CANpB,EASH,KATG,CAAA;EAAiB,IAAA,EAAA,MAAA;EAMT,OAAA,EAGoB,SAHR;CAAA,CAAA,EAAA,eAAA,CAAA,EAId,eAJc,CAAA,EAK/B,OAL+B,CAKvB,cALuB,CAAA"}
+{"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;;;AA2rChB;;;;;UAAiB,eAAA;UACP;UACA;cACI;;iBAMQ,YAAA,WACV,qBACA,4BACD;;WAA+B;sBACtB,kBACjB,QAAQ"}

package/dist/server.js

@@ -143,23 +143,31 @@ function handleNotFound(res, message) {
 }
 const CONTROL_PREFIX = "/__aimock";
 /**
-* Perform a full fixtures reset: clear the fixtures array, journal, video/fal
-* generation state, and the interaction/event-id counters, then zero the
-* `aimock_fixtures_loaded` gauge. Shared by `/reset/fixtures` and the
-* deprecated `/reset` alias.
+* Perform a full reset: clear the fixtures array, the journal (entries *and*
+* per-test fixture match-counts, i.e. sequence position), the video and fal.ai
+* job/queue state, and the Gemini interaction/event-id counters, then re-zero
+* the `aimock_fixtures_loaded` gauge.
+*
+* `targets` is `null` when no server is running (an in-process `reset()` before
+* `start()`). The process-global generation state is reset either way, since it
+* is not owned by any one server instance.
+*
+* Shared by `POST /__aimock/reset` (canonical), its deprecated
+* `POST /__aimock/reset/fixtures` alias, and `LLMock.reset()`.
 */
-function performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults) {
+function performFullReset(fixtures, targets) {
 	fixtures.length = 0;
-	journal.clear();
-	videoStates.clear();
-	openRouterVideoJobs.clear();
-	veoVideoJobs.clear();
-	grokVideoJobs.clear();
 	falJobs.clear();
 	falQueueStates.clear();
 	resetInteractionCounter();
 	resetEventIdCounter();
-	if (defaults.registry) defaults.registry.setGauge("aimock_fixtures_loaded", {}, fixtures.length);
+	if (!targets) return;
+	targets.journal.clear();
+	targets.videoStates.clear();
+	targets.openRouterVideoJobs.clear();
+	targets.veoVideoJobs.clear();
+	targets.grokVideoJobs.clear();
+	if (targets.defaults.registry) targets.defaults.registry.setGauge("aimock_fixtures_loaded", {}, fixtures.length);
 }
 /**
 * Handle requests under `/__aimock/`. Returns `true` if the request was
@@ -179,6 +187,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 {
@@ -228,8 +241,16 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		res.end(JSON.stringify({ cleared: true }));
 		return true;
 	}
-	if (subPath === "/reset/fixtures" && req.method === "POST") {
-		performFixturesReset(fixtures, journal, videoStates, openRouterVideoJobs, veoVideoJobs, grokVideoJobs, defaults);
+	const resetTargets = () => ({
+		journal,
+		videoStates,
+		openRouterVideoJobs,
+		veoVideoJobs,
+		grokVideoJobs,
+		defaults
+	});
+	if (subPath === "/reset" && req.method === "POST") {
+		performFullReset(fixtures, resetTargets());
 		res.writeHead(200, { "Content-Type": "application/json" });
 		res.end(JSON.stringify({ reset: true }));
 		return true;
@@ -240,10 +261,10 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 		res.end(JSON.stringify({ reset: true }));
 		return true;
 	}
-	if (subPath === "/reset" && req.method === "POST") {
-		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");
+	if (subPath === "/reset/fixtures" && req.method === "POST") {
+		performFullReset(fixtures, resetTargets());
+		const deprecation = "POST /__aimock/reset/fixtures is deprecated; use POST /__aimock/reset (full reset) or POST /__aimock/reset/journal (journal only)";
+		defaults.logger.warn("POST /__aimock/reset/fixtures is deprecated; use /__aimock/reset or /__aimock/reset/journal");
 		res.writeHead(200, {
 			"Content-Type": "application/json",
 			Deprecation: "true"
@@ -1837,7 +1858,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) {
@@ -1936,5 +1998,5 @@ async function createServerWithResolvedAuth(fixtures, options, resolvedAuth, mou
 }
 
 //#endregion
-export { createServer, createServerWithResolvedAuth };
+export { createServer, createServerWithResolvedAuth, performFullReset };
 //# sourceMappingURL=server.js.map