@copilotkit/aimock 1.45.0 → 1.46.0

package/dist/bedrock.cjs

@@ -1,39 +1,21 @@
+const require_errors = require('./response/errors.cjs');
+const require_policy = require('./response/policy.cjs');
 const require_helpers = require('./helpers.cjs');
 const require_sse_writer = require('./sse-writer.cjs');
 const require_chaos = require('./chaos.cjs');
 const require_misbehavior = require('./misbehavior.cjs');
 const require_router = require('./router.cjs');
+const require_anthropic = require('./response/usage/codecs/anthropic.cjs');
 const require_recorder = require('./recorder.cjs');
 const require_interruption = require('./interruption.cjs');
+const require_finish = require('./response/finish.cjs');
 const require_messages = require('./messages.cjs');
 const require_aws_event_stream = require('./aws-event-stream.cjs');
 
 //#region src/bedrock.ts
 function bedrockStopReason(overrideFinishReason, defaultReason) {
 	if (!overrideFinishReason) return defaultReason;
-	if (overrideFinishReason === "stop") return "end_turn";
-	if (overrideFinishReason === "tool_calls") return "tool_use";
-	if (overrideFinishReason === "length") return "max_tokens";
-	return overrideFinishReason;
-}
-/**
-* Build a Bedrock-style usage object from optional overrides.
-*
-* When no overrides are provided (the common case for mock fixtures),
-* returns all-zero token counts. This is intentional — aimock does not
-* attempt to estimate token usage from fixture content. Callers that
-* need realistic usage numbers should set `usage` in their fixture's
-* response overrides.
-*/
-function bedrockUsage(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 require_finish.mapAnthropicFinish(overrideFinishReason);
 }
 function extractTextContent(content) {
 	if (content === null || content === void 0) return "";
@@ -146,7 +128,7 @@ function buildBedrockTextResponse(content, model, reasoning, overrides) {
 		model: overrides?.model ?? model,
 		stop_reason: bedrockStopReason(overrides?.finishReason, "end_turn"),
 		stop_sequence: null,
-		usage: bedrockUsage(overrides)
+		usage: require_anthropic.render(require_anthropic.resolve(overrides), "bedrock-final")
 	};
 }
 function buildBedrockToolCallResponse(toolCalls, model, logger, reasoning, overrides) {
@@ -173,7 +155,7 @@ function buildBedrockToolCallResponse(toolCalls, model, logger, reasoning, overr
 		model: overrides?.model ?? model,
 		stop_reason: bedrockStopReason(overrides?.finishReason, "tool_use"),
 		stop_sequence: null,
-		usage: bedrockUsage(overrides)
+		usage: require_anthropic.render(require_anthropic.resolve(overrides), "bedrock-final")
 	};
 }
 /**
@@ -217,7 +199,7 @@ function buildBedrockBlocksResponse(blocks, model, logger, reasoning, overrides)
 		model: overrides?.model ?? model,
 		stop_reason: bedrockStopReason(overrides?.finishReason, outcome.hasToolCalls ? "tool_use" : "end_turn"),
 		stop_sequence: null,
-		usage: bedrockUsage(overrides)
+		usage: require_anthropic.render(require_anthropic.resolve(overrides), "bedrock-final")
 	};
 }
 async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults, setCorsHeaders) {
@@ -239,10 +221,10 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 				fixture: null
 			}
 		});
-		require_sse_writer.writeErrorResponse(res, 400, JSON.stringify({ error: {
+		require_sse_writer.writeErrorResponse(res, 400, require_errors.renderError("bedrock", "validation", {
 			message: `Malformed JSON: ${detail}`,
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	if (!require_helpers.isJsonObject(bedrockReq)) {
@@ -256,10 +238,10 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 				fixture: null
 			}
 		});
-		require_sse_writer.writeErrorResponse(res, 400, JSON.stringify({ error: {
+		require_sse_writer.writeErrorResponse(res, 400, require_errors.renderError("bedrock", "validation", {
 			message: "Request body must be a JSON object",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const bedrockShapeError = require_helpers.validateChatMessages(bedrockReq.messages, { checkToolCalls: false }) ?? require_helpers.validateToolsField(bedrockReq.tools) ?? (bedrockReq.system !== void 0 && bedrockReq.system !== null && typeof bedrockReq.system !== "string" && !Array.isArray(bedrockReq.system) ? "system must be a string or an array" : null);
@@ -274,10 +256,10 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 				fixture: null
 			}
 		});
-		require_sse_writer.writeErrorResponse(res, 400, JSON.stringify({ error: {
+		require_sse_writer.writeErrorResponse(res, 400, require_errors.renderError("bedrock", "validation", {
 			message: `Invalid request: ${bedrockShapeError}`,
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const completionReq = bedrockToCompletionRequest(bedrockReq, modelId, logger);
@@ -295,7 +277,7 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 		body: completionReq
 	}, fixture ? "fixture" : "proxy", defaults.registry, defaults.logger)) return;
 	if (!fixture) {
-		if (require_helpers.resolveStrictMode(defaults.strict, req.headers)) {
+		if (require_policy.resolveRequestDecisions(defaults.strict, req.headers).fixtureMiss) {
 			const strictStatus = 503;
 			const strictMessage = require_helpers.strictNoMatchMessage(skippedBySequenceOrTurn);
 			logger.error(require_helpers.strictNoMatchLogLine(req.method ?? "POST", urlPath, skippedBySequenceOrTurn));
@@ -307,13 +289,13 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 				response: {
 					status: strictStatus,
 					fixture: null,
-					...require_helpers.strictOverrideField(defaults.strict, req.headers)
+					...require_policy.strictOverrideField(defaults.strict, req.headers)
 				}
 			});
-			require_sse_writer.writeErrorResponse(res, strictStatus, JSON.stringify({ error: {
+			require_sse_writer.writeErrorResponse(res, strictStatus, require_errors.renderError("bedrock", "strict-miss", {
 				message: strictMessage,
 				type: "invalid_request_error"
-			} }));
+			}).body);
 			return;
 		}
 		if (defaults.record) {
@@ -342,13 +324,13 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 			response: {
 				status: 404,
 				fixture: null,
-				...require_helpers.strictOverrideField(defaults.strict, req.headers)
+				...require_policy.strictOverrideField(defaults.strict, req.headers)
 			}
 		});
-		require_sse_writer.writeErrorResponse(res, 404, JSON.stringify({ error: {
+		require_sse_writer.writeErrorResponse(res, 404, require_errors.renderError("bedrock", "fixture-miss", {
 			message: "No fixture matched",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const response = await require_helpers.resolveResponse(fixture, completionReq);
@@ -385,13 +367,13 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 				fixture
 			}
 		});
-		const type = misbehavior.status === 400 ? "ValidationException" : "NotImplementedException";
-		res.setHeader("x-amzn-errortype", type);
-		require_sse_writer.writeErrorResponse(res, misbehavior.status, JSON.stringify({
-			__type: type,
+		const renderedError = require_errors.renderError("bedrock", "misbehavior", {
+			type: misbehavior.status === 400 ? "ValidationException" : "NotImplementedException",
 			code: misbehavior.code,
 			message: misbehavior.message
-		}));
+		});
+		for (const [name, value] of Object.entries(renderedError.headers ?? {})) res.setHeader(name, value);
+		require_sse_writer.writeErrorResponse(res, misbehavior.status, renderedError.body);
 		return;
 	}
 	if (misbehavior.kind === "applied") {
@@ -400,7 +382,7 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 			stream: false,
 			chunkSize: Math.max(1, fixture.chunkSize ?? defaults.chunkSize),
 			logger,
-			strict: require_helpers.resolveStrictMode(defaults.strict, req.headers)
+			strict: require_policy.resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression
 		});
 		if (prepared.stream) throw new TypeError("InvokeModel requires non-stream Claude output");
 		require_misbehavior.recordMisbehaviorOutcome({
@@ -434,20 +416,13 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 				fixture
 			}
 		});
-		const anthropicError = {
-			type: "error",
-			error: {
-				type: response.error.type ?? "api_error",
-				message: response.error.message
-			}
-		};
-		require_sse_writer.writeErrorResponse(res, status, JSON.stringify(anthropicError), { retryAfter: response.retryAfter });
+		require_sse_writer.writeErrorResponse(res, status, require_errors.renderError("bedrock", "fixture", response.error).body, { retryAfter: response.retryAfter });
 		return;
 	}
 	if (require_helpers.isCombinedFixtureResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Bedrock API — ignoring");
 		const overrides = require_helpers.extractOverrides(response);
-		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_helpers.resolveStrictMode(defaults.strict, req.headers), logger);
+		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_policy.resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression, logger);
 		addResponseEntry({
 			method: req.method ?? "POST",
 			path: urlPath,
@@ -479,7 +454,7 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 	if (require_helpers.isTextResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Bedrock API — ignoring");
 		const overrides = require_helpers.extractOverrides(response);
-		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_helpers.resolveStrictMode(defaults.strict, req.headers), logger);
+		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_policy.resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression, logger);
 		addResponseEntry({
 			method: req.method ?? "POST",
 			path: urlPath,
@@ -498,7 +473,7 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 	if (require_helpers.isToolCallResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Bedrock API — ignoring");
 		const overrides = require_helpers.extractOverrides(response);
-		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_helpers.resolveStrictMode(defaults.strict, req.headers), logger);
+		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_policy.resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression, logger);
 		addResponseEntry({
 			method: req.method ?? "POST",
 			path: urlPath,
@@ -524,10 +499,10 @@ async function handleBedrock(req, res, raw, modelId, fixtures, journal, defaults
 			fixture
 		}
 	});
-	require_sse_writer.writeErrorResponse(res, 500, JSON.stringify({ error: {
+	require_sse_writer.writeErrorResponse(res, 500, require_errors.renderError("bedrock", "unknown-response", {
 		message: "Fixture response did not match any known type",
 		type: "server_error"
-	} }));
+	}).body);
 }
 const BEDROCK_INVOKE_STREAM_EVENT_TYPE = "chunk";
 /** Serialize Claude events as InvokeModel PayloadPart, leaving generic AWS framing unchanged. */
