@copilotkit/aimock 1.7.0

package/dist/cohere.js

@@ -0,0 +1,433 @@
+import { flattenHeaders, generateMessageId, generateToolCallId, isErrorResponse, isTextResponse, isToolCallResponse } from "./helpers.js";
+import { matchFixture } from "./router.js";
+import { calculateDelay, delay, writeErrorResponse } from "./sse-writer.js";
+import { createInterruptionSignal } from "./interruption.js";
+import { applyChaos } from "./chaos.js";
+import { proxyAndRecord } from "./recorder.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 cohereToCompletionRequest(req) {
+	const messages = [];
+	for (const msg of req.messages) if (msg.role === "system") messages.push({
+		role: "system",
+		content: msg.content
+	});
+	else if (msg.role === "user") messages.push({
+		role: "user",
+		content: msg.content
+	});
+	else if (msg.role === "assistant") messages.push({
+		role: "assistant",
+		content: msg.content
+	});
+	else if (msg.role === "tool") messages.push({
+		role: "tool",
+		content: msg.content,
+		tool_call_id: msg.tool_call_id
+	});
+	let tools;
+	if (req.tools && req.tools.length > 0) tools = req.tools.map((t) => ({
+		type: "function",
+		function: {
+			name: t.function.name,
+			description: t.function.description,
+			parameters: t.function.parameters
+		}
+	}));
+	return {
+		model: req.model,
+		messages,
+		stream: req.stream,
+		tools
+	};
+}
+function buildCohereTextResponse(content) {
+	return {
+		id: generateMessageId(),
+		finish_reason: "COMPLETE",
+		message: {
+			role: "assistant",
+			content: [{
+				type: "text",
+				text: content
+			}],
+			tool_calls: [],
+			tool_plan: "",
+			citations: []
+		},
+		usage: ZERO_USAGE
+	};
+}
+function buildCohereToolCallResponse(toolCalls, logger) {
+	const cohereCalls = toolCalls.map((tc) => {
+		try {
+			JSON.parse(tc.arguments || "{}");
+		} catch {
+			logger.warn(`Malformed JSON in fixture tool call arguments for "${tc.name}": ${tc.arguments}`);
+		}
+		return {
+			id: tc.id || generateToolCallId(),
+			type: "function",
+			function: {
+				name: tc.name,
+				arguments: tc.arguments || "{}"
+			}
+		};
+	});
+	return {
+		id: generateMessageId(),
+		finish_reason: "TOOL_CALL",
+		message: {
+			role: "assistant",
+			content: [],
+			tool_calls: cohereCalls,
+			tool_plan: "",
+			citations: []
+		},
+		usage: ZERO_USAGE
+	};
+}
+function buildCohereTextStreamEvents(content, chunkSize) {
+	const msgId = generateMessageId();
+	const events = [];
+	events.push({
+		id: msgId,
+		type: "message-start",
+		delta: { message: {
+			role: "assistant",
+			content: [],
+			tool_plan: "",
+			tool_calls: [],
+			citations: []
+		} }
+	});
+	events.push({
+		type: "content-start",
+		index: 0,
+		delta: { message: { content: { type: "text" } } }
+	});
+	for (let i = 0; i < content.length; i += chunkSize) {
+		const slice = content.slice(i, i + chunkSize);
+		events.push({
+			type: "content-delta",
+			index: 0,
+			delta: { message: { content: {
+				type: "text",
+				text: slice
+			} } }
+		});
+	}
+	events.push({
+		type: "content-end",
+		index: 0
+	});
+	events.push({
+		type: "message-end",
+		delta: {
+			finish_reason: "COMPLETE",
+			usage: ZERO_USAGE
+		}
+	});
+	return events;
+}
+function buildCohereToolCallStreamEvents(toolCalls, chunkSize, logger) {
+	const msgId = generateMessageId();
+	const events = [];
+	events.push({
+		id: msgId,
+		type: "message-start",
+		delta: { message: {
+			role: "assistant",
+			content: [],
+			tool_plan: "",
+			tool_calls: [],
+			citations: []
+		} }
+	});
+	events.push({
+		type: "tool-plan-delta",
+		delta: { message: { tool_plan: "I will use the requested tool." } }
+	});
+	for (let idx = 0; idx < toolCalls.length; idx++) {
+		const tc = toolCalls[idx];
+		const callId = tc.id || generateToolCallId();
+		let argsJson;
+		try {
+			JSON.parse(tc.arguments || "{}");
+			argsJson = tc.arguments || "{}";
+		} catch {
+			logger.warn(`Malformed JSON in fixture tool call arguments for "${tc.name}": ${tc.arguments}`);
+			argsJson = "{}";
+		}
+		events.push({
+			type: "tool-call-start",
+			index: idx,
+			delta: { message: { tool_calls: {
+				id: callId,
+				type: "function",
+				function: {
+					name: tc.name,
+					arguments: ""
+				}
+			} } }
+		});
+		for (let i = 0; i < argsJson.length; i += chunkSize) {
+			const slice = argsJson.slice(i, i + chunkSize);
+			events.push({
+				type: "tool-call-delta",
+				index: idx,
+				delta: { message: { tool_calls: { function: { arguments: slice } } } }
+			});
+		}
+		events.push({
+			type: "tool-call-end",
+			index: idx
+		});
+	}
+	events.push({
+		type: "message-end",
+		delta: {
+			finish_reason: "TOOL_CALL",
+			usage: ZERO_USAGE
+		}
+	});
+	return events;
+}
+async function writeCohereSSEStream(res, events, optionsOrLatency) {
+	const opts = typeof optionsOrLatency === "number" ? { latency: optionsOrLatency } : optionsOrLatency ?? {};
+	const latency = opts.latency ?? 0;
+	const profile = opts.streamingProfile;
+	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);
+		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.writableEnded) res.end();
+	return true;
+}
+async function handleCohere(req, res, raw, fixtures, journal, defaults, setCorsHeaders) {
+	const { logger } = defaults;
+	setCorsHeaders(res);
+	let cohereReq;
+	try {
+		cohereReq = JSON.parse(raw);
+	} catch {
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "/v2/chat",
+			headers: flattenHeaders(req.headers),
+			body: null,
+			response: {
+				status: 400,
+				fixture: null
+			}
+		});
+		writeErrorResponse(res, 400, JSON.stringify({ error: {
+			message: "Malformed JSON",
+			type: "invalid_request_error"
+		} }));
+		return;
+	}
+	if (!cohereReq.model) {
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "/v2/chat",
+			headers: flattenHeaders(req.headers),
+			body: null,
+			response: {
+				status: 400,
+				fixture: null
+			}
+		});
+		writeErrorResponse(res, 400, JSON.stringify({ error: {
+			message: "model is required",
+			type: "invalid_request_error"
+		} }));
+		return;
+	}
+	if (!cohereReq.messages || !Array.isArray(cohereReq.messages)) {
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "/v2/chat",
+			headers: flattenHeaders(req.headers),
+			body: null,
+			response: {
+				status: 400,
+				fixture: null
+			}
+		});
+		writeErrorResponse(res, 400, JSON.stringify({ error: {
+			message: "Invalid request: messages array is required",
+			type: "invalid_request_error"
+		} }));
+		return;
+	}
+	const completionReq = cohereToCompletionRequest(cohereReq);
+	const fixture = matchFixture(fixtures, completionReq, journal.fixtureMatchCounts);
+	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures);
+	if (applyChaos(res, fixture, defaults.chaos, req.headers, journal, {
+		method: req.method ?? "POST",
+		path: req.url ?? "/v2/chat",
+		headers: flattenHeaders(req.headers),
+		body: completionReq
+	}, defaults.registry, defaults.logger)) return;
+	if (!fixture) {
+		if (defaults.record) {
+			if (await proxyAndRecord(req, res, completionReq, "cohere", req.url ?? "/v2/chat", fixtures, defaults, raw)) {
+				journal.add({
+					method: req.method ?? "POST",
+					path: req.url ?? "/v2/chat",
+					headers: flattenHeaders(req.headers),
+					body: completionReq,
+					response: {
+						status: res.statusCode ?? 200,
+						fixture: null
+					}
+				});
+				return;
+			}
+		}
+		const strictStatus = defaults.strict ? 503 : 404;
+		const strictMessage = defaults.strict ? "Strict mode: no fixture matched" : "No fixture matched";
+		if (defaults.strict) logger.error(`STRICT: No fixture matched for ${req.method ?? "POST"} ${req.url ?? "/v2/chat"}`);
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "/v2/chat",
+			headers: flattenHeaders(req.headers),
+			body: completionReq,
+			response: {
+				status: strictStatus,
+				fixture: null
+			}
+		});
+		writeErrorResponse(res, strictStatus, JSON.stringify({ error: {
+			message: strictMessage,
+			type: "invalid_request_error"
+		} }));
+		return;
+	}
+	const response = fixture.response;
+	const latency = fixture.latency ?? defaults.latency;
+	const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);
+	if (isErrorResponse(response)) {
+		const status = response.status ?? 500;
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "/v2/chat",
+			headers: flattenHeaders(req.headers),
+			body: completionReq,
+			response: {
+				status,
+				fixture
+			}
+		});
+		writeErrorResponse(res, status, JSON.stringify(response));
+		return;
+	}
+	if (isTextResponse(response)) {
+		const journalEntry = journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "/v2/chat",
+			headers: flattenHeaders(req.headers),
+			body: completionReq,
+			response: {
+				status: 200,
+				fixture
+			}
+		});
+		if (cohereReq.stream !== true) {
+			const body = buildCohereTextResponse(response.content);
+			res.writeHead(200, { "Content-Type": "application/json" });
+			res.end(JSON.stringify(body));
+		} else {
+			const events = buildCohereTextStreamEvents(response.content, chunkSize);
+			const interruption = createInterruptionSignal(fixture);
+			if (!await writeCohereSSEStream(res, events, {
+				latency,
+				streamingProfile: fixture.streamingProfile,
+				signal: interruption?.signal,
+				onChunkSent: interruption?.tick
+			})) {
+				if (!res.writableEnded) res.destroy();
+				journalEntry.response.interrupted = true;
+				journalEntry.response.interruptReason = interruption?.reason();
+			}
+			interruption?.cleanup();
+		}
+		return;
+	}
+	if (isToolCallResponse(response)) {
+		const journalEntry = journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "/v2/chat",
+			headers: flattenHeaders(req.headers),
+			body: completionReq,
+			response: {
+				status: 200,
+				fixture
+			}
+		});
+		if (cohereReq.stream !== true) {
+			const body = buildCohereToolCallResponse(response.toolCalls, logger);
+			res.writeHead(200, { "Content-Type": "application/json" });
+			res.end(JSON.stringify(body));
+		} else {
+			const events = buildCohereToolCallStreamEvents(response.toolCalls, chunkSize, logger);
+			const interruption = createInterruptionSignal(fixture);
+			if (!await writeCohereSSEStream(res, events, {
+				latency,
+				streamingProfile: fixture.streamingProfile,
+				signal: interruption?.signal,
+				onChunkSent: interruption?.tick
+			})) {
+				if (!res.writableEnded) res.destroy();
+				journalEntry.response.interrupted = true;
+				journalEntry.response.interruptReason = interruption?.reason();
+			}
+			interruption?.cleanup();
+		}
+		return;
+	}
+	journal.add({
+		method: req.method ?? "POST",
+		path: req.url ?? "/v2/chat",
+		headers: flattenHeaders(req.headers),
+		body: completionReq,
+		response: {
+			status: 500,
+			fixture
+		}
+	});
+	writeErrorResponse(res, 500, JSON.stringify({ error: {
+		message: "Fixture response did not match any known type",
+		type: "server_error"
+	} }));
+}
+
+//#endregion
+export { cohereToCompletionRequest, handleCohere };
+//# sourceMappingURL=cohere.js.map

