@copilotkit/aimock 1.25.0 → 1.26.1

package/dist/server.js

@@ -1,4 +1,4 @@
-import { buildContentWithToolCallsChunks, buildContentWithToolCallsCompletion, buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, buildUsageChunk, estimatePromptTokens, estimateTokens, extractOverrides, flattenHeaders, getTestId, isAudioResponse, isContentWithToolCallsResponse, isErrorResponse, isTextResponse, isToolCallResponse, readBody, resolveResponse, resolveStrictMode, serializeErrorResponse, strictOverrideField } from "./helpers.js";
+import { buildContentWithToolCallsChunks, buildContentWithToolCallsCompletion, buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, buildUsageChunk, estimatePromptTokens, estimateTokens, extractOverrides, flattenHeaders, getContext, getTestId, isAudioResponse, isContentWithToolCallsResponse, isErrorResponse, isTextResponse, isToolCallResponse, readBody, resolveResponse, resolveStrictMode, serializeErrorResponse, strictOverrideField } from "./helpers.js";
 import { Logger } from "./logger.js";
 import { Journal } from "./journal.js";
 import { matchFixture } from "./router.js";
@@ -179,7 +179,7 @@ async function handleControlAPI(req, res, pathname, fixtures, journal, videoStat
 			res.end(JSON.stringify({ error: "Missing or invalid \"fixtures\" array" }));
 			return true;
 		}
-		const converted = parsed.fixtures.map(entryToFixture);
+		const converted = parsed.fixtures.map((e) => entryToFixture(e));
 		const errors = validateFixtures(converted).filter((i) => i.severity === "error");
 		if (errors.length > 0) {
 			res.writeHead(400, { "Content-Type": "application/json" });
@@ -341,6 +341,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 	const path = req.url ?? COMPLETIONS_PATH;
 	const flatHeaders = flattenHeaders(req.headers);
 	body._endpointType = "chat";
+	body._context = getContext(req);
 	const testId = getTestId(req);
 	const fixture = matchFixture(fixtures, body, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
 	if (fixture) {
@@ -512,7 +513,9 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 				streamingProfile: fixture.streamingProfile,
 				signal: interruption?.signal,
 				onChunkSent: interruption?.tick,
-				usageChunk
+				usageChunk,
+				recordedTimings: fixture.recordedTimings,
+				replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
 			})) {
 				if (!res.writableEnded) res.destroy();
 				journalEntry.response.interrupted = true;
@@ -556,7 +559,9 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 				streamingProfile: fixture.streamingProfile,
 				signal: interruption?.signal,
 				onChunkSent: interruption?.tick,
-				usageChunk
+				usageChunk,
+				recordedTimings: fixture.recordedTimings,
+				replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
 			})) {
 				if (!res.writableEnded) res.destroy();
 				journalEntry.response.interrupted = true;
@@ -601,7 +606,9 @@ async function handleCompletions(req, res, fixtures, journal, defaults, modelFal
 				streamingProfile: fixture.streamingProfile,
 				signal: interruption?.signal,
 				onChunkSent: interruption?.tick,
-				usageChunk
+				usageChunk,
+				recordedTimings: fixture.recordedTimings,
+				replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
 			})) {
 				if (!res.writableEnded) res.destroy();
 				journalEntry.response.interrupted = true;
@@ -635,6 +642,7 @@ async function createServer(fixtures, options, mounts, serviceFixtures) {
 	const defaults = {
 		latency: serverOptions.latency ?? 0,
 		chunkSize: Math.max(1, serverOptions.chunkSize ?? DEFAULT_CHUNK_SIZE),
+		replaySpeed: serverOptions.replaySpeed ?? 1,
 		logger,
 		get chaos() {
 			return serverOptions.chaos;