@@ -550,12 +525,12 @@ function buildBedrockInvokeMessageStart(model, overrides) {
 				model: overrides?.model ?? model,
 				stop_reason: null,
 				stop_sequence: null,
-				usage: bedrockUsage(overrides)
+				usage: require_anthropic.render(require_anthropic.resolve(overrides), "bedrock-start")
 			}
 		}
 	};
 }
-function buildBedrockInvokeMessageDelta(stopReason) {
+function buildBedrockInvokeMessageDelta(stopReason, overrides) {
 	return {
 		eventType: BEDROCK_INVOKE_STREAM_EVENT_TYPE,
 		payload: {
@@ -564,7 +539,7 @@ function buildBedrockInvokeMessageDelta(stopReason) {
 				stop_reason: stopReason,
 				stop_sequence: null
 			},
-			usage: { output_tokens: 0 }
+			usage: require_anthropic.render(require_anthropic.resolve(overrides), "bedrock-delta")
 		}
 	};
 }
@@ -660,7 +635,7 @@ function buildBedrockStreamTextEvents(content, model, chunkSize, reasoning, over
 			index: textBlockIndex
 		}
 	});
-	events.push(buildBedrockInvokeMessageDelta(bedrockStopReason(overrides?.finishReason, "end_turn")));
+	events.push(buildBedrockInvokeMessageDelta(bedrockStopReason(overrides?.finishReason, "end_turn"), overrides));
 	events.push(buildBedrockInvokeMessageStop());
 	return events;
 }
