@copilotkit/aimock 1.45.0 → 1.46.0

package/dist/messages.js

@@ -1,10 +1,15 @@
-import { estimatePromptTokens, estimateTokens, extractOverrides, flattenHeaders, generateMessageId, generateToolUseId, getContext, getTestId, isCombinedFixtureResponse, isErrorResponse, isJsonObject, isTextResponse, isToolCallResponse, requireEmittedFunctionToolCalls, resolveFixtureBlockOutcome, resolveReasoningArtifactsForModel, resolveReasoningForModel, resolveResponse, resolveStrictMode, servedToolArgs, strictNoMatchLogLine, strictNoMatchMessage, strictOverrideField, toolArgsForWire } from "./helpers.js";
+import { missingRequiredHeader } from "./response/profiles.js";
+import { renderError } from "./response/errors.js";
+import { resolvePolicy, resolveRequestDecisions, strictOverrideField } from "./response/policy.js";
+import { estimatePromptTokens, estimateTokens, extractOverrides, flattenHeaders, generateMessageId, generateToolUseId, getContext, getTestId, isCombinedFixtureResponse, isErrorResponse, isJsonObject, isTextResponse, isToolCallResponse, requireEmittedFunctionToolCalls, resolveFixtureBlockOutcome, resolveReasoningArtifactsForModel, resolveReasoningForModel, resolveRequestId, resolveResponse, servedToolArgs, strictNoMatchLogLine, strictNoMatchMessage, toolArgsForWire } from "./helpers.js";
 import { calculateDelay, delay, writeErrorResponse } from "./sse-writer.js";
 import { applyChaosAsync } from "./chaos.js";
 import { planMisbehavior, recordMisbehaviorOutcome } from "./misbehavior.js";
 import { matchFixtureDiagnostic, recordMatchOptions } from "./router.js";
+import { render, resolve } from "./response/usage/codecs/anthropic.js";
 import { proxyAndRecord } from "./recorder.js";
 import { createInterruptionSignal } from "./interruption.js";