package/dist/cohere.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.js","names":[],"sources":["../src/cohere.ts"],"sourcesContent":["/**\n * Cohere v2 Chat API endpoint support.\n *\n * Translates incoming /v2/chat requests into the ChatCompletionRequest\n * format used by the fixture router, and converts fixture responses back into\n * Cohere's typed SSE streaming (or non-streaming) format.\n *\n * Cohere uses typed SSE events (event: + data: lines), similar to the\n * Claude Messages handler in messages.ts.\n */\n\nimport type * as http from \"node:http\";\nimport type {\n  ChatCompletionRequest,\n  ChatMessage,\n  Fixture,\n  HandlerDefaults,\n  StreamingProfile,\n  ToolCall,\n  ToolDefinition,\n} from \"./types.js\";\nimport {\n  generateMessageId,\n  generateToolCallId,\n  isTextResponse,\n  isToolCallResponse,\n  isErrorResponse,\n  flattenHeaders,\n} from \"./helpers.js\";\nimport { matchFixture } from \"./router.js\";\nimport { writeErrorResponse, delay, calculateDelay } from \"./sse-writer.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport type { Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\nimport { applyChaos } from \"./chaos.js\";\nimport { proxyAndRecord } from \"./recorder.js\";\n\n// ─── Cohere v2 Chat request types ───────────────────────────────────────────\n\ninterface CohereMessage {\n  role: \"user\" | \"assistant\" | \"system\" | \"tool\";\n  content: string;\n  tool_call_id?: string;\n}\n\ninterface CohereToolDef {\n  type: string;\n  function: {\n    name: string;\n    description?: string;\n    parameters?: object;\n  };\n}\n\ninterface CohereRequest {\n  model: string;\n  messages: CohereMessage[];\n  stream?: boolean;\n  tools?: CohereToolDef[];\n  response_format?: { type: string; json_schema?: object };\n}\n\n// ─── Cohere SSE event types ─────────────────────────────────────────────────\n\ninterface CohereSSEEvent {\n  type: string;\n  [key: string]: unknown;\n}\n\n// ─── Zero-value usage block ─────────────────────────────────────────────────\n\nconst ZERO_USAGE = {\n  billed_units: { input_tokens: 0, output_tokens: 0, search_units: 0, classifications: 0 },\n  tokens: { input_tokens: 0, output_tokens: 0 },\n};\n\n// ─── Input conversion: Cohere → ChatCompletionRequest ───────────────────────\n\nexport function cohereToCompletionRequest(req: CohereRequest): ChatCompletionRequest {\n  const messages: ChatMessage[] = [];\n\n  for (const msg of req.messages) {\n    if (msg.role === \"system\") {\n      messages.push({ role: \"system\", content: msg.content });\n    } else if (msg.role === \"user\") {\n      messages.push({ role: \"user\", content: msg.content });\n    } else if (msg.role === \"assistant\") {\n      messages.push({ role: \"assistant\", content: msg.content });\n    } else if (msg.role === \"tool\") {\n      messages.push({\n        role: \"tool\",\n        content: msg.content,\n        tool_call_id: msg.tool_call_id,\n      });\n    }\n  }\n\n  // Convert tools\n  let tools: ToolDefinition[] | undefined;\n  if (req.tools && req.tools.length > 0) {\n    tools = req.tools.map((t) => ({\n      type: \"function\" as const,\n      function: {\n        name: t.function.name,\n        description: t.function.description,\n        parameters: t.function.parameters,\n      },\n    }));\n  }\n\n  return {\n    model: req.model,\n    messages,\n    stream: req.stream,\n    tools,\n  };\n}\n\n// ─── Response building: fixture → Cohere v2 Chat format ─────────────────────\n\n// Non-streaming text response\nfunction buildCohereTextResponse(content: string): object {\n  return {\n    id: generateMessageId(),\n    finish_reason: \"COMPLETE\",\n    message: {\n      role: \"assistant\",\n      content: [{ type: \"text\", text: content }],\n      tool_calls: [],\n      tool_plan: \"\",\n      citations: [],\n    },\n    usage: ZERO_USAGE,\n  };\n}\n\n// Non-streaming tool call response\nfunction buildCohereToolCallResponse(toolCalls: ToolCall[], logger: Logger): object {\n  const cohereCalls = toolCalls.map((tc) => {\n    // Validate arguments JSON\n    try {\n      JSON.parse(tc.arguments || \"{}\");\n    } catch {\n      logger.warn(\n        `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n      );\n    }\n    return {\n      id: tc.id || generateToolCallId(),\n      type: \"function\",\n      function: {\n        name: tc.name,\n        arguments: tc.arguments || \"{}\",\n      },\n    };\n  });\n\n  return {\n    id: generateMessageId(),\n    finish_reason: \"TOOL_CALL\",\n    message: {\n      role: \"assistant\",\n      content: [],\n      tool_calls: cohereCalls,\n      tool_plan: \"\",\n      citations: [],\n    },\n    usage: ZERO_USAGE,\n  };\n}\n\n// ─── Streaming event builders ───────────────────────────────────────────────\n\nfunction buildCohereTextStreamEvents(content: string, chunkSize: number): CohereSSEEvent[] {\n  const msgId = generateMessageId();\n  const events: CohereSSEEvent[] = [];\n\n  // message-start\n  events.push({\n    id: msgId,\n    type: \"message-start\",\n    delta: {\n      message: {\n        role: \"assistant\",\n        content: [],\n        tool_plan: \"\",\n        tool_calls: [],\n        citations: [],\n      },\n    },\n  });\n\n  // content-start (type: \"text\" only, no text field)\n  events.push({\n    type: \"content-start\",\n    index: 0,\n    delta: {\n      message: {\n        content: { type: \"text\" },\n      },\n    },\n  });\n\n  // content-delta — text chunks\n  for (let i = 0; i < content.length; i += chunkSize) {\n    const slice = content.slice(i, i + chunkSize);\n    events.push({\n      type: \"content-delta\",\n      index: 0,\n      delta: {\n        message: {\n          content: { type: \"text\", text: slice },\n        },\n      },\n    });\n  }\n\n  // content-end\n  events.push({\n    type: \"content-end\",\n    index: 0,\n  });\n\n  // message-end\n  events.push({\n    type: \"message-end\",\n    delta: {\n      finish_reason: \"COMPLETE\",\n      usage: ZERO_USAGE,\n    },\n  });\n\n  return events;\n}\n\nfunction buildCohereToolCallStreamEvents(\n  toolCalls: ToolCall[],\n  chunkSize: number,\n  logger: Logger,\n): CohereSSEEvent[] {\n  const msgId = generateMessageId();\n  const events: CohereSSEEvent[] = [];\n\n  // message-start\n  events.push({\n    id: msgId,\n    type: \"message-start\",\n    delta: {\n      message: {\n        role: \"assistant\",\n        content: [],\n        tool_plan: \"\",\n        tool_calls: [],\n        citations: [],\n      },\n    },\n  });\n\n  // tool-plan-delta\n  events.push({\n    type: \"tool-plan-delta\",\n    delta: {\n      message: {\n        tool_plan: \"I will use the requested tool.\",\n      },\n    },\n  });\n\n  for (let idx = 0; idx < toolCalls.length; idx++) {\n    const tc = toolCalls[idx];\n    const callId = tc.id || generateToolCallId();\n\n    // Validate arguments JSON\n    let argsJson: string;\n    try {\n      JSON.parse(tc.arguments || \"{}\");\n      argsJson = tc.arguments || \"{}\";\n    } catch {\n      logger.warn(\n        `Malformed JSON in fixture tool call arguments for \"${tc.name}\": ${tc.arguments}`,\n      );\n      argsJson = \"{}\";\n    }\n\n    // tool-call-start\n    events.push({\n      type: \"tool-call-start\",\n      index: idx,\n      delta: {\n        message: {\n          tool_calls: {\n            id: callId,\n            type: \"function\",\n            function: {\n              name: tc.name,\n              arguments: \"\",\n            },\n          },\n        },\n      },\n    });\n\n    // tool-call-delta — chunked arguments\n    for (let i = 0; i < argsJson.length; i += chunkSize) {\n      const slice = argsJson.slice(i, i + chunkSize);\n      events.push({\n        type: \"tool-call-delta\",\n        index: idx,\n        delta: {\n          message: {\n            tool_calls: {\n              function: {\n                arguments: slice,\n              },\n            },\n          },\n        },\n      });\n    }\n\n    // tool-call-end\n    events.push({\n      type: \"tool-call-end\",\n      index: idx,\n    });\n  }\n\n  // message-end\n  events.push({\n    type: \"message-end\",\n    delta: {\n      finish_reason: \"TOOL_CALL\",\n      usage: ZERO_USAGE,\n    },\n  });\n\n  return events;\n}\n\n// ─── SSE writer for Cohere typed events ─────────────────────────────────────\n\ninterface CohereStreamOptions {\n  latency?: number;\n  streamingProfile?: StreamingProfile;\n  signal?: AbortSignal;\n  onChunkSent?: () => void;\n}\n\nasync function writeCohereSSEStream(\n  res: http.ServerResponse,\n  events: CohereSSEEvent[],\n  optionsOrLatency?: number | CohereStreamOptions,\n): Promise<boolean> {\n  const opts: CohereStreamOptions =\n    typeof optionsOrLatency === \"number\" ? { latency: optionsOrLatency } : (optionsOrLatency ?? {});\n  const latency = opts.latency ?? 0;\n  const profile = opts.streamingProfile;\n  const signal = opts.signal;\n  const onChunkSent = opts.onChunkSent;\n\n  if (res.writableEnded) return true;\n  res.setHeader(\"Content-Type\", \"text/event-stream\");\n  res.setHeader(\"Cache-Control\", \"no-cache\");\n  res.setHeader(\"Connection\", \"keep-alive\");\n\n  let chunkIndex = 0;\n  for (const event of events) {\n    const chunkDelay = calculateDelay(chunkIndex, profile, latency);\n    if (chunkDelay > 0) await delay(chunkDelay, signal);\n    if (signal?.aborted) return false;\n    if (res.writableEnded) return true;\n    res.write(`event: ${event.type}\\ndata: ${JSON.stringify(event)}\\n\\n`);\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n    chunkIndex++;\n  }\n\n  if (!res.writableEnded) {\n    res.end();\n  }\n  return true;\n}\n\n// ─── Request handler ────────────────────────────────────────────────────────\n\nexport async function handleCohere(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  raw: string,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: HandlerDefaults,\n  setCorsHeaders: (res: http.ServerResponse) => void,\n): Promise<void> {\n  const { logger } = defaults;\n  setCorsHeaders(res);\n\n  let cohereReq: CohereRequest;\n  try {\n    cohereReq = JSON.parse(raw) as CohereRequest;\n  } catch {\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v2/chat\",\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 400, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      400,\n      JSON.stringify({\n        error: {\n          message: \"Malformed JSON\",\n          type: \"invalid_request_error\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Validate required model field\n  if (!cohereReq.model) {\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v2/chat\",\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 400, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      400,\n      JSON.stringify({\n        error: {\n          message: \"model is required\",\n          type: \"invalid_request_error\",\n        },\n      }),\n    );\n    return;\n  }\n\n  if (!cohereReq.messages || !Array.isArray(cohereReq.messages)) {\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v2/chat\",\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 400, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      400,\n      JSON.stringify({\n        error: {\n          message: \"Invalid request: messages array is required\",\n          type: \"invalid_request_error\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Convert to ChatCompletionRequest for fixture matching\n  const completionReq = cohereToCompletionRequest(cohereReq);\n\n  const fixture = matchFixture(fixtures, completionReq, journal.fixtureMatchCounts);\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures);\n  }\n\n  if (\n    applyChaos(\n      res,\n      fixture,\n      defaults.chaos,\n      req.headers,\n      journal,\n      {\n        method: req.method ?? \"POST\",\n        path: req.url ?? \"/v2/chat\",\n        headers: flattenHeaders(req.headers),\n        body: completionReq,\n      },\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\n\n  if (!fixture) {\n    if (defaults.record) {\n      const proxied = await proxyAndRecord(\n        req,\n        res,\n        completionReq,\n        \"cohere\",\n        req.url ?? \"/v2/chat\",\n        fixtures,\n        defaults,\n        raw,\n      );\n      if (proxied) {\n        journal.add({\n          method: req.method ?? \"POST\",\n          path: req.url ?? \"/v2/chat\",\n          headers: flattenHeaders(req.headers),\n          body: completionReq,\n          response: { status: res.statusCode ?? 200, fixture: null },\n        });\n        return;\n      }\n    }\n    const strictStatus = defaults.strict ? 503 : 404;\n    const strictMessage = defaults.strict\n      ? \"Strict mode: no fixture matched\"\n      : \"No fixture matched\";\n    if (defaults.strict) {\n      logger.error(\n        `STRICT: No fixture matched for ${req.method ?? \"POST\"} ${req.url ?? \"/v2/chat\"}`,\n      );\n    }\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v2/chat\",\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status: strictStatus, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      strictStatus,\n      JSON.stringify({\n        error: {\n          message: strictMessage,\n          type: \"invalid_request_error\",\n        },\n      }),\n    );\n    return;\n  }\n\n  const response = fixture.response;\n  const latency = fixture.latency ?? defaults.latency;\n  const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);\n\n  // Error response\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v2/chat\",\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    writeErrorResponse(res, status, JSON.stringify(response));\n    return;\n  }\n\n  // Text response\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v2/chat\",\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    if (cohereReq.stream !== true) {\n      const body = buildCohereTextResponse(response.content);\n      res.writeHead(200, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify(body));\n    } else {\n      const events = buildCohereTextStreamEvents(response.content, chunkSize);\n      const interruption = createInterruptionSignal(fixture);\n      const completed = await writeCohereSSEStream(res, events, {\n        latency,\n        streamingProfile: fixture.streamingProfile,\n        signal: interruption?.signal,\n        onChunkSent: interruption?.tick,\n      });\n      if (!completed) {\n        if (!res.writableEnded) res.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n      }\n      interruption?.cleanup();\n    }\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v2/chat\",\n      headers: flattenHeaders(req.headers),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    if (cohereReq.stream !== true) {\n      const body = buildCohereToolCallResponse(response.toolCalls, logger);\n      res.writeHead(200, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify(body));\n    } else {\n      const events = buildCohereToolCallStreamEvents(response.toolCalls, chunkSize, logger);\n      const interruption = createInterruptionSignal(fixture);\n      const completed = await writeCohereSSEStream(res, events, {\n        latency,\n        streamingProfile: fixture.streamingProfile,\n        signal: interruption?.signal,\n        onChunkSent: interruption?.tick,\n      });\n      if (!completed) {\n        if (!res.writableEnded) res.destroy();\n        journalEntry.response.interrupted = true;\n        journalEntry.response.interruptReason = interruption?.reason();\n      }\n      interruption?.cleanup();\n    }\n    return;\n  }\n\n  // Unknown response type\n  journal.add({\n    method: req.method ?? \"POST\",\n    path: req.url ?? \"/v2/chat\",\n    headers: flattenHeaders(req.headers),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  writeErrorResponse(\n    res,\n    500,\n    JSON.stringify({\n      error: {\n        message: \"Fixture response did not match any known type\",\n        type: \"server_error\",\n      },\n    }),\n  );\n}\n"],"mappings":";;;;;;;;AAuEA,MAAM,aAAa;CACjB,cAAc;EAAE,cAAc;EAAG,eAAe;EAAG,cAAc;EAAG,iBAAiB;EAAG;CACxF,QAAQ;EAAE,cAAc;EAAG,eAAe;EAAG;CAC9C;AAID,SAAgB,0BAA0B,KAA2C;CACnF,MAAM,WAA0B,EAAE;AAElC,MAAK,MAAM,OAAO,IAAI,SACpB,KAAI,IAAI,SAAS,SACf,UAAS,KAAK;EAAE,MAAM;EAAU,SAAS,IAAI;EAAS,CAAC;UAC9C,IAAI,SAAS,OACtB,UAAS,KAAK;EAAE,MAAM;EAAQ,SAAS,IAAI;EAAS,CAAC;UAC5C,IAAI,SAAS,YACtB,UAAS,KAAK;EAAE,MAAM;EAAa,SAAS,IAAI;EAAS,CAAC;UACjD,IAAI,SAAS,OACtB,UAAS,KAAK;EACZ,MAAM;EACN,SAAS,IAAI;EACb,cAAc,IAAI;EACnB,CAAC;CAKN,IAAI;AACJ,KAAI,IAAI,SAAS,IAAI,MAAM,SAAS,EAClC,SAAQ,IAAI,MAAM,KAAK,OAAO;EAC5B,MAAM;EACN,UAAU;GACR,MAAM,EAAE,SAAS;GACjB,aAAa,EAAE,SAAS;GACxB,YAAY,EAAE,SAAS;GACxB;EACF,EAAE;AAGL,QAAO;EACL,OAAO,IAAI;EACX;EACA,QAAQ,IAAI;EACZ;EACD;;AAMH,SAAS,wBAAwB,SAAyB;AACxD,QAAO;EACL,IAAI,mBAAmB;EACvB,eAAe;EACf,SAAS;GACP,MAAM;GACN,SAAS,CAAC;IAAE,MAAM;IAAQ,MAAM;IAAS,CAAC;GAC1C,YAAY,EAAE;GACd,WAAW;GACX,WAAW,EAAE;GACd;EACD,OAAO;EACR;;AAIH,SAAS,4BAA4B,WAAuB,QAAwB;CAClF,MAAM,cAAc,UAAU,KAAK,OAAO;AAExC,MAAI;AACF,QAAK,MAAM,GAAG,aAAa,KAAK;UAC1B;AACN,UAAO,KACL,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;;AAEH,SAAO;GACL,IAAI,GAAG,MAAM,oBAAoB;GACjC,MAAM;GACN,UAAU;IACR,MAAM,GAAG;IACT,WAAW,GAAG,aAAa;IAC5B;GACF;GACD;AAEF,QAAO;EACL,IAAI,mBAAmB;EACvB,eAAe;EACf,SAAS;GACP,MAAM;GACN,SAAS,EAAE;GACX,YAAY;GACZ,WAAW;GACX,WAAW,EAAE;GACd;EACD,OAAO;EACR;;AAKH,SAAS,4BAA4B,SAAiB,WAAqC;CACzF,MAAM,QAAQ,mBAAmB;CACjC,MAAM,SAA2B,EAAE;AAGnC,QAAO,KAAK;EACV,IAAI;EACJ,MAAM;EACN,OAAO,EACL,SAAS;GACP,MAAM;GACN,SAAS,EAAE;GACX,WAAW;GACX,YAAY,EAAE;GACd,WAAW,EAAE;GACd,EACF;EACF,CAAC;AAGF,QAAO,KAAK;EACV,MAAM;EACN,OAAO;EACP,OAAO,EACL,SAAS,EACP,SAAS,EAAE,MAAM,QAAQ,EAC1B,EACF;EACF,CAAC;AAGF,MAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK,WAAW;EAClD,MAAM,QAAQ,QAAQ,MAAM,GAAG,IAAI,UAAU;AAC7C,SAAO,KAAK;GACV,MAAM;GACN,OAAO;GACP,OAAO,EACL,SAAS,EACP,SAAS;IAAE,MAAM;IAAQ,MAAM;IAAO,EACvC,EACF;GACF,CAAC;;AAIJ,QAAO,KAAK;EACV,MAAM;EACN,OAAO;EACR,CAAC;AAGF,QAAO,KAAK;EACV,MAAM;EACN,OAAO;GACL,eAAe;GACf,OAAO;GACR;EACF,CAAC;AAEF,QAAO;;AAGT,SAAS,gCACP,WACA,WACA,QACkB;CAClB,MAAM,QAAQ,mBAAmB;CACjC,MAAM,SAA2B,EAAE;AAGnC,QAAO,KAAK;EACV,IAAI;EACJ,MAAM;EACN,OAAO,EACL,SAAS;GACP,MAAM;GACN,SAAS,EAAE;GACX,WAAW;GACX,YAAY,EAAE;GACd,WAAW,EAAE;GACd,EACF;EACF,CAAC;AAGF,QAAO,KAAK;EACV,MAAM;EACN,OAAO,EACL,SAAS,EACP,WAAW,kCACZ,EACF;EACF,CAAC;AAEF,MAAK,IAAI,MAAM,GAAG,MAAM,UAAU,QAAQ,OAAO;EAC/C,MAAM,KAAK,UAAU;EACrB,MAAM,SAAS,GAAG,MAAM,oBAAoB;EAG5C,IAAI;AACJ,MAAI;AACF,QAAK,MAAM,GAAG,aAAa,KAAK;AAChC,cAAW,GAAG,aAAa;UACrB;AACN,UAAO,KACL,sDAAsD,GAAG,KAAK,KAAK,GAAG,YACvE;AACD,cAAW;;AAIb,SAAO,KAAK;GACV,MAAM;GACN,OAAO;GACP,OAAO,EACL,SAAS,EACP,YAAY;IACV,IAAI;IACJ,MAAM;IACN,UAAU;KACR,MAAM,GAAG;KACT,WAAW;KACZ;IACF,EACF,EACF;GACF,CAAC;AAGF,OAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK,WAAW;GACnD,MAAM,QAAQ,SAAS,MAAM,GAAG,IAAI,UAAU;AAC9C,UAAO,KAAK;IACV,MAAM;IACN,OAAO;IACP,OAAO,EACL,SAAS,EACP,YAAY,EACV,UAAU,EACR,WAAW,OACZ,EACF,EACF,EACF;IACF,CAAC;;AAIJ,SAAO,KAAK;GACV,MAAM;GACN,OAAO;GACR,CAAC;;AAIJ,QAAO,KAAK;EACV,MAAM;EACN,OAAO;GACL,eAAe;GACf,OAAO;GACR;EACF,CAAC;AAEF,QAAO;;AAYT,eAAe,qBACb,KACA,QACA,kBACkB;CAClB,MAAM,OACJ,OAAO,qBAAqB,WAAW,EAAE,SAAS,kBAAkB,GAAI,oBAAoB,EAAE;CAChG,MAAM,UAAU,KAAK,WAAW;CAChC,MAAM,UAAU,KAAK;CACrB,MAAM,SAAS,KAAK;CACpB,MAAM,cAAc,KAAK;AAEzB,KAAI,IAAI,cAAe,QAAO;AAC9B,KAAI,UAAU,gBAAgB,oBAAoB;AAClD,KAAI,UAAU,iBAAiB,WAAW;AAC1C,KAAI,UAAU,cAAc,aAAa;CAEzC,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,aAAa,eAAe,YAAY,SAAS,QAAQ;AAC/D,MAAI,aAAa,EAAG,OAAM,MAAM,YAAY,OAAO;AACnD,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,IAAI,cAAe,QAAO;AAC9B,MAAI,MAAM,UAAU,MAAM,KAAK,UAAU,KAAK,UAAU,MAAM,CAAC,MAAM;AACrE,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;AAC5B;;AAGF,KAAI,CAAC,IAAI,cACP,KAAI,KAAK;AAEX,QAAO;;AAKT,eAAsB,aACpB,KACA,KACA,KACA,UACA,SACA,UACA,gBACe;CACf,MAAM,EAAE,WAAW;AACnB,gBAAe,IAAI;CAEnB,IAAI;AACJ,KAAI;AACF,cAAY,KAAK,MAAM,IAAI;SACrB;AACN,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACP,EACF,CAAC,CACH;AACD;;AAIF,KAAI,CAAC,UAAU,OAAO;AACpB,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACP,EACF,CAAC,CACH;AACD;;AAGF,KAAI,CAAC,UAAU,YAAY,CAAC,MAAM,QAAQ,UAAU,SAAS,EAAE;AAC7D,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACP,EACF,CAAC,CACH;AACD;;CAIF,MAAM,gBAAgB,0BAA0B,UAAU;CAE1D,MAAM,UAAU,aAAa,UAAU,eAAe,QAAQ,mBAAmB;AAEjF,KAAI,QACF,SAAQ,2BAA2B,SAAS,SAAS;AAGvD,KACE,WACE,KACA,SACA,SAAS,OACT,IAAI,SACJ,SACA;EACE,QAAQ,IAAI,UAAU;EACtB,MAAM,IAAI,OAAO;EACjB,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACP,EACD,SAAS,UACT,SAAS,OACV,CAED;AAEF,KAAI,CAAC,SAAS;AACZ,MAAI,SAAS,QAWX;OAVgB,MAAM,eACpB,KACA,KACA,eACA,UACA,IAAI,OAAO,YACX,UACA,UACA,IACD,EACY;AACX,YAAQ,IAAI;KACV,QAAQ,IAAI,UAAU;KACtB,MAAM,IAAI,OAAO;KACjB,SAAS,eAAe,IAAI,QAAQ;KACpC,MAAM;KACN,UAAU;MAAE,QAAQ,IAAI,cAAc;MAAK,SAAS;MAAM;KAC3D,CAAC;AACF;;;EAGJ,MAAM,eAAe,SAAS,SAAS,MAAM;EAC7C,MAAM,gBAAgB,SAAS,SAC3B,oCACA;AACJ,MAAI,SAAS,OACX,QAAO,MACL,kCAAkC,IAAI,UAAU,OAAO,GAAG,IAAI,OAAO,aACtE;AAEH,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAc,SAAS;IAAM;GAClD,CAAC;AACF,qBACE,KACA,cACA,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACP,EACF,CAAC,CACH;AACD;;CAGF,MAAM,WAAW,QAAQ;CACzB,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;AAGtE,KAAI,gBAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,qBAAmB,KAAK,QAAQ,KAAK,UAAU,SAAS,CAAC;AACzD;;AAIF,KAAI,eAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,MAAI,UAAU,WAAW,MAAM;GAC7B,MAAM,OAAO,wBAAwB,SAAS,QAAQ;AACtD,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,KAAK,CAAC;SACxB;GACL,MAAM,SAAS,4BAA4B,SAAS,SAAS,UAAU;GACvE,MAAM,eAAe,yBAAyB,QAAQ;AAOtD,OAAI,CANc,MAAM,qBAAqB,KAAK,QAAQ;IACxD;IACA,kBAAkB,QAAQ;IAC1B,QAAQ,cAAc;IACtB,aAAa,cAAc;IAC5B,CAAC,EACc;AACd,QAAI,CAAC,IAAI,cAAe,KAAI,SAAS;AACrC,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,iBAAc,SAAS;;AAEzB;;AAIF,KAAI,mBAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,MAAI,UAAU,WAAW,MAAM;GAC7B,MAAM,OAAO,4BAA4B,SAAS,WAAW,OAAO;AACpE,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,KAAK,CAAC;SACxB;GACL,MAAM,SAAS,gCAAgC,SAAS,WAAW,WAAW,OAAO;GACrF,MAAM,eAAe,yBAAyB,QAAQ;AAOtD,OAAI,CANc,MAAM,qBAAqB,KAAK,QAAQ;IACxD;IACA,kBAAkB,QAAQ;IAC1B,QAAQ,cAAc;IACtB,aAAa,cAAc;IAC5B,CAAC,EACc;AACd,QAAI,CAAC,IAAI,cAAe,KAAI,SAAS;AACrC,iBAAa,SAAS,cAAc;AACpC,iBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,iBAAc,SAAS;;AAEzB;;AAIF,SAAQ,IAAI;EACV,QAAQ,IAAI,UAAU;EACtB,MAAM,IAAI,OAAO;EACjB,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;AACF,oBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;EACL,SAAS;EACT,MAAM;EACP,EACF,CAAC,CACH"}