@@ -794,7 +769,7 @@ function buildBedrockStreamContentWithToolCallsEvents(content, toolCalls, model,
 			});
 			blockIndex++;
 		}
-		events.push(buildBedrockInvokeMessageDelta(bedrockStopReason(overrides?.finishReason, outcome.hasToolCalls ? "tool_use" : "end_turn")));
+		events.push(buildBedrockInvokeMessageDelta(bedrockStopReason(overrides?.finishReason, outcome.hasToolCalls ? "tool_use" : "end_turn"), overrides));
 		events.push(buildBedrockInvokeMessageStop());
 		return events;
 	}
@@ -871,7 +846,7 @@ function buildBedrockStreamContentWithToolCallsEvents(content, toolCalls, model,
 			}
 		});
 	}
-	events.push(buildBedrockInvokeMessageDelta(bedrockStopReason(overrides?.finishReason, "tool_use")));
+	events.push(buildBedrockInvokeMessageDelta(bedrockStopReason(overrides?.finishReason, "tool_use"), overrides));
 	events.push(buildBedrockInvokeMessageStop());
 	return events;
 }
@@ -966,7 +941,7 @@ function buildBedrockStreamToolCallEvents(toolCalls, model, chunkSize, logger, r
 			}
 		});
 	}
-	events.push(buildBedrockInvokeMessageDelta(bedrockStopReason(overrides?.finishReason, "tool_use")));
+	events.push(buildBedrockInvokeMessageDelta(bedrockStopReason(overrides?.finishReason, "tool_use"), overrides));
 	events.push(buildBedrockInvokeMessageStop());
 	return events;
 }
