@copilotkit/aimock 1.46.0 → 1.47.0

package/dist/messages.cjs

@@ -8,7 +8,10 @@ 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_timing = require('./response/timing.cjs');
+const require_deliver = require('./response/deliver.cjs');
+const require_proxy = require('./response/proxy.cjs');
+const require_raw = require('./response/raw.cjs');
 const require_finish = require('./response/finish.cjs');
 
 //#region src/messages.ts
@@ -232,6 +235,76 @@ function pushRedactedThinkingStreamEvents(events, startIndex, redactedThinking)
 	}
 	return blockIndex;
 }
+/**
+* Fixture `citations` as Anthropic `web_search_result_location` citations, the
+* URL-sourced citation type (see `CitationsWebSearchResultLocation` in
+* `@anthropic-ai/sdk`), grouped per text block. `texts` are the turn's text
+* blocks in order; citation offsets index their joined text, the same
+* convention as Chat Completions and Gemini. Each citation goes only to the
+* last text block whose start offset in the joined text is at or before the
+* citation's `startIndex`, with `cited_text` sliced from that block's own text
+* (cut off at the block's end).
+* `web_search_result_location` carries no offsets, so nothing is rebased.
+* `encrypted_index` is opaque on the real wire, so aimock fills a stable
+* base64 placeholder. File citations are skipped (see {@link resolveCitations}):
+* Anthropic has no citation type for an OpenAI file ID.
+*/
+function claudeBlockCitations(texts, overrides) {
+	const starts = [];
+	let offset = 0;
+	for (const text of texts) {
+		starts.push(offset);
+		offset += text.length;
+	}
+	const perBlock = texts.map(() => []);
+	if (texts.length === 0) {
+		require_helpers.resolveCitations("", overrides?.citations);
+		return perBlock;
+	}
+	require_helpers.resolveCitations(texts.join(""), overrides?.citations).forEach((c, i) => {
+		let owner = 0;
+		for (let k = 1; k < starts.length; k++) if (starts[k] <= c.startIndex) owner = k;
+		const start = starts[owner];
+		perBlock[owner].push({
+			type: "web_search_result_location",
+			cited_text: texts[owner].slice(c.startIndex - start, c.endIndex - start),
+			url: c.url,
+			title: c.title,
+			encrypted_index: Buffer.from(`aimock-citation:${i}:${c.url}`).toString("base64")
+		});
+	});
+	return perBlock;
+}
+/** {@link claudeBlockCitations} for a turn with a single text block. */
+function claudeCitations(text, overrides) {
+	return claudeBlockCitations([text], overrides)[0];
+}
+/** A text content block; `citations` is added only when it has some. */
+function claudeTextBlock(text, citations) {
+	return citations.length > 0 ? {
+		type: "text",
+		text,
+		citations
+	} : {
+		type: "text",
+		text
+	};
+}
+/**
+* Streamed form of a text block's citations: one `citations_delta` per
+* citation, after the block's text deltas and before its `content_block_stop`,
+* as in the Anthropic citations streaming example. Nothing when it has none.
+*/
+function claudeCitationDeltas(citations, index) {
+	return citations.map((citation) => ({
+		type: "content_block_delta",
+		index,
+		delta: {
+			type: "citations_delta",
+			citation
+		}
+	}));
+}
 function buildClaudeTextStreamEvents(content, model, chunkSize, reasoning, overrides, reasoningSignature, redactedThinking) {
 	const msgId = overrides?.id ?? require_helpers.generateMessageId();
 	const effectiveModel = overrides?.model ?? model;
@@ -306,6 +379,7 @@ function buildClaudeTextStreamEvents(content, model, chunkSize, reasoning, overr
 			}
 		});
 	}
