@copilotkit/aimock 1.45.1 → 1.46.0

package/dist/cohere.js

@@ -1,47 +1,22 @@
-import { extractOverrides, flattenHeaders, generateDeterministicEmbedding, generateMessageId, generateToolCallId, getContext, getTestId, isCombinedFixtureResponse, isEmbeddingResponse, isErrorResponse, isJsonObject, isTextResponse, isToolCallResponse, prepareOpenAIChatMisbehavior, requireEmittedFunctionToolCalls, resolveFixtureBlockOutcome, resolveOpenAIChatMisbehaviorUsage, resolveReasoningForModel, resolveResponse, resolveStrictMode, serializeErrorResponse, servedToolArgs, strictNoMatchLogLine, strictNoMatchMessage, strictOverrideField, validateChatMessages, validateToolsField } from "./helpers.js";
+import { renderError } from "./response/errors.js";
+import { resolveRequestDecisions, strictOverrideField } from "./response/policy.js";
+import { extractOverrides, flattenHeaders, generateDeterministicEmbedding, generateMessageId, generateToolCallId, getContext, getTestId, isCombinedFixtureResponse, isEmbeddingResponse, isErrorResponse, isJsonObject, isTextResponse, isToolCallResponse, prepareOpenAIChatMisbehavior, requireEmittedFunctionToolCalls, resolveFixtureBlockOutcome, resolveOpenAIChatMisbehaviorUsage, resolveReasoningForModel, resolveResponse, servedToolArgs, strictNoMatchLogLine, strictNoMatchMessage, validateChatMessages, validateToolsField } from "./helpers.js";
 import { calculateDelay, delay, writeErrorResponse } from "./sse-writer.js";
 import { applyChaosAsync } from "./chaos.js";
 import { normalizeDirectToolSchema, planMisbehavior, recordMisbehaviorOutcome } from "./misbehavior.js";
 import { matchFixtureDiagnostic, recordMatchOptions } from "./router.js";
 import { proxyAndRecord } from "./recorder.js";
 import { createInterruptionSignal } from "./interruption.js";
+import { mapCohereFinish } from "./response/finish.js";
+import { render } from "./response/usage/codecs/cohere.js";
 
 //#region src/cohere.ts