+import { mapAnthropicFinish } from "./response/finish.js";
 
 //#region src/messages.ts
 /**
@@ -191,20 +196,7 @@ function thinkingViolationMessage(v) {
 }
 function claudeStopReason(finishReason, defaultReason) {
 	if (!finishReason) return defaultReason;
-	if (finishReason === "stop") return "end_turn";
-	if (finishReason === "tool_calls") return "tool_use";
-	if (finishReason === "length") return "max_tokens";
-	return finishReason;
-}
-function claudeUsage(overrides) {
-	if (!overrides?.usage) return {
-		input_tokens: 0,
-		output_tokens: 0
-	};
-	return {
-		input_tokens: overrides.usage.input_tokens ?? overrides.usage.prompt_tokens ?? 0,
-		output_tokens: overrides.usage.output_tokens ?? overrides.usage.completion_tokens ?? 0
-	};
+	return mapAnthropicFinish(finishReason);
 }
 /**
 * Emit faithful Anthropic `redacted_thinking` content-block events at the head
@@ -255,7 +247,7 @@ function buildClaudeTextStreamEvents(content, model, chunkSize, reasoning, overr
 			model: effectiveModel,
 			stop_reason: null,
 			stop_sequence: null,
-			usage: claudeUsage(overrides)
+			usage: render(resolve(overrides), "messages-start")
 		}
 	});
 	let blockIndex = 0;
@@ -324,7 +316,7 @@ function buildClaudeTextStreamEvents(content, model, chunkSize, reasoning, overr
 			stop_reason: claudeStopReason(overrides?.finishReason, "end_turn"),
 			stop_sequence: null
 		},
-		usage: { output_tokens: claudeUsage(overrides).output_tokens }
+		usage: render(resolve(overrides), "messages-delta")
 	});
 	events.push({ type: "message_stop" });
 	return events;
@@ -344,7 +336,7 @@ function buildClaudeToolCallStreamEvents(toolCalls, model, chunkSize, logger, re
 			model: effectiveModel,
 			stop_reason: null,
 			stop_sequence: null,
-			usage: claudeUsage(overrides)
+			usage: render(resolve(overrides), "messages-start")
 		}
 	});
 	let blockIndex = 0;
@@ -420,7 +412,7 @@ function buildClaudeToolCallStreamEvents(toolCalls, model, chunkSize, logger, re
 			stop_reason: claudeStopReason(overrides?.finishReason, "tool_use"),
 			stop_sequence: null
 		},
-		usage: { output_tokens: claudeUsage(overrides).output_tokens }
+		usage: render(resolve(overrides), "messages-delta")
 	});
 	events.push({ type: "message_stop" });
 	return events;
@@ -463,7 +455,7 @@ function buildClaudeTextResponse(content, model, reasoning, overrides, reasoning
 		model: overrides?.model ?? model,
 		stop_reason: claudeStopReason(overrides?.finishReason, "end_turn"),
 		stop_sequence: null,
-		usage: claudeUsage(overrides)
+		usage: render(resolve(overrides), "messages-final")
 	};
 }
 function buildClaudeToolCallResponse(toolCalls, model, logger, reasoning, overrides, reasoningSignature, redactedThinking) {
@@ -491,7 +483,7 @@ function buildClaudeToolCallResponse(toolCalls, model, logger, reasoning, overri
 		model: overrides?.model ?? model,
 		stop_reason: claudeStopReason(overrides?.finishReason, "tool_use"),
 		stop_sequence: null,
-		usage: claudeUsage(overrides)
+		usage: render(resolve(overrides), "messages-final")
 	};
 }
 function buildClaudeContentWithToolCallsStreamEvents(content, toolCalls, model, chunkSize, logger, reasoning, overrides, reasoningSignature, redactedThinking, blocks) {
@@ -509,7 +501,7 @@ function buildClaudeContentWithToolCallsStreamEvents(content, toolCalls, model,
 			model: effectiveModel,
 			stop_reason: null,
 			stop_sequence: null,
-			usage: claudeUsage(overrides)
+			usage: render(resolve(overrides), "messages-start")
 		}
 	});
 	let blockIndex = 0;
@@ -613,7 +605,7 @@ function buildClaudeContentWithToolCallsStreamEvents(content, toolCalls, model,
 				stop_reason: claudeStopReason(overrides?.finishReason, outcome.hasToolCalls ? "tool_use" : "end_turn"),
 				stop_sequence: null
 			},
-			usage: { output_tokens: claudeUsage(overrides).output_tokens }
+			usage: render(resolve(overrides), "messages-delta")
 		});
 		events.push({ type: "message_stop" });
 		return events;
@@ -678,7 +670,7 @@ function buildClaudeContentWithToolCallsStreamEvents(content, toolCalls, model,
 			stop_reason: claudeStopReason(overrides?.finishReason, "tool_use"),
 			stop_sequence: null
 		},
-		usage: { output_tokens: claudeUsage(overrides).output_tokens }
+		usage: render(resolve(overrides), "messages-delta")
 	});
 	events.push({ type: "message_stop" });
 	return events;
@@ -729,7 +721,7 @@ function buildClaudeContentWithToolCallsResponse(content, toolCalls, model, logg
 		model: overrides?.model ?? model,
 		stop_reason: claudeStopReason(overrides?.finishReason, blockHasTools === false ? "end_turn" : "tool_use"),
 		stop_sequence: null,
-		usage: claudeUsage(overrides)
+		usage: render(resolve(overrides), "messages-final")
 	};
 }
 /**
@@ -783,10 +775,11 @@ function prepareClaudeMisbehavior(plan, context) {
 	const reasoning = plan.summary.fault === "reasoning-only" ? plan.reasoning ?? "" : resolveReasoningForModel(response.reasoning, context.request.model, context.strict, context.logger);
 	const artifacts = resolveReasoningArtifactsForModel(response.reasoningSignature, response.redactedThinking, context.request.model, context.strict, context.logger);
 	const content = outcome?.content ?? ("content" in response ? response.content ?? "" : "");
-	const usage = {
-		input_tokens: estimatePromptTokens(context.request.messages),
-		output_tokens: estimateTokens(content + servedToolCalls.map((call) => call.name + call.arguments).join("") + (reasoning ?? "") + (plan.refusal ?? ""))
-	};
+	const resolvedUsage = resolve(void 0, {
+		inputTokens: estimatePromptTokens(context.request.messages),
+		outputTokens: estimateTokens(content + servedToolCalls.map((call) => call.name + call.arguments).join("") + (reasoning ?? "") + (plan.refusal ?? ""))
+	});
+	const usage = render(resolvedUsage, "messages-final", "applied-misbehavior");
 	const overrides = {
 		...extractOverrides(response),
 		...plan.stop === "length" ? { finishReason: "length" } : {},
@@ -857,7 +850,7 @@ function prepareClaudeMisbehavior(plan, context) {
 				stop_reason: "max_tokens",
 				stop_sequence: null
 			},
-			usage: { output_tokens: usage.output_tokens }
+			usage: render(resolvedUsage, "messages-delta", "applied-misbehavior")
 		}, { type: "message_stop" });
 		return {
 			stream: true,
@@ -908,7 +901,7 @@ function prepareClaudeMisbehavior(plan, context) {
 						stop_sequence: null,
 						...detail
 					},
-					usage: { output_tokens: usage.output_tokens }
+					usage: render(resolvedUsage, "messages-delta", "applied-misbehavior")
 				},
 				{ type: "message_stop" }
 			],
@@ -958,29 +951,66 @@ async function writeClaudeSSEStream(res, events, optionsOrLatency) {
 	const latency = opts.latency ?? 0;
 	const profile = opts.streamingProfile;
 	const { recordedTimings, replaySpeed } = opts;
-	const signal = opts.signal;
 	const onChunkSent = opts.onChunkSent;
 	if (res.writableEnded) return true;
-	res.setHeader("Content-Type", "text/event-stream");
-	res.setHeader("Cache-Control", "no-cache");
-	res.setHeader("Connection", "keep-alive");
-	let chunkIndex = 0;
-	for (const event of events) {
-		const chunkDelay = calculateDelay(chunkIndex, profile, latency, recordedTimings, replaySpeed);
-		if (chunkDelay > 0) await delay(chunkDelay, signal);
-		if (signal?.aborted) return false;
-		if (res.writableEnded) return true;
-		res.write(`event: ${event.type}\ndata: ${JSON.stringify(event)}\n\n`);
-		onChunkSent?.();
-		if (signal?.aborted) return false;
-		chunkIndex++;
+	if (res.destroyed) return false;
+	const controller = new AbortController();
+	const signal = controller.signal;
+	const abort = () => controller.abort();
+	const onClose = () => {
+		if (!res.writableFinished) abort();
+	};
+	res.once("close", onClose);
+	opts.signal?.addEventListener("abort", abort, { once: true });
+	if (opts.signal?.aborted) abort();
+	try {
+		res.setHeader("Content-Type", "text/event-stream");
+		res.setHeader("Cache-Control", "no-cache");
+		res.setHeader("Connection", "keep-alive");
+		let chunkIndex = 0;
+		for (const event of events) {
+			const chunkDelay = calculateDelay(chunkIndex, profile, latency, recordedTimings, replaySpeed);
+			if (chunkDelay > 0) await delay(chunkDelay, signal);
+			if (signal.aborted || res.destroyed) return false;
+			if (res.writableEnded) return true;
+			res.write(`event: ${event.type}\ndata: ${JSON.stringify(event)}\n\n`);
+			onChunkSent?.();
+			if (signal.aborted || res.destroyed) return false;
+			chunkIndex++;
+		}
+		if (!res.writableEnded) res.end();
+		return true;
+	} finally {
+		res.off("close", onClose);
+		opts.signal?.removeEventListener("abort", abort);
 	}
-	if (!res.writableEnded) res.end();
-	return true;
 }
 async function handleMessages(req, res, raw, fixtures, journal, defaults, setCorsHeaders) {
 	const { logger } = defaults;
 	setCorsHeaders(res);
+	const missingHeader = resolvePolicy(defaults, req.headers).requiredHeaders === "enforce" ? missingRequiredHeader("messages", req.headers) : void 0;
+	if (missingHeader) {
+		const { id: requestId } = resolveRequestId(req.headers);
+		req.headers["x-request-id"] = requestId;
+		res.setHeader("X-Request-Id", requestId);
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "/v1/messages",
+			headers: flattenHeaders(req.headers),
+			body: null,
+			response: {
+				status: 400,
+				fixture: null,
+				source: "internal",
+				...strictOverrideField(defaults.strict, req.headers)
+			}
+		});
+		writeErrorResponse(res, 400, renderError("messages", "thinking", {
+			message: `${missingHeader}: header is required`,
+			requestId
+		}).body);
+		return;
+	}
 	let claudeReq;
 	try {
 		claudeReq = JSON.parse(raw);
@@ -996,10 +1026,10 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("messages", "validation", {
 			message: `Malformed JSON: ${detail}`,
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	if (!isJsonObject(claudeReq)) {
@@ -1013,10 +1043,10 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("messages", "validation", {
 			message: "Request body must be a JSON object",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const messagesError = !Array.isArray(claudeReq.messages) ? "messages must be an array" : claudeReq.messages.some((message) => !isJsonObject(message)) ? "messages entries must be JSON objects" : claudeReq.messages.some((message) => (message.role === "user" || message.role === "assistant") && typeof message.content === "number") ? "messages content must be a string or an array of content blocks" : null;
@@ -1031,15 +1061,15 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("messages", "validation", {
 			message: messagesError,
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const thinkingViolation = validateThinkingInvariants(claudeReq);
 	if (thinkingViolation) {
-		const effectiveStrict = resolveStrictMode(defaults.strict, req.headers);
+		const effectiveStrict = resolveRequestDecisions(defaults.strict, req.headers).thinkingRejection;
 		const violationMessage = thinkingViolationMessage(thinkingViolation);
 		if (effectiveStrict) {
 			logger.error(`THINKING: ${violationMessage}`);
@@ -1054,13 +1084,7 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 					...strictOverrideField(defaults.strict, req.headers)
 				}
 			});
-			writeErrorResponse(res, 400, JSON.stringify({
-				type: "error",
-				error: {
-					type: "invalid_request_error",
-					message: violationMessage
-				}
-			}));
+			writeErrorResponse(res, 400, renderError("messages", "thinking", { message: violationMessage }).body);
 			return;
 		}
 		logger.warn(`THINKING: ${violationMessage} (strict off — replaying anyway)`);
@@ -1078,10 +1102,10 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 					fixture: null
 				}
 			});
-			writeErrorResponse(res, 400, JSON.stringify({ error: {
+			writeErrorResponse(res, 400, renderError("messages", "validation", {
 				message: toolsError,
 				type: "invalid_request_error"
-			} }));
+			}).body);
 			return;
 		}
 	}
@@ -1104,7 +1128,7 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 		body: completionReq
 	}, fixture ? "fixture" : "proxy", defaults.registry, defaults.logger)) return;
 	if (!fixture) {
-		if (resolveStrictMode(defaults.strict, req.headers)) {
+		if (resolveRequestDecisions(defaults.strict, req.headers).fixtureMiss) {
 			const strictStatus = 503;
 			const strictMessage = strictNoMatchMessage(skippedBySequenceOrTurn);
 			logger.error(strictNoMatchLogLine(req.method ?? "POST", req.url ?? "/v1/messages", skippedBySequenceOrTurn));
@@ -1119,10 +1143,10 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 					...strictOverrideField(defaults.strict, req.headers)
 				}
 			});
-			writeErrorResponse(res, strictStatus, JSON.stringify({ error: {
+			writeErrorResponse(res, strictStatus, renderError("messages", "strict-miss", {
 				message: strictMessage,
 				type: "invalid_request_error"
-			} }));
+			}).body);
 			return;
 		}
 		if (defaults.record) {
@@ -1154,10 +1178,10 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 				...strictOverrideField(defaults.strict, req.headers)
 			}
 		});
-		writeErrorResponse(res, 404, JSON.stringify({ error: {
+		writeErrorResponse(res, 404, renderError("messages", "fixture-miss", {
 			message: "No fixture matched",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const response = await resolveResponse(fixture, completionReq);
@@ -1189,14 +1213,11 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			testId
 		});
 		if (!misbehavior.summary?.evaluations.length) logger.error(`POST /v1/messages: ${misbehavior.message}`);
-		writeErrorResponse(res, misbehavior.status, JSON.stringify({
-			type: "error",
-			error: {
-				type: misbehavior.status === 400 ? "invalid_request_error" : "api_error",
-				code: misbehavior.code,
-				message: misbehavior.message
-			}
-		}));
+		writeErrorResponse(res, misbehavior.status, renderError("messages", "misbehavior", {
+			type: misbehavior.status === 400 ? "invalid_request_error" : "api_error",
+			code: misbehavior.code,
+			message: misbehavior.message
+		}).body);
 		return;
 	}
 	const latency = fixture.latency ?? defaults.latency;
@@ -1207,7 +1228,7 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			stream: claudeReq.stream === true,
 			chunkSize,
 			logger,
-			strict: resolveStrictMode(defaults.strict, req.headers)
+			strict: resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression
 		});
 		const entry = journal.add({
 			method: req.method ?? "POST",
@@ -1241,7 +1262,7 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 				})) {
 					if (!res.writableEnded) res.destroy();
 					entry.response.interrupted = true;
-					entry.response.interruptReason = interruption?.reason();
+					entry.response.interruptReason = interruption?.reason() ?? "client aborted";
 				}
 			} finally {
 				interruption?.cleanup();
@@ -1266,20 +1287,13 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			defaults,
 			testId
 		});
-		const anthropicError = {
-			type: "error",
-			error: {
-				type: response.error.type ?? "api_error",
-				message: response.error.message
-			}
-		};
-		writeErrorResponse(res, status, JSON.stringify(anthropicError), { retryAfter: response.retryAfter });
+		writeErrorResponse(res, status, renderError("messages", "fixture", response.error).body, { retryAfter: response.retryAfter });
 		return;
 	}
 	if (isCombinedFixtureResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Claude Messages API — ignoring");
 		const overrides = extractOverrides(response);
-		const effectiveStrict = resolveStrictMode(defaults.strict, req.headers);
+		const effectiveStrict = resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression;
 		const effReasoning = resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger);
 		const { reasoningSignature: effReasoningSignature, redactedThinking: effRedactedThinking } = resolveReasoningArtifactsForModel(response.reasoningSignature, response.redactedThinking, completionReq.model, effectiveStrict, defaults.logger);
 		const journalEntry = journal.add({
@@ -1316,7 +1330,7 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			})) {
 				if (!res.writableEnded) res.destroy();
 				journalEntry.response.interrupted = true;
-				journalEntry.response.interruptReason = interruption?.reason();
+				journalEntry.response.interruptReason = interruption?.reason() ?? "client aborted";
 			}
 			interruption?.cleanup();
 		}
@@ -1325,7 +1339,7 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 	if (isTextResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Claude Messages API — ignoring");
 		const overrides = extractOverrides(response);
-		const effectiveStrict = resolveStrictMode(defaults.strict, req.headers);
+		const effectiveStrict = resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression;
 		const effReasoning = resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger);
 		const { reasoningSignature: effReasoningSignature, redactedThinking: effRedactedThinking } = resolveReasoningArtifactsForModel(response.reasoningSignature, response.redactedThinking, completionReq.model, effectiveStrict, defaults.logger);
 		const journalEntry = journal.add({
@@ -1361,7 +1375,7 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			})) {
 				if (!res.writableEnded) res.destroy();
 				journalEntry.response.interrupted = true;
-				journalEntry.response.interruptReason = interruption?.reason();
+				journalEntry.response.interruptReason = interruption?.reason() ?? "client aborted";
 			}
 			interruption?.cleanup();
 		}
@@ -1370,7 +1384,7 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 	if (isToolCallResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Claude Messages API — ignoring");
 		const overrides = extractOverrides(response);
-		const effectiveStrict = resolveStrictMode(defaults.strict, req.headers);
+		const effectiveStrict = resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression;
 		const effReasoning = resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger);
 		const { reasoningSignature: effReasoningSignature, redactedThinking: effRedactedThinking } = resolveReasoningArtifactsForModel(response.reasoningSignature, response.redactedThinking, completionReq.model, effectiveStrict, defaults.logger);
 		const journalEntry = journal.add({
@@ -1407,7 +1421,7 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			})) {
 				if (!res.writableEnded) res.destroy();
 				journalEntry.response.interrupted = true;
-				journalEntry.response.interruptReason = interruption?.reason();
+				journalEntry.response.interruptReason = interruption?.reason() ?? "client aborted";
 			}
 			interruption?.cleanup();
 		}
@@ -1428,12 +1442,12 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 		defaults,
 		testId
 	});
-	writeErrorResponse(res, 500, JSON.stringify({ error: {
+	writeErrorResponse(res, 500, renderError("messages", "unknown-response", {
 		message: "Fixture response did not match any known type",
 		type: "server_error"
-	} }));
+	}).body);
 }
 
 //#endregion
-export { handleMessages, prepareClaudeMisbehavior };
+export { claudeToCompletionRequest, handleMessages, prepareClaudeMisbehavior };
 //# sourceMappingURL=messages.js.map