+	events.push(...claudeCitationDeltas(claudeCitations(content, overrides), blockIndex));
 	events.push({
 		type: "content_block_stop",
 		index: blockIndex
@@ -322,6 +396,7 @@ function buildClaudeTextStreamEvents(content, model, chunkSize, reasoning, overr
 	return events;
 }
 function buildClaudeToolCallStreamEvents(toolCalls, model, chunkSize, logger, reasoning, overrides, reasoningSignature, redactedThinking) {
+	require_helpers.resolveCitations("", overrides?.citations);
 	const msgId = overrides?.id ?? require_helpers.generateMessageId();
 	const effectiveModel = overrides?.model ?? model;
 	const signature = reasoningSignature ?? PLACEHOLDER_SIGNATURE;
@@ -443,10 +518,7 @@ function buildClaudeTextResponse(content, model, reasoning, overrides, reasoning
 		thinking: reasoning,
 		signature: reasoningSignature ?? PLACEHOLDER_SIGNATURE
 	});
-	contentBlocks.push({
-		type: "text",
-		text: content
-	});
+	contentBlocks.push(claudeTextBlock(content, claudeCitations(content, overrides)));
 	return {
 		id: overrides?.id ?? require_helpers.generateMessageId(),
 		type: "message",
@@ -459,6 +531,7 @@ function buildClaudeTextResponse(content, model, reasoning, overrides, reasoning
 	};
 }
 function buildClaudeToolCallResponse(toolCalls, model, logger, reasoning, overrides, reasoningSignature, redactedThinking) {
+	require_helpers.resolveCitations("", overrides?.citations);
 	const contentBlocks = [];
 	pushRedactedThinkingBlocks(contentBlocks, redactedThinking);
 	if (reasoning) contentBlocks.push({
@@ -544,6 +617,8 @@ function buildClaudeContentWithToolCallsStreamEvents(content, toolCalls, model,
 	if (blocks && blocks.length > 0) {
 		const outcome = require_helpers.resolveFixtureBlockOutcome(blocks);
 		const ordered = outcome.ordered;
+		const textCitations = claudeBlockCitations(ordered.flatMap((b) => b.type === "text" ? [b.text] : []), overrides);
+		let textPos = 0;
 		for (const block of ordered) if (block.type === "text") {
 			events.push({
 				type: "content_block_start",
@@ -564,6 +639,7 @@ function buildClaudeContentWithToolCallsStreamEvents(content, toolCalls, model,
 					}
 				});
 			}
+			events.push(...claudeCitationDeltas(textCitations[textPos++] ?? [], blockIndex));
 			events.push({
 				type: "content_block_stop",
 				index: blockIndex
@@ -629,6 +705,7 @@ function buildClaudeContentWithToolCallsStreamEvents(content, toolCalls, model,
 			}
 		});
 	}
+	events.push(...claudeCitationDeltas(claudeCitations(content, overrides), blockIndex));
 	events.push({
 		type: "content_block_stop",
 		index: blockIndex
@@ -696,21 +773,17 @@ function buildClaudeContentWithToolCallsResponse(content, toolCalls, model, logg
 	if (blocks && blocks.length > 0) {
 		const outcome = require_helpers.resolveFixtureBlockOutcome(blocks);
 		const ordered = outcome.ordered;
+		const textCitations = claudeBlockCitations(ordered.flatMap((b) => b.type === "text" ? [b.text] : []), overrides);
+		let textPos = 0;
 		blockHasTools = outcome.hasToolCalls;
-		for (const block of ordered) if (block.type === "text") contentBlocks.push({
-			type: "text",
-			text: block.text
-		});
+		for (const block of ordered) if (block.type === "text") contentBlocks.push(claudeTextBlock(block.text, textCitations[textPos++] ?? []));
 		else contentBlocks.push(toolUseBlock({
 			name: block.name,
 			arguments: block.arguments,
 			id: block.id
 		}));
 	} else {
-		contentBlocks.push({
-			type: "text",
-			text: content
-		});
+		contentBlocks.push(claudeTextBlock(content, claudeCitations(content, overrides)));
 		for (const tc of toolCalls) contentBlocks.push(toolUseBlock(tc));
 	}
 	return {
@@ -780,8 +853,10 @@ function prepareClaudeMisbehavior(plan, context) {
 		outputTokens: require_helpers.estimateTokens(content + servedToolCalls.map((call) => call.name + call.arguments).join("") + (reasoning ?? "") + (plan.refusal ?? ""))
 	});
 	const usage = require_anthropic.render(resolvedUsage, "messages-final", "applied-misbehavior");
+	const { citations, ...rest } = require_helpers.extractOverrides(response);
 	const overrides = {
-		...require_helpers.extractOverrides(response),
+		...rest,
+		...context.citations ? { citations } : {},
 		...plan.stop === "length" ? { finishReason: "length" } : {},
 		usage
 	};
@@ -804,6 +879,7 @@ function prepareClaudeMisbehavior(plan, context) {
 			usage
 		};
 		if (!context.stream) return {
+			reasoning,
 			stream: false,
 			body,
 			plan: preparedPlan,
@@ -853,6 +929,7 @@ function prepareClaudeMisbehavior(plan, context) {
 			usage: require_anthropic.render(resolvedUsage, "messages-delta", "applied-misbehavior")
 		}, { type: "message_stop" });
 		return {
+			reasoning,
 			stream: true,
 			events,
 			plan: preparedPlan,
@@ -878,12 +955,14 @@ function prepareClaudeMisbehavior(plan, context) {
 			usage
 		};
 		if (!context.stream) return {
+			reasoning,
 			stream: false,
 			body,
 			plan: preparedPlan,
 			usage
 		};
 		return {
+			reasoning,
 			stream: true,
 			events: [
 				{
@@ -920,6 +999,7 @@ function prepareClaudeMisbehavior(plan, context) {
 		if (plan.summary.fault === "stop-length-mid-tool") {
 			const cutIndex = events.filter((event) => event.type === "content_block_start" && require_helpers.isJsonObject(event.content_block) && event.content_block.type === "tool_use")[plan.target?.index ?? 0]?.index;
 			return {
+				reasoning,
 				stream: true,
 				events: events.filter((event) => !(event.type === "content_block_stop" && event.index === cutIndex)),
 				plan: preparedPlan,
@@ -927,6 +1007,7 @@ function prepareClaudeMisbehavior(plan, context) {
 			};
 		}
 		return {
+			reasoning,
 			stream: true,
 			events,
 			plan: preparedPlan,
@@ -940,51 +1021,13 @@ function prepareClaudeMisbehavior(plan, context) {
 		artifacts.redactedThinking
 	];
 	return {
+		reasoning,
 		stream: false,
 		body: require_helpers.isCombinedFixtureResponse(preparedResponse) ? buildClaudeContentWithToolCallsResponse(preparedResponse.content ?? "", toolCalls, model, context.logger, ...args, preparedResponse.blocks) : require_helpers.isToolCallResponse(preparedResponse) ? buildClaudeToolCallResponse(toolCalls, model, context.logger, ...args) : buildClaudeTextResponse(content, model, ...args),
 		plan: preparedPlan,
 		usage
 	};
 }
-async function writeClaudeSSEStream(res, events, optionsOrLatency) {
-	const opts = typeof optionsOrLatency === "number" ? { latency: optionsOrLatency } : optionsOrLatency ?? {};
-	const latency = opts.latency ?? 0;
-	const profile = opts.streamingProfile;
-	const { recordedTimings, replaySpeed } = opts;
-	const onChunkSent = opts.onChunkSent;
-	if (res.writableEnded) return true;
-	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 = require_sse_writer.calculateDelay(chunkIndex, profile, latency, recordedTimings, replaySpeed);
-			if (chunkDelay > 0) await require_sse_writer.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);
-	}
-}
 async function handleMessages(req, res, raw, fixtures, journal, defaults, setCorsHeaders) {
 	const { logger } = defaults;
 	setCorsHeaders(res);
@@ -1112,7 +1155,10 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 	const completionReq = claudeToCompletionRequest(claudeReq);
 	completionReq._context = require_helpers.getContext(req);
 	const testId = require_helpers.getTestId(req);
-	const { fixture, skippedBySequenceOrTurn } = require_router.matchFixtureDiagnostic(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform, require_router.recordMatchOptions(!!defaults.record, defaults.logger));
+	const { fixture, skippedBySequenceOrTurn } = require_router.matchFixtureDiagnostic(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform, {
+		...require_router.recordMatchOptions(!!defaults.record, defaults.logger),
+		wire: "anthropic"
+	});
 	if (fixture) {
 		journal.incrementFixtureMatchCount(fixture, fixtures, testId);
 		logger.debug(`Fixture matched: ${JSON.stringify(fixture.match).slice(0, 120)}`);
@@ -1150,7 +1196,7 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			return;
 		}
 		if (defaults.record) {
-			const outcome = await require_recorder.proxyAndRecord(req, res, completionReq, "anthropic", req.url ?? "/v1/messages", fixtures, defaults, raw);
+			const { outcome, clientHungUp } = await require_proxy.withHangUpDetection(req, res, () => require_recorder.proxyAndRecord(req, res, completionReq, "anthropic", req.url ?? "/v1/messages", fixtures, defaults, raw));
 			if (outcome === "handled_by_hook") return;
 			if (outcome !== "not_configured") {
 				journal.add({
@@ -1161,7 +1207,8 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 					response: {
 						status: res.statusCode ?? 200,
 						fixture: null,
-						source: "proxy"
+						source: "proxy",
+						...require_proxy.proxyInterruption(res, clientHungUp)
 					}
 				});
 				return;
@@ -1222,14 +1269,22 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 	}
 	const latency = fixture.latency ?? defaults.latency;
 	const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);
+	const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;
+	const streamTiming = {
+		latency,
+		streamingProfile: fixture.streamingProfile,
+		recordedTimings: fixture.recordedTimings,
+		replaySpeed
+	};
+	const jsonOpts = (payloadLength, duplicateId = false) => ({
+		surface: "other",
+		payloadLength,
+		chunkSize,
+		recordedTimings: fixture.recordedTimings,
+		duplicateId,
+		replaySpeed
+	});
 	if (misbehavior.kind === "applied") {
-		const prepared = prepareClaudeMisbehavior(misbehavior, {
-			request: completionReq,
-			stream: claudeReq.stream === true,
-			chunkSize,
-			logger,
-			strict: require_policy.resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression
-		});
 		const entry = journal.add({
 			method: req.method ?? "POST",
 			path: req.url ?? "/v1/messages",
@@ -1240,6 +1295,15 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 				fixture
 			}
 		});
+		require_helpers.rejectGeneratedFiles(misbehavior.response, "Anthropic Messages");
+		const prepared = prepareClaudeMisbehavior(misbehavior, {
+			request: completionReq,
+			stream: claudeReq.stream === true,
+			chunkSize,
+			logger,
+			strict: require_policy.resolveRequestDecisions(defaults.strict, req.headers).reasoningSuppression,
+			citations: true
+		});
 		require_misbehavior.recordMisbehaviorOutcome({
 			entry,
 			summary: prepared.plan.summary,
@@ -1247,27 +1311,26 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			testId
 		});
 		if (!prepared.stream) {
-			res.writeHead(200, { "Content-Type": "application/json" });
-			res.end(JSON.stringify(prepared.body));
-		} else {
-			const interruption = require_interruption.createInterruptionSignal(fixture);
-			try {
-				if (!await writeClaudeSSEStream(res, prepared.events, {
-					latency,
-					streamingProfile: fixture.streamingProfile,
-					recordedTimings: fixture.recordedTimings,
-					replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,
-					signal: interruption?.signal,
-					onChunkSent: interruption?.tick
-				})) {
-					if (!res.writableEnded) res.destroy();
-					entry.response.interrupted = true;
-					entry.response.interruptReason = interruption?.reason() ?? "client aborted";
-				}
-			} finally {
-				interruption?.cleanup();
-			}
-		}
+			const preparedBody = prepared.body;
+			await require_deliver.deliverJson(res, entry, fixture, () => preparedBody, jsonOpts(require_timing.misbehaviorPayloadLength(misbehavior, misbehavior.reasoning === void 0 ? prepared.reasoning : void 0), !!misbehavior.duplicateId));
+		} else await require_deliver.deliverStream(res, entry, fixture, require_sse_writer.typedSseCodec(), prepared.events, streamTiming);
+		return;
+	}
+	if (require_helpers.isRawResponse(response)) {
+		await require_raw.serveRawResponse({
+			wire: "anthropic",
+			stream: claudeReq.stream === true,
+			req,
+			res,
+			fixture,
+			response,
+			body: completionReq,
+			journal,
+			defaults,
+			testId,
+			misbehaviorSummary: misbehavior.summary,
+			errorWire: "messages"
+		});
 		return;
 	}
 	if (require_helpers.isErrorResponse(response)) {
@@ -1312,27 +1375,12 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			defaults,
 			testId
 		});
+		require_helpers.rejectGeneratedFiles(response, "Anthropic Messages");
 		const functionToolCalls = require_helpers.requireEmittedFunctionToolCalls(response, "Anthropic Messages");
-		if (claudeReq.stream !== true) {
-			const body = buildClaudeContentWithToolCallsResponse(response.content ?? "", functionToolCalls, completionReq.model, logger, effReasoning, overrides, effReasoningSignature, effRedactedThinking, response.blocks);
-			res.writeHead(200, { "Content-Type": "application/json" });
-			res.end(JSON.stringify(body));
-		} else {
+		if (claudeReq.stream !== true) await require_deliver.deliverJson(res, journalEntry, fixture, () => buildClaudeContentWithToolCallsResponse(response.content ?? "", functionToolCalls, completionReq.model, logger, effReasoning, overrides, effReasoningSignature, effRedactedThinking, response.blocks), jsonOpts(require_timing.servedPayloadLength(response, effReasoning)));
+		else {
 			const events = buildClaudeContentWithToolCallsStreamEvents(response.content ?? "", functionToolCalls, completionReq.model, chunkSize, logger, effReasoning, overrides, effReasoningSignature, effRedactedThinking, response.blocks);
-			const interruption = require_interruption.createInterruptionSignal(fixture);
-			if (!await writeClaudeSSEStream(res, events, {
-				latency,
-				streamingProfile: fixture.streamingProfile,
-				recordedTimings: fixture.recordedTimings,
-				replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,
-				signal: interruption?.signal,
-				onChunkSent: interruption?.tick
-			})) {
-				if (!res.writableEnded) res.destroy();
-				journalEntry.response.interrupted = true;
-				journalEntry.response.interruptReason = interruption?.reason() ?? "client aborted";
-			}
-			interruption?.cleanup();
+			await require_deliver.deliverStream(res, journalEntry, fixture, require_sse_writer.typedSseCodec(), events, streamTiming);
 		}
 		return;
 	}
@@ -1358,26 +1406,11 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			defaults,
 			testId
 		});
-		if (claudeReq.stream !== true) {
-			const body = buildClaudeTextResponse(response.content, completionReq.model, effReasoning, overrides, effReasoningSignature, effRedactedThinking);
-			res.writeHead(200, { "Content-Type": "application/json" });
-			res.end(JSON.stringify(body));
-		} else {
+		require_helpers.rejectGeneratedFiles(response, "Anthropic Messages");
+		if (claudeReq.stream !== true) await require_deliver.deliverJson(res, journalEntry, fixture, () => buildClaudeTextResponse(response.content, completionReq.model, effReasoning, overrides, effReasoningSignature, effRedactedThinking), jsonOpts(require_timing.servedPayloadLength(response, effReasoning)));
+		else {
 			const events = buildClaudeTextStreamEvents(response.content, completionReq.model, chunkSize, effReasoning, overrides, effReasoningSignature, effRedactedThinking);
-			const interruption = require_interruption.createInterruptionSignal(fixture);
-			if (!await writeClaudeSSEStream(res, events, {
-				latency,
-				streamingProfile: fixture.streamingProfile,
-				recordedTimings: fixture.recordedTimings,
-				replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,
-				signal: interruption?.signal,
-				onChunkSent: interruption?.tick
-			})) {
-				if (!res.writableEnded) res.destroy();
-				journalEntry.response.interrupted = true;
-				journalEntry.response.interruptReason = interruption?.reason() ?? "client aborted";
-			}
-			interruption?.cleanup();
+			await require_deliver.deliverStream(res, journalEntry, fixture, require_sse_writer.typedSseCodec(), events, streamTiming);
 		}
 		return;
 	}
@@ -1403,27 +1436,12 @@ async function handleMessages(req, res, raw, fixtures, journal, defaults, setCor
 			defaults,
 			testId
 		});
+		require_helpers.rejectGeneratedFiles(response, "Anthropic Messages");
 		const functionToolCalls = require_helpers.requireEmittedFunctionToolCalls(response, "Anthropic Messages");
-		if (claudeReq.stream !== true) {
-			const body = buildClaudeToolCallResponse(functionToolCalls, completionReq.model, logger, effReasoning, overrides, effReasoningSignature, effRedactedThinking);
-			res.writeHead(200, { "Content-Type": "application/json" });
-			res.end(JSON.stringify(body));
-		} else {
+		if (claudeReq.stream !== true) await require_deliver.deliverJson(res, journalEntry, fixture, () => buildClaudeToolCallResponse(functionToolCalls, completionReq.model, logger, effReasoning, overrides, effReasoningSignature, effRedactedThinking), jsonOpts(require_timing.servedPayloadLength(response, effReasoning)));
+		else {
 			const events = buildClaudeToolCallStreamEvents(functionToolCalls, completionReq.model, chunkSize, logger, effReasoning, overrides, effReasoningSignature, effRedactedThinking);
-			const interruption = require_interruption.createInterruptionSignal(fixture);
-			if (!await writeClaudeSSEStream(res, events, {
-				latency,
-				streamingProfile: fixture.streamingProfile,
-				recordedTimings: fixture.recordedTimings,
-				replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,
-				signal: interruption?.signal,
-				onChunkSent: interruption?.tick
-			})) {
-				if (!res.writableEnded) res.destroy();
-				journalEntry.response.interrupted = true;
-				journalEntry.response.interruptReason = interruption?.reason() ?? "client aborted";
-			}
-			interruption?.cleanup();
+			await require_deliver.deliverStream(res, journalEntry, fixture, require_sse_writer.typedSseCodec(), events, streamTiming);
 		}
 		return;
 	}