@copilotkit/aimock 1.46.0 → 1.47.0

package/dist/server.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.cts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;AA8X+B,UAAd,cAAA,CAAc;mBAEhB,CAAA,MAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;QACJ,EADD,MAAA,CAAK,MACJ;SAEC,EAFD,OAEC;KACG,EAAA,MAAA;UACQ,EAFX,eAEW;aACP,EAFD,aAEC;qBACC,EAFM,qBAEN;cACI,EAFL,cAEK;EAAmB,aAAA,EADvB,eACuB;EAo9EvB,iBAAA,EAp9EI,mBAo9EW;;;;;;;;UAAf,eAAA;UACP;UACA;cACI;;aAED;;iBAMS,YAAA,WACV,qBACA,4BACD;;WAA+B;sBACtB,kBACjB,QAAQ"}
+{"version":3,"file":"server.d.cts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;AA8X+B,UAAd,cAAA,CAAc;mBAEhB,CAAA,MAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;QACJ,EADD,MAAA,CAAK,MACJ;SAEC,EAFD,OAEC;KACG,EAAA,MAAA;UACQ,EAFX,eAEW;aACP,EAFD,aAEC;qBACC,EAFM,qBAEN;cACI,EAFL,cAEK;EAAmB,aAAA,EADvB,eACuB;EA88EvB,iBAAA,EA98EI,mBA88EW;;;;;;;;UAAf,eAAA;UACP;UACA;cACI;;aAED;;iBAMS,YAAA,WACV,qBACA,4BACD;;WAA+B;sBACtB,kBACjB,QAAQ"}