package/dist/config-loader.cjs

@@ -0,0 +1,111 @@
+const require_runtime = require('./_virtual/_rolldown/runtime.cjs');
+const require_a2a_mock = require('./a2a-mock.cjs');
+const require_logger = require('./logger.cjs');
+const require_llmock = require('./llmock.cjs');
+const require_mcp_mock = require('./mcp-mock.cjs');
+const require_vector_mock = require('./vector-mock.cjs');
+let node_path = require("node:path");
+node_path = require_runtime.__toESM(node_path);
+let node_fs = require("node:fs");
+node_fs = require_runtime.__toESM(node_fs);
+
+//#region src/config-loader.ts
+function loadConfig(configPath) {
+	const raw = node_fs.readFileSync(configPath, "utf-8");
+	return JSON.parse(raw);
+}
+async function startFromConfig(config, overrides) {
+	const logger = new require_logger.Logger("info");
+	const llmock = new require_llmock.LLMock({
+		port: overrides?.port ?? config.port ?? 0,
+		host: overrides?.host ?? config.host ?? "127.0.0.1",
+		chaos: config.llm?.chaos,
+		record: config.llm?.record,
+		metrics: config.metrics,
+		strict: config.strict
+	});
+	if (config.llm?.fixtures) {
+		const fixturePath = node_path.resolve(config.llm.fixtures);
+		if (node_fs.statSync(fixturePath).isDirectory()) llmock.loadFixtureDir(fixturePath);
+		else llmock.loadFixtureFile(fixturePath);
+	}
+	if (config.mcp) {
+		const mcpConfig = config.mcp;
+		const mcp = new require_mcp_mock.MCPMock({ serverInfo: mcpConfig.serverInfo });
+		if (mcpConfig.tools) for (const tool of mcpConfig.tools) {
+			const { result, ...def } = tool;
+			mcp.addTool(def);
+			if (result !== void 0) mcp.onToolCall(def.name, () => result);
+		}
+		if (mcpConfig.resources) for (const res of mcpConfig.resources) mcp.addResource({
+			uri: res.uri,
+			name: res.name,
+			mimeType: res.mimeType,
+			description: res.description
+		}, res.text !== void 0 || res.blob !== void 0 ? {
+			text: res.text,
+			blob: res.blob,
+			mimeType: res.mimeType
+		} : void 0);
+		if (mcpConfig.prompts) for (const prompt of mcpConfig.prompts) {
+			const { result, ...def } = prompt;
+			if (result) mcp.addPrompt(def, () => result);
+			else mcp.addPrompt(def);
+		}
+		const mcpPath = mcpConfig.path ?? "/mcp";
+		llmock.mount(mcpPath, mcp);
+		logger.info(`MCPMock mounted at ${mcpPath}`);
+	}
+	if (config.a2a) {
+		const a2aConfig = config.a2a;
+		const a2a = new require_a2a_mock.A2AMock();
+		if (a2aConfig.agents) for (const agentConfig of a2aConfig.agents) {
+			const { messages, tasks, streamingTasks, ...def } = agentConfig;
+			a2a.registerAgent(def);
+			if (messages) for (const m of messages) a2a.onMessage(def.name, m.pattern, m.parts ?? [{ text: "" }]);
+			if (tasks) for (const t of tasks) a2a.onTask(def.name, t.pattern, t.artifacts ?? []);
+			if (streamingTasks) for (const s of streamingTasks) a2a.onStreamingTask(def.name, s.pattern, s.events ?? [], s.delayMs);
+		}
+		const a2aPath = a2aConfig.path ?? "/a2a";
+		llmock.mount(a2aPath, a2a);
+		logger.info(`A2AMock mounted at ${a2aPath}`);
+	}
+	if (config.vector) {
+		const vectorConfig = config.vector;
+		const vector = new require_vector_mock.VectorMock();
+		if (vectorConfig.collections) for (const col of vectorConfig.collections) {
+			vector.addCollection(col.name, { dimension: col.dimension });
+			if (col.vectors && col.vectors.length > 0) vector.upsert(col.name, col.vectors);
+			if (col.queryResults) vector.onQuery(col.name, col.queryResults);
+		}
+		const vectorPath = vectorConfig.path ?? "/vector";
+		llmock.mount(vectorPath, vector);
+		logger.info(`VectorMock mounted at ${vectorPath}`);
+	}
+	if (config.services) {
+		if (config.services.search) {
+			llmock.onSearch(/.*/, []);
+			logger.info("Search service enabled with default empty results");
+		}
+		if (config.services.rerank) {
+			llmock.onRerank(/.*/, []);
+			logger.info("Rerank service enabled with default empty results");
+		}
+		if (config.services.moderate) {
+			llmock.onModerate(/.*/, {
+				flagged: false,
+				categories: {}
+			});
+			logger.info("Moderation service enabled with default unflagged results");
+		}
+	}
+	return {
+		llmock,
+		url: await llmock.start()
+	};
+}
+
+//#endregion
+exports.loadConfig = loadConfig;
+exports.startFromConfig = startFromConfig;
+//# sourceMappingURL=config-loader.cjs.map