@@ -989,10 +964,10 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 				fixture: null
 			}
 		});
-		require_sse_writer.writeErrorResponse(res, 400, JSON.stringify({ error: {
+		require_sse_writer.writeErrorResponse(res, 400, require_errors.renderError("bedrock", "validation", {
 			message: `Malformed JSON: ${detail}`,
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	if (!require_helpers.isJsonObject(bedrockReq)) {
@@ -1006,10 +981,10 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 				fixture: null
 			}
 		});
-		require_sse_writer.writeErrorResponse(res, 400, JSON.stringify({ error: {
+		require_sse_writer.writeErrorResponse(res, 400, require_errors.renderError("bedrock", "validation", {
 			message: "Request body must be a JSON object",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const bedrockShapeError = require_helpers.validateChatMessages(bedrockReq.messages, { checkToolCalls: false }) ?? require_helpers.validateToolsField(bedrockReq.tools) ?? (bedrockReq.system !== void 0 && bedrockReq.system !== null && typeof bedrockReq.system !== "string" && !Array.isArray(bedrockReq.system) ? "system must be a string or an array" : null);
@@ -1024,10 +999,10 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 				fixture: null
 			}
 		});
-		require_sse_writer.writeErrorResponse(res, 400, JSON.stringify({ error: {
+		require_sse_writer.writeErrorResponse(res, 400, require_errors.renderError("bedrock", "validation", {
 			message: `Invalid request: ${bedrockShapeError}`,
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const completionReq = bedrockToCompletionRequest(bedrockReq, modelId, logger);
@@ -1046,7 +1021,7 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 		body: completionReq
 	}, fixture ? "fixture" : "proxy", defaults.registry, defaults.logger)) return;
 	if (!fixture) {
-		if (require_helpers.resolveStrictMode(defaults.strict, req.headers)) {
+		if (require_policy.resolveRequestDecisions(defaults.strict, req.headers).fixtureMiss) {
 			const strictStatus = 503;
 			const strictMessage = require_helpers.strictNoMatchMessage(skippedBySequenceOrTurn);
 			logger.error(require_helpers.strictNoMatchLogLine(req.method ?? "POST", urlPath, skippedBySequenceOrTurn));
@@ -1058,13 +1033,13 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 				response: {
 					status: strictStatus,
 					fixture: null,
-					...require_helpers.strictOverrideField(defaults.strict, req.headers)
+					...require_policy.strictOverrideField(defaults.strict, req.headers)
 				}
 			});
-			require_sse_writer.writeErrorResponse(res, strictStatus, JSON.stringify({ error: {
+			require_sse_writer.writeErrorResponse(res, strictStatus, require_errors.renderError("bedrock", "strict-miss", {
 				message: strictMessage,
 				type: "invalid_request_error"
-			} }));
+			}).body);
 			return;
 		}
 		if (defaults.record) {
@@ -1093,13 +1068,13 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 			response: {
 				status: 404,
 				fixture: null,
-				...require_helpers.strictOverrideField(defaults.strict, req.headers)
+				...require_policy.strictOverrideField(defaults.strict, req.headers)
 			}
 		});
-		require_sse_writer.writeErrorResponse(res, 404, JSON.stringify({ error: {
+		require_sse_writer.writeErrorResponse(res, 404, require_errors.renderError("bedrock", "fixture-miss", {
 			message: "No fixture matched",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const response = await require_helpers.resolveResponse(fixture, completionReq);
@@ -1136,13 +1111,13 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 				fixture
 			}
 		});
-		const type = misbehavior.status === 400 ? "ValidationException" : "NotImplementedException";
-		res.setHeader("x-amzn-errortype", type);
-		require_sse_writer.writeErrorResponse(res, misbehavior.status, JSON.stringify({
-			__type: type,
+		const renderedError = require_errors.renderError("bedrock", "misbehavior", {
+			type: misbehavior.status === 400 ? "ValidationException" : "NotImplementedException",
 			code: misbehavior.code,
 			message: misbehavior.message
-		}));
+		});
+		for (const [name, value] of Object.entries(renderedError.headers ?? {})) res.setHeader(name, value);
+		require_sse_writer.writeErrorResponse(res, misbehavior.status, renderedError.body);
 		return;
 	}
 	const latency = fixture.latency ?? defaults.latency;
@@ -1153,7 +1128,7 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 			stream: true,
 			chunkSize,
 			logger,
-			strict: require_helpers.resolveStrictMode(defaults.strict, req.headers)
+			strict: require_policy.resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression
 		});
 		if (!prepared.stream) throw new TypeError("InvokeModel stream requires Claude events");
 		const entry = journal.add({
@@ -1207,20 +1182,13 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 				fixture
 			}
 		});
-		const anthropicError = {
-			type: "error",
-			error: {
-				type: response.error.type ?? "api_error",
-				message: response.error.message
-			}
-		};
-		require_sse_writer.writeErrorResponse(res, status, JSON.stringify(anthropicError), { retryAfter: response.retryAfter });
+		require_sse_writer.writeErrorResponse(res, status, require_errors.renderError("bedrock", "fixture", response.error).body, { retryAfter: response.retryAfter });
 		return;
 	}
 	if (require_helpers.isCombinedFixtureResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Bedrock API — ignoring");
 		const overrides = require_helpers.extractOverrides(response);
-		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_helpers.resolveStrictMode(defaults.strict, req.headers), logger);
+		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_policy.resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression, logger);
 		const journalEntry = addResponseEntry({
 			method: req.method ?? "POST",
 			path: urlPath,
@@ -1252,7 +1220,7 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 	if (require_helpers.isTextResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Bedrock API — ignoring");
 		const overrides = require_helpers.extractOverrides(response);
-		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_helpers.resolveStrictMode(defaults.strict, req.headers), logger);
+		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_policy.resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression, logger);
 		const journalEntry = addResponseEntry({
 			method: req.method ?? "POST",
 			path: urlPath,
@@ -1283,7 +1251,7 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 	if (require_helpers.isToolCallResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Bedrock API — ignoring");
 		const overrides = require_helpers.extractOverrides(response);
-		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_helpers.resolveStrictMode(defaults.strict, req.headers), logger);
+		const effReasoning = require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, require_policy.resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression, logger);
 		const journalEntry = addResponseEntry({
 			method: req.method ?? "POST",
 			path: urlPath,
@@ -1321,10 +1289,10 @@ async function handleBedrockStream(req, res, raw, modelId, fixtures, journal, de
 			fixture
 		}
 	});
-	require_sse_writer.writeErrorResponse(res, 500, JSON.stringify({ error: {
+	require_sse_writer.writeErrorResponse(res, 500, require_errors.renderError("bedrock", "unknown-response", {
 		message: "Fixture response did not match any known type",
 		type: "server_error"
-	} }));
+	}).body);
 }
 
 //#endregion