package/dist/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;AA8X+B,UAAd,cAAA,CAAc;mBAEhB,CAAA,MAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;QACJ,EADD,MAAA,CAAK,MACJ;SAEC,EAFD,OAEC;KACG,EAAA,MAAA;UACQ,EAFX,eAEW;aACP,EAFD,aAEC;qBACC,EAFM,qBAEN;cACI,EAFL,cAEK;EAAmB,aAAA,EADvB,eACuB;EAo9EvB,iBAAA,EAp9EI,mBAo9EW;;;;;;;;UAAf,eAAA;UACP;UACA;cACI;;aAED;;iBAMS,YAAA,WACV,qBACA,4BACD;;WAA+B;sBACtB,kBACjB,QAAQ"}
+{"version":3,"file":"server.d.ts","names":[],"sources":["../src/server.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;AA8X+B,UAAd,cAAA,CAAc;mBAEhB,CAAA,MAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;QACJ,EADD,MAAA,CAAK,MACJ;SAEC,EAFD,OAEC;KACG,EAAA,MAAA;UACQ,EAFX,eAEW;aACP,EAFD,aAEC;qBACC,EAFM,qBAEN;cACI,EAFL,cAEK;EAAmB,aAAA,EADvB,eACuB;EA88EvB,iBAAA,EA98EI,mBA88EW;;;;;;;;UAAf,eAAA;UACP;UACA;cACI;;aAED;;iBAMS,YAAA,WACV,qBACA,4BACD;;WAA+B;sBACtB,kBACjB,QAAQ"}

package/dist/server.js

@@ -4,9 +4,9 @@ import { DEFAULT_TEST_ID } from "./constants.js";
 import { build, msg, quote } from "./message-text.js";
 import { markAuthenticatedRequest, resolveInboundAuth, validateRequestApiKey, writeApiKeyHttpRejection, writeApiKeyUpgradeRejection } from "./api-key-auth.js";
 import { parsePolicy, resolvePolicy, resolveRequestDecisions, strictOverrideField } from "./response/policy.js";
-import { RequestBodyTooLargeError, buildContentWithToolCallsChunks, buildContentWithToolCallsCompletion, buildOpenAIContentFilterChunks, buildOpenAIContentFilterCompletion, buildOpenAIReasoningChunks, buildOpenAIReasoningCompletion, buildOpenAIRefusalChunks, buildOpenAIRefusalCompletion, buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, buildUsageChunk, createServerFixtureList, describeMatch, extractOverrides, fixtureToolCallErrorCode, flattenHeaders, getContext, getTestId, googleFixtureToolCallErrorDetails, isAudioResponse, isCombinedFixtureResponse, isEmbeddingResponse, isErrorResponse, isExtendedResponsesToolsList, isFixtureToolCallError, isImageResponse, isJSONResponse, isJsonObject, isResponseFactory, isServerOwnedFixtures, isTextResponse, isToolCallResponse, isTranscriptionResponse, isVideoResponse, markMintedRequestId, prepareOpenAIChatMisbehavior, readBody, readBodyBufferBounded, requireEmittedFunctionToolCalls, requireServedFunctionToolCalls, resolveFixtureBlockOutcome, resolveOpenAIChatMisbehaviorUsage, resolveReasoningForModel, resolveRequestId, resolveResponse, resolveTestId, resolveUsage, runWithToolArgumentsScope, serializeErrorResponse, strictNoMatchLogLine, strictNoMatchMessage, validateToolsField, withoutResponsesToolKeys, wouldProxyMiss } from "./helpers.js";
+import { RequestBodyTooLargeError, buildContentWithToolCallsChunks, buildContentWithToolCallsCompletion, buildOpenAIContentFilterChunks, buildOpenAIContentFilterCompletion, buildOpenAIReasoningChunks, buildOpenAIReasoningCompletion, buildOpenAIRefusalChunks, buildOpenAIRefusalCompletion, buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, buildUsageChunk, createServerFixtureList, describeMatch, extractOverrides, fixtureToolCallErrorCode, flattenHeaders, getContext, getTestId, googleFixtureToolCallErrorDetails, isAudioResponse, isCombinedFixtureResponse, isEmbeddingResponse, isErrorResponse, isExtendedResponsesToolsList, isFixtureToolCallError, isImageResponse, isJSONResponse, isJsonObject, isRawResponse, isResponseFactory, isServerOwnedFixtures, isTextResponse, isToolCallResponse, isTranscriptionResponse, isVideoResponse, markMintedRequestId, prepareOpenAIChatMisbehavior, readBody, readBodyBufferBounded, rejectGeneratedFiles, requireEmittedFunctionToolCalls, requireServedFunctionToolCalls, resolveFixtureBlockOutcome, resolveOpenAIChatMisbehaviorUsage, resolveReasoningForModel, resolveRequestId, resolveResponse, resolveTestId, resolveUsage, runWithToolArgumentsScope, serializeErrorResponse, strictNoMatchLogLine, strictNoMatchMessage, validateToolsField, withoutResponsesToolKeys, wouldProxyMiss } from "./helpers.js";
 import { normalizeLiveOptions } from "./live-fixture.js";
-import { calculateTotalDelay, delay, writeErrorResponse, writeSSEStream } from "./sse-writer.js";
+import { sseCodec, writeErrorResponse } from "./sse-writer.js";
 import { CHAOS_FIELDS, CHAOS_FIELD_NAMES, applyChaosAction, awaitChaosLatency, describeUnwritableReason, evaluateChaos, isChaosScope, parseChaosField, resetChaosWarnings, resolveChaosConfig, responseGoneReason } from "./chaos.js";
 import { MISBEHAVIOR_CATALOG, WIRE_SUPPORT, getFixtureMisbehaviorPosition, parseMisbehavior, parseMisbehaviorHeader, planMisbehavior, recordMisbehaviorOutcome, resolveMisbehaviorSeed, resolveMisbehaviorShortCircuit, setFixtureMisbehaviorPosition } from "./misbehavior.js";
 import { FixtureLoadError, INJECTED_STATUS_RANGE, MisbehaviorConfigError, claimOneShotError, clearFixtureQueue, enableHeldFixtureMisbehavior, entryToFixture, isInjectableStatus, isOneShotError, markFixtureResponsesToolsExtended, queueOneShotError, releaseOneShotError, responsesToolsExtendedView, responsesToolsViewSource, validateFixtures } from "./fixture-loader.js";
@@ -14,12 +14,16 @@ import { matchFixtureDiagnostic, recordMatchOptions } from "./router.js";
 import { proxyAndRecord } from "./recorder.js";
 import { Logger } from "./logger.js";
 import { Journal } from "./journal.js";
-import { createInterruptionSignal } from "./interruption.js";
+import { CLIENT_ABORTED } from "./response/lifecycle.js";
+import { extendRecordedTimings, misbehaviorPayloadLength } from "./response/timing.js";
+import { deliverJson, deliverStream } from "./response/deliver.js";
+import { proxyInterruption, withHangUpDetection } from "./response/proxy.js";
 import { collect, idsFor } from "./response/catalog/collect.js";
-import { anthropicPage } from "./response/catalog/anthropic-page.js";
-import { geminiPage } from "./response/catalog/gemini-page.js";
 import { coreModelIds, renderAnthropicModel, renderAnthropicModels, renderGeminiModel, renderGeminiModels, renderOllamaTags, renderOpenAIModel, renderOpenAIModels } from "./response/catalog/render.js";
 import { buildOpenRouterCandidates, handleOpenRouterCredits, handleOpenRouterKey, handleOpenRouterModels, isOpenRouterPath, resolveOpenRouterShaping, serializeOpenRouterError, shapeOpenRouterChunks, shapeOpenRouterCompletion } from "./openrouter-chat.js";
+import { serveRawResponse } from "./response/raw.js";
+import { anthropicPage } from "./response/catalog/anthropic-page.js";
+import { geminiPage } from "./response/catalog/gemini-page.js";
 import { handleResponses } from "./responses.js";
 import { handleMessages } from "./messages.js";
 import { handleGemini } from "./gemini.js";
@@ -211,6 +215,7 @@ function fixtureResponseKind(response) {
 	if (isCombinedFixtureResponse(response)) return "contentWithToolCalls";
 	if (isToolCallResponse(response)) return "toolCalls";
 	if (isTextResponse(response)) return "text";
+	if (isRawResponse(response)) return "raw";
 	return "unknown";
 }
 /**
@@ -896,7 +901,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 		const clientAbort = clientAbortedMidBody(req, res, err);
 		const bodyLimitDisconnect = clientFault && (req.destroyed || res.destroyed);
 		if (clientAbort || bodyLimitDisconnect) {
-			const interruptReason = bodyLimitDisconnect ? "request body exceeded size limit" : "client aborted";
+			const interruptReason = bodyLimitDisconnect ? "request body exceeded size limit" : CLIENT_ABORTED;
 			defaults.logger.warn(bodyLimitDisconnect ? `${route}: ${msg}` : `${route}: client aborted mid-body (${msg})`);
 			journal.add({
 				method: req.method ?? "POST",
@@ -1045,7 +1050,10 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 	body._endpointType = "chat";
 	body._context = getContext(req);
 	const testId = getTestId(req);
-	const matchOptions = recordMatchOptions(!!(defaults.record && providerKey), defaults.logger);
+	const matchOptions = {
+		...recordMatchOptions(!!(defaults.record && providerKey), defaults.logger),
+		wire: "openai-chat"
+	};
 	let fixture = null;
 	let skippedBySequenceOrTurn = 0;
 	let preResolvedResponse = null;
@@ -1075,10 +1083,11 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 			skippedBySequenceOrTurn = Math.max(skippedBySequenceOrTurn, attempt.skippedBySequenceOrTurn);
 			if (!attempt.fixture) return null;
 			const response = await resolveResponse(attempt.fixture, probe);
+			const isError = isErrorResponse(response) || isRawResponse(response) && (response.raw.status ?? 200) >= 400;
 			return {
 				fixture: attempt.fixture,
 				response,
-				isError: isErrorResponse(response)
+				isError
 			};
 		};
 		for (const candidate of candidates) {
@@ -1092,7 +1101,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 			matchCounts = journal.getFixtureMatchCountsForTest(testId);
 			responseModel = candidate;
 			if (outcome.isError) {
-				const errorFallsThrough = outcome.response.fallthrough !== false;
+				const errorFallsThrough = !isErrorResponse(outcome.response) || outcome.response.fallthrough !== false;
 				if (allowFallbacks && errorFallsThrough) {
 					lastErrorFixture = outcome.fixture;
 					lastErrorResponse = outcome.response;
@@ -1209,22 +1218,9 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 					});
 				}
 			} : void 0;
-			let clientHungUp = false;
-			const noteClientHangUp = () => {
-				if (!res.writableEnded) clientHungUp = true;
-			};
-			req.socket?.on("end", noteClientHangUp);
-			req.socket?.on("error", noteClientHangUp);
-			let outcome;
-			try {
-				outcome = await proxyAndRecord(req, res, body, providerKey, req.url ?? COMPLETIONS_PATH, fixtures, defaults, raw, hookOptions);
-			} finally {
-				req.socket?.off("end", noteClientHangUp);
-				req.socket?.off("error", noteClientHangUp);
-			}
+			const { outcome, clientHungUp } = await withHangUpDetection(req, res, () => proxyAndRecord(req, res, body, providerKey, req.url ?? COMPLETIONS_PATH, fixtures, defaults, raw, hookOptions));
 			if (outcome === "handled_by_hook") return;
 			if (outcome !== "not_configured") {
-				const destroyedMidStream = res.destroyed && !res.writableEnded;
 				journal.add({
 					method: req.method ?? "POST",
 					path: req.url ?? COMPLETIONS_PATH,
@@ -1234,10 +1230,7 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 						status: res.statusCode ?? 200,
 						fixture: null,
 						source: "proxy",
-						...destroyedMidStream ? {
-							interrupted: true,
-							interruptReason: clientHungUp ? "client aborted" : "proxy stream destroyed"
-						} : {}
+						...proxyInterruption(res, clientHungUp)
 					}
 				});
 				recordBypass("proxied");
@@ -1324,20 +1317,23 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 	const emitStreamingUsage = body.stream === true && body.stream_options?.include_usage === true || openRouter;
 	const streamingPromptText = body.messages.map((m) => typeof m.content === "string" ? m.content : Array.isArray(m.content) ? m.content.map((p) => p.text ?? "").join("") : "").join("");
 	const resolveStreamingUsageTokens = (overrides, completionText) => appliedUsage ?? resolveUsage(overrides, streamingPromptText, completionText);
-	const resolveRecordedTimings = (chunkCount) => {
-		const timings = fixture.recordedTimings;
-		if (!appliedPlan?.duplicateId || !timings) return timings;
-		const gaps = timings.interChunkDelaysMs;
-		return {
-			...timings,
-			interChunkDelaysMs: Array.from({ length: Math.max(gaps.length, chunkCount - 1) }, (_, index) => gaps[index] ?? gaps.at(-1) ?? 0)
-		};
-	};
-	const awaitNonStreamingTiming = (contentLength) => {
-		if (fixture.nonStreamingTiming !== true) return Promise.resolve();
-		if (!fixture.streamingProfile && !fixture.recordedTimings) return Promise.resolve();
-		const chunkCount = Math.max(1, Math.ceil(contentLength / chunkSize));
-		return delay(calculateTotalDelay(contentLength, chunkSize, fixture.streamingProfile, resolveRecordedTimings(chunkCount), fixture.replaySpeed ?? defaults.replaySpeed));
+	const nonStreamingOpts = (payloadLength) => ({
+		surface: "openai-chat",
+		payloadLength,
+		chunkSize,
+		recordedTimings: fixture.recordedTimings,
+		duplicateId: !!appliedPlan?.duplicateId,
+		replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
+	});
+	const streamTiming = (frameCount) => ({
+		latency,
+		streamingProfile: fixture.streamingProfile,
+		recordedTimings: extendRecordedTimings(fixture.recordedTimings, frameCount, !!appliedPlan?.duplicateId),
+		replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
+	});
+	const commitNonStream = async (entry, reply, payloadLength) => {
+		recordOutcome(entry);
+		await deliverJson(res, entry, fixture, () => reply, nonStreamingOpts(payloadLength));
 	};
 	const shapeORCompletion = (completion, overrides) => {
 		if (appliedUsage) completion.usage = appliedUsage;
@@ -1351,6 +1347,24 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 		if (usageChunk) shapeOpenRouterChunks([usageChunk], shaping, overrides?.id !== void 0);
 	};
 	const openRouterProcessing = !!(openRouter && fixture.openRouterProcessing);
+	if (isRawResponse(response)) {
+		await serveRawResponse({
+			wire: "openai-chat",
+			stream: body.stream === true,
+			req,
+			res,
+			fixture,
+			response,
+			body,
+			journal,
+			defaults,
+			testId,
+			misbehaviorSummary: evaluation.summary,
+			errorWire: "openai",
+			openRouter
+		});
+		return;
+	}
 	if (isErrorResponse(response)) {
 		const status = response.status ?? 500;
 		recordOutcome(journal.add({
@@ -1395,32 +1409,18 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 				fixture
 			}
 		});
-		recordOutcome(entry);
-		if (body.stream !== true) {
-			const completion = appliedPlan.stop === "refusal" ? buildOpenAIRefusalCompletion(appliedPlan.refusal ?? "", responseModel, overrides, body.messages) : appliedPlan.stop === "content_filter" ? buildOpenAIContentFilterCompletion(responseModel, overrides, body.messages) : buildOpenAIReasoningCompletion(appliedPlan.reasoning ?? "", responseModel, openRouter, overrides, body.messages);
-			await awaitNonStreamingTiming((appliedPlan.stop === "refusal" ? appliedPlan.refusal ?? "" : appliedPlan.stop === "content_filter" ? "" : appliedPlan.reasoning ?? "").length);
-			res.writeHead(200, { "Content-Type": "application/json" });
-			res.end(JSON.stringify(shapeORCompletion(completion, overrides)));
-		} else {
+		rejectGeneratedFiles(response, wire);
+		if (body.stream !== true) await commitNonStream(entry, shapeORCompletion(appliedPlan.stop === "refusal" ? buildOpenAIRefusalCompletion(appliedPlan.refusal ?? "", responseModel, overrides, body.messages) : appliedPlan.stop === "content_filter" ? buildOpenAIContentFilterCompletion(responseModel, overrides, body.messages) : buildOpenAIReasoningCompletion(appliedPlan.reasoning ?? "", responseModel, openRouter, overrides, body.messages), overrides), misbehaviorPayloadLength(appliedPlan, appliedReasoning));
+		else {
 			const chunks = appliedPlan.stop === "refusal" ? buildOpenAIRefusalChunks(appliedPlan.refusal ?? "", responseModel, chunkSize, overrides) : appliedPlan.stop === "content_filter" ? buildOpenAIContentFilterChunks(responseModel, overrides) : buildOpenAIReasoningChunks(appliedPlan.reasoning ?? "", responseModel, chunkSize, openRouter, overrides);
 			const usageChunk = emitStreamingUsage ? buildUsageChunk(chunks[0].id, overrides?.model ?? responseModel, chunks[0].created, resolveStreamingUsageTokens(overrides, ""), overrides?.systemFingerprint) : void 0;
 			shapeORChunks(chunks, usageChunk, overrides);
-			const interruption = createInterruptionSignal(fixture);
-			if (!await writeSSEStream(res, chunks, {
-				latency,
-				streamingProfile: fixture.streamingProfile,
-				signal: interruption?.signal,
-				onChunkSent: interruption?.tick,
+			recordOutcome(entry);
+			await deliverStream(res, entry, fixture, sseCodec({
+				done: true,
 				usageChunk,
-				recordedTimings: resolveRecordedTimings(chunks.length),
-				replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,
 				openRouterProcessing
-			})) {
-				if (!res.writableEnded) res.destroy();
-				entry.response.interrupted = true;
-				entry.response.interruptReason = interruption?.reason();
-			}
-			interruption?.cleanup();
+			}), chunks, streamTiming(chunks.length));
 		}
 		return;
 	}
@@ -1441,37 +1441,26 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 				fixture
 			}
 		});
-		recordOutcome(journalEntry);
+		rejectGeneratedFiles(response, wire);
 		const toolCalls = requireEmittedFunctionToolCalls(response, wire);
 		const completionText = blockOutcome ? blockOutcome.content + blockOutcome.toolCalls.map((tc) => tc.name + tc.arguments).join("") : (response.content ?? "") + toolCalls.map((tc) => tc.name + tc.arguments).join("");
 		if (!streaming) {
-			await awaitNonStreamingTiming((effReasoning?.length ?? 0) + completionText.length);
 			const completion = buildContentWithToolCallsCompletion(blockOutcome?.content ?? response.content ?? "", blockOutcome?.toolCalls ?? toolCalls, responseModel, effReasoning, blockOutcome ? {
 				...overrides,
 				finishReason: overrides?.finishReason ?? (blockOutcome.hasToolCalls ? "tool_calls" : "stop")
 			} : overrides, body.messages);
-			res.writeHead(200, { "Content-Type": "application/json" });
-			res.end(JSON.stringify(shapeORCompletion(completion, overrides)));
+			const contentLength = (effReasoning?.length ?? 0) + completionText.length;
+			await commitNonStream(journalEntry, shapeORCompletion(completion, overrides), contentLength);
 		} else {
 			const chunks = buildContentWithToolCallsChunks(blockOutcome?.content ?? response.content ?? "", blockOutcome?.toolCalls ?? toolCalls, responseModel, chunkSize, effReasoning, overrides, blockOutcome?.ordered);
 			const usageChunk = emitStreamingUsage ? buildUsageChunk(chunks[0]?.id ?? "chatcmpl-unknown", overrides?.model ?? responseModel, chunks[0]?.created ?? Math.floor(Date.now() / 1e3), resolveStreamingUsageTokens(overrides, completionText), overrides?.systemFingerprint) : void 0;
 			shapeORChunks(chunks, usageChunk, overrides);
-			const interruption = createInterruptionSignal(fixture);
-			if (!await writeSSEStream(res, chunks, {
-				latency,
-				streamingProfile: fixture.streamingProfile,
-				signal: interruption?.signal,
-				onChunkSent: interruption?.tick,
+			recordOutcome(journalEntry);
+			await deliverStream(res, journalEntry, fixture, sseCodec({
+				done: true,
 				usageChunk,
-				recordedTimings: resolveRecordedTimings(chunks.length),
-				replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,
 				openRouterProcessing
-			})) {
-				if (!res.writableEnded) res.destroy();
-				journalEntry.response.interrupted = true;
-				journalEntry.response.interruptReason = interruption?.reason();
-			}
-			interruption?.cleanup();
+			}), chunks, streamTiming(chunks.length));
 		}
 		return;
 	}
@@ -1490,32 +1479,21 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 				fixture
 			}
 		});
-		recordOutcome(journalEntry);
+		rejectGeneratedFiles(response, wire);
 		if (body.stream !== true) {
-			await awaitNonStreamingTiming((effReasoning?.length ?? 0) + response.content.length);
 			const completion = buildTextCompletion(response.content, responseModel, effReasoning, overrides, body.messages);
-			res.writeHead(200, { "Content-Type": "application/json" });
-			res.end(JSON.stringify(shapeORCompletion(completion, overrides)));
+			const contentLength = (effReasoning?.length ?? 0) + response.content.length;
+			await commitNonStream(journalEntry, shapeORCompletion(completion, overrides), contentLength);
 		} else {
 			const chunks = buildTextChunks(response.content, responseModel, chunkSize, effReasoning, overrides);
 			const usageChunk = emitStreamingUsage ? buildUsageChunk(chunks[0]?.id ?? "chatcmpl-unknown", overrides?.model ?? responseModel, chunks[0]?.created ?? Math.floor(Date.now() / 1e3), resolveStreamingUsageTokens(overrides, response.content), overrides?.systemFingerprint) : void 0;
 			shapeORChunks(chunks, usageChunk, overrides);
-			const interruption = createInterruptionSignal(fixture);
-			if (!await writeSSEStream(res, chunks, {
-				latency,
-				streamingProfile: fixture.streamingProfile,
-				signal: interruption?.signal,
-				onChunkSent: interruption?.tick,
+			recordOutcome(journalEntry);
+			await deliverStream(res, journalEntry, fixture, sseCodec({
+				done: true,
 				usageChunk,
-				recordedTimings: resolveRecordedTimings(chunks.length),
-				replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,
 				openRouterProcessing
-			})) {
-				if (!res.writableEnded) res.destroy();
-				journalEntry.response.interrupted = true;
-				journalEntry.response.interruptReason = interruption?.reason();
-			}
-			interruption?.cleanup();
+			}), chunks, streamTiming(chunks.length));
 		}
 		return;
 	}
@@ -1534,35 +1512,24 @@ async function handleCompletions(req, res, fixtures, journal, defaults, getReque
 				fixture
 			}
 		});
-		recordOutcome(journalEntry);
+		rejectGeneratedFiles(response, wire);
 		const toolCalls = requireServedFunctionToolCalls(response, wire);
 		if (body.stream !== true) {
 			const toolCallText = toolCalls.map((tc) => tc.name + tc.arguments).join("");
-			await awaitNonStreamingTiming((effReasoning?.length ?? 0) + toolCallText.length);
 			const completion = buildToolCallCompletion(toolCalls, responseModel, effReasoning, overrides, body.messages);
-			res.writeHead(200, { "Content-Type": "application/json" });
-			res.end(JSON.stringify(shapeORCompletion(completion, overrides)));
+			const contentLength = (effReasoning?.length ?? 0) + toolCallText.length;
+			await commitNonStream(journalEntry, shapeORCompletion(completion, overrides), contentLength);
 		} else {
 			const chunks = buildToolCallChunks(toolCalls, responseModel, chunkSize, effReasoning, overrides);
 			const completionText = toolCalls.map((tc) => tc.name + tc.arguments).join("");
 			const usageChunk = emitStreamingUsage ? buildUsageChunk(chunks[0]?.id ?? "chatcmpl-unknown", overrides?.model ?? responseModel, chunks[0]?.created ?? Math.floor(Date.now() / 1e3), resolveStreamingUsageTokens(overrides, completionText), overrides?.systemFingerprint) : void 0;
 			shapeORChunks(chunks, usageChunk, overrides);
-			const interruption = createInterruptionSignal(fixture);
-			if (!await writeSSEStream(res, chunks, {
-				latency,
-				streamingProfile: fixture.streamingProfile,
-				signal: interruption?.signal,
-				onChunkSent: interruption?.tick,
+			recordOutcome(journalEntry);
+			await deliverStream(res, journalEntry, fixture, sseCodec({
+				done: true,
 				usageChunk,
-				recordedTimings: resolveRecordedTimings(chunks.length),
-				replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,
 				openRouterProcessing
-			})) {
-				if (!res.writableEnded) res.destroy();
-				journalEntry.response.interrupted = true;
-				journalEntry.response.interruptReason = interruption?.reason();
-			}
-			interruption?.cleanup();
+			}), chunks, streamTiming(chunks.length));
 		}
 		return;
 	}
@@ -1822,10 +1789,12 @@ async function startServer(fixtures, options, resolvedAuth, mountList, logger, s
 		return status >= 400 && status < 500 ? "invalid_request_error" : "server_error";
 	}
 	/**
-	* Envelope for a fixture tool-call error (see isFixtureToolCallError: invalid
-	* JSON arguments on an object wire, a custom tool call on a non-Responses
-	* wire, or a malformed fixture tool call). It is the provider envelope aimock
-	* emits for that route, with the aimock `code` in it: Google's
+	* Envelope for a fixture error (see isFixtureToolCallError: invalid JSON
+	* arguments on an object wire, a custom tool call on a non-Responses wire, a
+	* malformed fixture tool call, generated `files` on a wire that cannot carry
+	* them, or a malformed fixture `citations` or `files` value). It is the
+	* provider envelope aimock emits for that route, with the aimock `code` in
+	* it: Google's
 	* `google.rpc.ErrorInfo` detail on Gemini and Vertex AI, a string
 	* `error.code` on Gemini Interactions and Anthropic Messages, a `reason`
 	* member on Bedrock (a `code` member would replace `__type` as the AWS SDK's
@@ -1955,7 +1924,7 @@ async function startServer(fixtures, options, resolvedAuth, mountList, logger, s
 		const existing = lastJournalEntryFor(req);
 		const bodyLimitDisconnect = clientFault && (req.destroyed || res.destroyed);
 		if ((clientAbort || bodyLimitDisconnect) && !res.headersSent) {
-			const interruptReason = bodyLimitDisconnect ? "request body exceeded size limit" : "client aborted";
+			const interruptReason = bodyLimitDisconnect ? "request body exceeded size limit" : CLIENT_ABORTED;
 			if (existing) {
 				if (!existing.response.interrupted) {
 					existing.response.status = 0;
@@ -2011,7 +1980,7 @@ async function startServer(fixtures, options, resolvedAuth, mountList, logger, s
 				type: errorTypeForStatus(status)
 			}).body);
 		} else {
-			const midStreamReason = clientAbort ? "client aborted" : "handler crashed mid-stream";
+			const midStreamReason = clientAbort ? CLIENT_ABORTED : "handler crashed mid-stream";
 			if (existing) {
 				if (res.writableEnded) {
 					if (existing.response.error === void 0) existing.response.error = msg;