package/dist/config-loader.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"config-loader.cjs","names":["fs","Logger","LLMock","path","MCPMock","A2AMock","VectorMock"],"sources":["../src/config-loader.ts"],"sourcesContent":["import * as fs from \"node:fs\";\nimport * as path from \"node:path\";\nimport { LLMock } from \"./llmock.js\";\nimport { MCPMock } from \"./mcp-mock.js\";\nimport { A2AMock } from \"./a2a-mock.js\";\nimport type { ChaosConfig, RecordConfig } from \"./types.js\";\nimport type { MCPToolDefinition, MCPPromptDefinition } from \"./mcp-types.js\";\nimport type { A2AAgentDefinition, A2APart, A2AArtifact, A2AStreamEvent } from \"./a2a-types.js\";\nimport { VectorMock } from \"./vector-mock.js\";\nimport type { QueryResult } from \"./vector-types.js\";\nimport { Logger } from \"./logger.js\";\n\nexport interface MCPConfigTool extends MCPToolDefinition {\n  result?: string;\n}\n\nexport interface MCPConfigResource {\n  uri: string;\n  name: string;\n  mimeType?: string;\n  description?: string;\n  text?: string;\n  blob?: string;\n}\n\nexport interface MCPConfigPrompt extends MCPPromptDefinition {\n  result?: {\n    messages: Array<{ role: string; content: { type: string; text: string } }>;\n  };\n}\n\nexport interface MCPConfig {\n  path?: string;\n  serverInfo?: { name: string; version: string };\n  tools?: MCPConfigTool[];\n  resources?: MCPConfigResource[];\n  prompts?: MCPConfigPrompt[];\n}\n\nexport interface A2AConfigPattern {\n  pattern: string;\n  parts?: A2APart[];\n  artifacts?: A2AArtifact[];\n  events?: A2AStreamEvent[];\n  delayMs?: number;\n}\n\nexport interface A2AConfigAgent extends A2AAgentDefinition {\n  messages?: A2AConfigPattern[];\n  tasks?: A2AConfigPattern[];\n  streamingTasks?: A2AConfigPattern[];\n}\n\nexport interface A2AConfig {\n  path?: string;\n  agents?: A2AConfigAgent[];\n}\n\nexport interface VectorConfigCollection {\n  name: string;\n  dimension: number;\n  vectors?: Array<{\n    id: string;\n    values: number[];\n    metadata?: Record<string, unknown>;\n  }>;\n  queryResults?: QueryResult[];\n}\n\nexport interface VectorConfig {\n  path?: string;\n  collections?: VectorConfigCollection[];\n}\n\nexport interface AimockConfig {\n  llm?: {\n    fixtures?: string;\n    chaos?: ChaosConfig;\n    record?: RecordConfig;\n  };\n  mcp?: MCPConfig;\n  a2a?: A2AConfig;\n  vector?: VectorConfig;\n  services?: { search?: boolean; rerank?: boolean; moderate?: boolean };\n  metrics?: boolean;\n  strict?: boolean;\n  port?: number;\n  host?: string;\n}\n\nexport function loadConfig(configPath: string): AimockConfig {\n  const raw = fs.readFileSync(configPath, \"utf-8\");\n  return JSON.parse(raw) as AimockConfig;\n}\n\nexport async function startFromConfig(\n  config: AimockConfig,\n  overrides?: { port?: number; host?: string },\n): Promise<{ llmock: LLMock; url: string }> {\n  const logger = new Logger(\"info\");\n\n  // Load fixtures if specified\n  const llmock = new LLMock({\n    port: overrides?.port ?? config.port ?? 0,\n    host: overrides?.host ?? config.host ?? \"127.0.0.1\",\n    chaos: config.llm?.chaos,\n    record: config.llm?.record,\n    metrics: config.metrics,\n    strict: config.strict,\n  });\n\n  if (config.llm?.fixtures) {\n    const fixturePath = path.resolve(config.llm.fixtures);\n    const stat = fs.statSync(fixturePath);\n    if (stat.isDirectory()) {\n      llmock.loadFixtureDir(fixturePath);\n    } else {\n      llmock.loadFixtureFile(fixturePath);\n    }\n  }\n\n  // MCP\n  if (config.mcp) {\n    const mcpConfig = config.mcp;\n    const mcp = new MCPMock({\n      serverInfo: mcpConfig.serverInfo,\n    });\n\n    if (mcpConfig.tools) {\n      for (const tool of mcpConfig.tools) {\n        const { result, ...def } = tool;\n        mcp.addTool(def);\n        if (result !== undefined) {\n          mcp.onToolCall(def.name, () => result);\n        }\n      }\n    }\n\n    if (mcpConfig.resources) {\n      for (const res of mcpConfig.resources) {\n        mcp.addResource(\n          { uri: res.uri, name: res.name, mimeType: res.mimeType, description: res.description },\n          res.text !== undefined || res.blob !== undefined\n            ? { text: res.text, blob: res.blob, mimeType: res.mimeType }\n            : undefined,\n        );\n      }\n    }\n\n    if (mcpConfig.prompts) {\n      for (const prompt of mcpConfig.prompts) {\n        const { result, ...def } = prompt;\n        if (result) {\n          mcp.addPrompt(def, () => result as import(\"./mcp-types.js\").MCPPromptResult);\n        } else {\n          mcp.addPrompt(def);\n        }\n      }\n    }\n\n    const mcpPath = mcpConfig.path ?? \"/mcp\";\n    llmock.mount(mcpPath, mcp);\n    logger.info(`MCPMock mounted at ${mcpPath}`);\n  }\n\n  // A2A\n  if (config.a2a) {\n    const a2aConfig = config.a2a;\n    const a2a = new A2AMock();\n\n    if (a2aConfig.agents) {\n      for (const agentConfig of a2aConfig.agents) {\n        const { messages, tasks, streamingTasks, ...def } = agentConfig;\n        a2a.registerAgent(def);\n\n        if (messages) {\n          for (const m of messages) {\n            a2a.onMessage(def.name, m.pattern, m.parts ?? [{ text: \"\" }]);\n          }\n        }\n\n        if (tasks) {\n          for (const t of tasks) {\n            a2a.onTask(def.name, t.pattern, t.artifacts ?? []);\n          }\n        }\n\n        if (streamingTasks) {\n          for (const s of streamingTasks) {\n            a2a.onStreamingTask(def.name, s.pattern, s.events ?? [], s.delayMs);\n          }\n        }\n      }\n    }\n\n    const a2aPath = a2aConfig.path ?? \"/a2a\";\n    llmock.mount(a2aPath, a2a);\n    logger.info(`A2AMock mounted at ${a2aPath}`);\n  }\n\n  // Vector\n  if (config.vector) {\n    const vectorConfig = config.vector;\n    const vector = new VectorMock();\n\n    if (vectorConfig.collections) {\n      for (const col of vectorConfig.collections) {\n        vector.addCollection(col.name, { dimension: col.dimension });\n\n        if (col.vectors && col.vectors.length > 0) {\n          vector.upsert(col.name, col.vectors);\n        }\n\n        if (col.queryResults) {\n          vector.onQuery(col.name, col.queryResults);\n        }\n      }\n    }\n\n    const vectorPath = vectorConfig.path ?? \"/vector\";\n    llmock.mount(vectorPath, vector);\n    logger.info(`VectorMock mounted at ${vectorPath}`);\n  }\n\n  // Services — configure default catch-all responses\n  if (config.services) {\n    if (config.services.search) {\n      llmock.onSearch(/.*/, []);\n      logger.info(\"Search service enabled with default empty results\");\n    }\n    if (config.services.rerank) {\n      llmock.onRerank(/.*/, []);\n      logger.info(\"Rerank service enabled with default empty results\");\n    }\n    if (config.services.moderate) {\n      llmock.onModerate(/.*/, { flagged: false, categories: {} });\n      logger.info(\"Moderation service enabled with default unflagged results\");\n    }\n  }\n\n  const url = await llmock.start();\n  return { llmock, url };\n}\n"],"mappings":";;;;;;;;;;;;AA0FA,SAAgB,WAAW,YAAkC;CAC3D,MAAM,MAAMA,QAAG,aAAa,YAAY,QAAQ;AAChD,QAAO,KAAK,MAAM,IAAI;;AAGxB,eAAsB,gBACpB,QACA,WAC0C;CAC1C,MAAM,SAAS,IAAIC,sBAAO,OAAO;CAGjC,MAAM,SAAS,IAAIC,sBAAO;EACxB,MAAM,WAAW,QAAQ,OAAO,QAAQ;EACxC,MAAM,WAAW,QAAQ,OAAO,QAAQ;EACxC,OAAO,OAAO,KAAK;EACnB,QAAQ,OAAO,KAAK;EACpB,SAAS,OAAO;EAChB,QAAQ,OAAO;EAChB,CAAC;AAEF,KAAI,OAAO,KAAK,UAAU;EACxB,MAAM,cAAcC,UAAK,QAAQ,OAAO,IAAI,SAAS;AAErD,MADaH,QAAG,SAAS,YAAY,CAC5B,aAAa,CACpB,QAAO,eAAe,YAAY;MAElC,QAAO,gBAAgB,YAAY;;AAKvC,KAAI,OAAO,KAAK;EACd,MAAM,YAAY,OAAO;EACzB,MAAM,MAAM,IAAII,yBAAQ,EACtB,YAAY,UAAU,YACvB,CAAC;AAEF,MAAI,UAAU,MACZ,MAAK,MAAM,QAAQ,UAAU,OAAO;GAClC,MAAM,EAAE,QAAQ,GAAG,QAAQ;AAC3B,OAAI,QAAQ,IAAI;AAChB,OAAI,WAAW,OACb,KAAI,WAAW,IAAI,YAAY,OAAO;;AAK5C,MAAI,UAAU,UACZ,MAAK,MAAM,OAAO,UAAU,UAC1B,KAAI,YACF;GAAE,KAAK,IAAI;GAAK,MAAM,IAAI;GAAM,UAAU,IAAI;GAAU,aAAa,IAAI;GAAa,EACtF,IAAI,SAAS,UAAa,IAAI,SAAS,SACnC;GAAE,MAAM,IAAI;GAAM,MAAM,IAAI;GAAM,UAAU,IAAI;GAAU,GAC1D,OACL;AAIL,MAAI,UAAU,QACZ,MAAK,MAAM,UAAU,UAAU,SAAS;GACtC,MAAM,EAAE,QAAQ,GAAG,QAAQ;AAC3B,OAAI,OACF,KAAI,UAAU,WAAW,OAAmD;OAE5E,KAAI,UAAU,IAAI;;EAKxB,MAAM,UAAU,UAAU,QAAQ;AAClC,SAAO,MAAM,SAAS,IAAI;AAC1B,SAAO,KAAK,sBAAsB,UAAU;;AAI9C,KAAI,OAAO,KAAK;EACd,MAAM,YAAY,OAAO;EACzB,MAAM,MAAM,IAAIC,0BAAS;AAEzB,MAAI,UAAU,OACZ,MAAK,MAAM,eAAe,UAAU,QAAQ;GAC1C,MAAM,EAAE,UAAU,OAAO,gBAAgB,GAAG,QAAQ;AACpD,OAAI,cAAc,IAAI;AAEtB,OAAI,SACF,MAAK,MAAM,KAAK,SACd,KAAI,UAAU,IAAI,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC;AAIjE,OAAI,MACF,MAAK,MAAM,KAAK,MACd,KAAI,OAAO,IAAI,MAAM,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;AAItD,OAAI,eACF,MAAK,MAAM,KAAK,eACd,KAAI,gBAAgB,IAAI,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ;;EAM3E,MAAM,UAAU,UAAU,QAAQ;AAClC,SAAO,MAAM,SAAS,IAAI;AAC1B,SAAO,KAAK,sBAAsB,UAAU;;AAI9C,KAAI,OAAO,QAAQ;EACjB,MAAM,eAAe,OAAO;EAC5B,MAAM,SAAS,IAAIC,gCAAY;AAE/B,MAAI,aAAa,YACf,MAAK,MAAM,OAAO,aAAa,aAAa;AAC1C,UAAO,cAAc,IAAI,MAAM,EAAE,WAAW,IAAI,WAAW,CAAC;AAE5D,OAAI,IAAI,WAAW,IAAI,QAAQ,SAAS,EACtC,QAAO,OAAO,IAAI,MAAM,IAAI,QAAQ;AAGtC,OAAI,IAAI,aACN,QAAO,QAAQ,IAAI,MAAM,IAAI,aAAa;;EAKhD,MAAM,aAAa,aAAa,QAAQ;AACxC,SAAO,MAAM,YAAY,OAAO;AAChC,SAAO,KAAK,yBAAyB,aAAa;;AAIpD,KAAI,OAAO,UAAU;AACnB,MAAI,OAAO,SAAS,QAAQ;AAC1B,UAAO,SAAS,MAAM,EAAE,CAAC;AACzB,UAAO,KAAK,oDAAoD;;AAElE,MAAI,OAAO,SAAS,QAAQ;AAC1B,UAAO,SAAS,MAAM,EAAE,CAAC;AACzB,UAAO,KAAK,oDAAoD;;AAElE,MAAI,OAAO,SAAS,UAAU;AAC5B,UAAO,WAAW,MAAM;IAAE,SAAS;IAAO,YAAY,EAAE;IAAE,CAAC;AAC3D,UAAO,KAAK,4DAA4D;;;AAK5E,QAAO;EAAE;EAAQ,KADL,MAAM,OAAO,OAAO;EACV"}