-const ZERO_USAGE = {
-	billed_units: {
-		input_tokens: 0,
-		output_tokens: 0,
-		search_units: 0,
-		classifications: 0
-	},
-	tokens: {
-		input_tokens: 0,
-		output_tokens: 0
-	}
-};
 function cohereFinishReason(overrideFinishReason, defaultReason) {
 	if (!overrideFinishReason) return defaultReason;
-	if (overrideFinishReason === "stop") return "COMPLETE";
-	if (overrideFinishReason === "tool_calls") return "TOOL_CALL";
-	if (overrideFinishReason === "length") return "MAX_TOKENS";
-	return overrideFinishReason;
+	return mapCohereFinish(overrideFinishReason);
 }
 function cohereUsage(overrides) {
-	if (!overrides?.usage) return ZERO_USAGE;
-	const inputTokens = overrides.usage.input_tokens ?? overrides.usage.prompt_tokens ?? 0;
-	const outputTokens = overrides.usage.output_tokens ?? overrides.usage.completion_tokens ?? 0;
-	return {
-		billed_units: {
-			input_tokens: inputTokens,
-			output_tokens: outputTokens,
-			search_units: 0,
-			classifications: 0
-		},
-		tokens: {
-			input_tokens: inputTokens,
-			output_tokens: outputTokens
-		}
-	};
+	return render(overrides);
 }
 /** Extract plain text from structured content (array of { type, text } parts) or passthrough string. */
 function extractTextContent(content) {
@@ -576,10 +551,10 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("cohere", "validation", {
 			message: `Malformed JSON body: ${detail}`,
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	if (!isJsonObject(cohereReq)) {
@@ -593,10 +568,10 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("cohere", "validation", {
 			message: "Request body must be a JSON object",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	if (!cohereReq.model) {
@@ -610,10 +585,10 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("cohere", "validation", {
 			message: "model is required",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	let shapeError = validateChatMessages(cohereReq.messages) ?? validateToolsField(cohereReq.tools);
@@ -632,10 +607,10 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("cohere", "validation", {
 			message: `Invalid request: ${shapeError}`,
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const completionReq = cohereToCompletionRequest(cohereReq);
@@ -654,7 +629,7 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 		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 ?? "/v2/chat", skippedBySequenceOrTurn));
@@ -669,10 +644,10 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 					...strictOverrideField(defaults.strict, req.headers)
 				}
 			});
-			writeErrorResponse(res, strictStatus, JSON.stringify({ error: {
+			writeErrorResponse(res, strictStatus, renderError("cohere", "strict-miss", {
 				message: strictMessage,
 				type: "invalid_request_error"
-			} }));
+			}).body);
 			return;
 		}
 		if (defaults.record) {
@@ -704,10 +679,10 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 				...strictOverrideField(defaults.strict, req.headers)
 			}
 		});
-		writeErrorResponse(res, 404, JSON.stringify({ error: {
+		writeErrorResponse(res, 404, renderError("cohere", "fixture-miss", {
 			message: "No fixture matched",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const resolvedResponse = await resolveResponse(fixture, completionReq);
@@ -739,15 +714,15 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 			testId
 		});
 		if (!misbehavior.summary?.evaluations.length) logger.error(misbehavior.message);
-		writeErrorResponse(res, misbehavior.status, JSON.stringify({
+		writeErrorResponse(res, misbehavior.status, renderError("cohere", "misbehavior", {
 			message: misbehavior.message,
 			code: misbehavior.code
-		}));
+		}).body);
 		return;
 	}
 	const prepared = misbehavior.kind === "applied" ? prepareOpenAIChatMisbehavior(misbehavior) : void 0;
 	const observedK5Stream = cohereReq.stream === true && prepared?.summary.fault === "stop-length-mid-tool";
-	const appliedReasoning = prepared && "reasoning" in prepared.response ? resolveReasoningForModel(prepared.response.reasoning, cohereReq.model, resolveStrictMode(defaults.strict, req.headers), logger) : void 0;
+	const appliedReasoning = prepared && "reasoning" in prepared.response ? resolveReasoningForModel(prepared.response.reasoning, cohereReq.model, resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression, logger) : void 0;
 	const usage = prepared ? resolveOpenAIChatMisbehaviorUsage(prepared, completionReq, appliedReasoning !== void 0) : void 0;
 	const response = prepared && usage ? {
 		...prepared.response,
@@ -775,13 +750,13 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 			defaults,
 			testId
 		});
-		writeErrorResponse(res, status, serializeErrorResponse(response), { retryAfter: response.retryAfter });
+		writeErrorResponse(res, status, renderError("openai", "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 Cohere v2 Chat API — ignoring");
 		const overrides = extractOverrides(response);
-		const effectiveStrict = resolveStrictMode(defaults.strict, req.headers);
+		const effectiveStrict = resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression;
 		const effReasoning = prepared ? appliedReasoning : resolveReasoningForModel(response.reasoning, cohereReq.model, effectiveStrict, logger);
 		const journalEntry = journal.add({
 			method: req.method ?? "POST",
@@ -827,7 +802,7 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 	if (isTextResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Cohere v2 Chat API — ignoring");
 		const overrides = extractOverrides(response);
-		const effectiveStrict = resolveStrictMode(defaults.strict, req.headers);
+		const effectiveStrict = resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression;
 		const effReasoning = prepared ? appliedReasoning : resolveReasoningForModel(response.reasoning, cohereReq.model, effectiveStrict, logger);
 		const journalEntry = journal.add({
 			method: req.method ?? "POST",
@@ -881,7 +856,7 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 	if (isToolCallResponse(response)) {
 		if (response.webSearches?.length) logger.warn("webSearches in fixture response are not supported for Cohere v2 Chat API — ignoring");
 		const overrides = extractOverrides(response);
-		const effectiveStrict = resolveStrictMode(defaults.strict, req.headers);
+		const effectiveStrict = resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression;
 		const effReasoning = prepared ? appliedReasoning : resolveReasoningForModel(response.reasoning, cohereReq.model, effectiveStrict, logger);
 		const journalEntry = journal.add({
 			method: req.method ?? "POST",
@@ -939,10 +914,10 @@ async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsH
 		defaults,
 		testId
 	});
-	writeErrorResponse(res, 500, JSON.stringify({ error: {
+	writeErrorResponse(res, 500, renderError("cohere", "unknown-response", {
 		message: "Fixture response did not match any known type",
 		type: "server_error"
-	} }));
+	}).body);
 }
 async function handleCohereEmbed(req, res, raw, fixtures, journal, defaults, setCorsHeaders) {
 	const { logger } = defaults;
@@ -962,10 +937,10 @@ async function handleCohereEmbed(req, res, raw, fixtures, journal, defaults, set
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("cohere", "validation", {
 			message: `Malformed JSON body: ${detail}`,
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	if (!isJsonObject(embedReq)) {
@@ -979,10 +954,10 @@ async function handleCohereEmbed(req, res, raw, fixtures, journal, defaults, set
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("cohere", "validation", {
 			message: "Request body must be a JSON object",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	if (!embedReq.model) {
@@ -996,10 +971,10 @@ async function handleCohereEmbed(req, res, raw, fixtures, journal, defaults, set
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("cohere", "validation", {
 			message: "model is required",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	if (!embedReq.texts || !Array.isArray(embedReq.texts)) {
@@ -1013,10 +988,10 @@ async function handleCohereEmbed(req, res, raw, fixtures, journal, defaults, set
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("cohere", "validation", {
 			message: "Invalid request: texts array is required",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	if (!embedReq.texts.every((text) => typeof text === "string")) {
@@ -1030,10 +1005,10 @@ async function handleCohereEmbed(req, res, raw, fixtures, journal, defaults, set
 				fixture: null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("cohere", "validation", {
 			message: "Invalid request: texts must be an array of strings",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	const combinedInput = embedReq.texts.join(" ");
@@ -1071,10 +1046,10 @@ async function handleCohereEmbed(req, res, raw, fixtures, journal, defaults, set
 				fixture: fixture ?? null
 			}
 		});
-		writeErrorResponse(res, 400, JSON.stringify({ error: {
+		writeErrorResponse(res, 400, renderError("cohere", "validation", {
 			message: isArray ? "Invalid request: embedding_types must be an array of strings" : "Invalid request: embedding_types must be an array",
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return true;
 	}
 	if (fixture) {
@@ -1091,7 +1066,7 @@ async function handleCohereEmbed(req, res, raw, fixtures, journal, defaults, set
 					fixture
 				}
 			});
-			writeErrorResponse(res, status, serializeErrorResponse(response), { retryAfter: response.retryAfter });
+			writeErrorResponse(res, status, renderError("openai", "fixture", response.error).body, { retryAfter: response.retryAfter });
 			return;
 		}
 		if (isEmbeddingResponse(response)) {
@@ -1129,13 +1104,13 @@ async function handleCohereEmbed(req, res, raw, fixtures, journal, defaults, set
 				fixture
 			}
 		});
-		writeErrorResponse(res, 500, JSON.stringify({ error: {
+		writeErrorResponse(res, 500, renderError("cohere", "unknown-response", {
 			message: "Fixture response did not match any known embedding type (must have embedding or error)",
 			type: "server_error"
-		} }));
+		}).body);
 		return;
 	}
-	if (resolveStrictMode(defaults.strict, req.headers)) {
+	if (resolveRequestDecisions(defaults.strict, req.headers).fixtureMiss) {
 		const strictMessage = strictNoMatchMessage(skippedBySequenceOrTurn);
 		logger.error(strictNoMatchLogLine(req.method ?? "POST", req.url ?? "/v2/embed", skippedBySequenceOrTurn));
 		journal.add({
@@ -1149,10 +1124,10 @@ async function handleCohereEmbed(req, res, raw, fixtures, journal, defaults, set
 				...strictOverrideField(defaults.strict, req.headers)
 			}
 		});
-		writeErrorResponse(res, 503, JSON.stringify({ error: {
+		writeErrorResponse(res, 503, renderError("cohere", "strict-miss", {
 			message: strictMessage,
 			type: "invalid_request_error"
-		} }));
+		}).body);
 		return;
 	}
 	if (defaults.record) {