@copilotkit/aimock 1.7.0

package/dist/server.js

@@ -0,0 +1,933 @@
+import { buildTextChunks, buildTextCompletion, buildToolCallChunks, buildToolCallCompletion, flattenHeaders, isErrorResponse, isTextResponse, isToolCallResponse } from "./helpers.js";
+import { Journal } from "./journal.js";
+import { matchFixture } from "./router.js";
+import { entryToFixture, validateFixtures } from "./fixture-loader.js";
+import { writeErrorResponse, writeSSEStream } from "./sse-writer.js";
+import { createInterruptionSignal } from "./interruption.js";
+import { applyChaos } from "./chaos.js";
+import { proxyAndRecord } from "./recorder.js";
+import { handleResponses } from "./responses.js";
+import { handleMessages } from "./messages.js";
+import { handleGemini } from "./gemini.js";
+import { handleBedrock, handleBedrockStream } from "./bedrock.js";
+import { handleConverse, handleConverseStream } from "./bedrock-converse.js";
+import { handleEmbeddings } from "./embeddings.js";
+import { handleOllama, handleOllamaGenerate } from "./ollama.js";
+import { handleCohere } from "./cohere.js";
+import { handleSearch } from "./search.js";
+import { handleRerank } from "./rerank.js";
+import { handleModeration } from "./moderation.js";
+import { upgradeToWebSocket } from "./ws-framing.js";
+import { handleWebSocketResponses } from "./ws-responses.js";
+import { handleWebSocketRealtime } from "./ws-realtime.js";
+import { handleWebSocketGeminiLive } from "./ws-gemini-live.js";
+import { Logger } from "./logger.js";
+import { createMetricsRegistry, normalizePathLabel } from "./metrics.js";
+import * as http from "node:http";
+
+//#region src/server.ts
+const COMPLETIONS_PATH = "/v1/chat/completions";
+const RESPONSES_PATH = "/v1/responses";
+const REALTIME_PATH = "/v1/realtime";
+const GEMINI_LIVE_PATH = "/ws/google.ai.generativelanguage.v1beta.GenerativeService.BidiGenerateContent";
+const MESSAGES_PATH = "/v1/messages";
+const EMBEDDINGS_PATH = "/v1/embeddings";
+const COHERE_CHAT_PATH = "/v2/chat";
+const SEARCH_PATH = "/search";
+const RERANK_PATH = "/v2/rerank";
+const MODERATIONS_PATH = "/v1/moderations";
+const DEFAULT_CHUNK_SIZE = 20;
+const GEMINI_PATH_RE = /^\/v1beta\/models\/([^:]+):(generateContent|streamGenerateContent)$/;
+const AZURE_DEPLOYMENT_RE = /^\/openai\/deployments\/([^/]+)\/(chat\/completions|embeddings)$/;
+const BEDROCK_INVOKE_RE = /^\/model\/([^/]+)\/invoke$/;
+const BEDROCK_STREAM_RE = /^\/model\/([^/]+)\/invoke-with-response-stream$/;
+const BEDROCK_CONVERSE_RE = /^\/model\/([^/]+)\/converse$/;
+const BEDROCK_CONVERSE_STREAM_RE = /^\/model\/([^/]+)\/converse-stream$/;
+const VERTEX_AI_RE = /^\/v1\/projects\/[^/]+\/locations\/[^/]+\/publishers\/google\/models\/([^/:]+):(generateContent|streamGenerateContent)$/;
+const OLLAMA_CHAT_PATH = "/api/chat";
+const OLLAMA_GENERATE_PATH = "/api/generate";
+const OLLAMA_TAGS_PATH = "/api/tags";
+const HEALTH_PATH = "/health";
+const READY_PATH = "/ready";
+const MODELS_PATH = "/v1/models";
+const REQUESTS_PATH = "/v1/_requests";
+const DEFAULT_MODELS = [
+	"gpt-4",
+	"gpt-4o",
+	"claude-3-5-sonnet-20241022",
+	"gemini-2.0-flash",
+	"text-embedding-3-small"
+];
+const CORS_HEADERS = {
+	"Access-Control-Allow-Origin": "*",
+	"Access-Control-Allow-Methods": "GET, POST, DELETE, OPTIONS",
+	"Access-Control-Allow-Headers": "Content-Type, Authorization"
+};
+function setCorsHeaders(res) {
+	for (const [key, value] of Object.entries(CORS_HEADERS)) res.setHeader(key, value);
+}
+async function readBody(req) {
+	const buffers = [];
+	for await (const chunk of req) buffers.push(chunk);
+	return Buffer.concat(buffers).toString();
+}
+function handleOptions(res) {
+	setCorsHeaders(res);
+	res.writeHead(204);
+	res.end();
+}
+function handleNotFound(res, message) {
+	setCorsHeaders(res);
+	writeErrorResponse(res, 404, JSON.stringify({ error: {
+		message,
+		type: "not_found"
+	} }));
+}
+const CONTROL_PREFIX = "/__aimock";
+/**
+* Handle requests under `/__aimock/`. Returns `true` if the request was
+* handled, `false` if the path doesn't match the control prefix.
+*/
+async function handleControlAPI(req, res, pathname, fixtures, journal) {
+	if (!pathname.startsWith(CONTROL_PREFIX)) return false;
+	const subPath = pathname.slice(9);
+	setCorsHeaders(res);
+	if (subPath === "/health" && req.method === "GET") {
+		res.writeHead(200, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ status: "ok" }));
+		return true;
+	}
+	if (subPath === "/journal" && req.method === "GET") {
+		res.writeHead(200, { "Content-Type": "application/json" });
+		res.end(JSON.stringify(journal.getAll()));
+		return true;
+	}
+	if (subPath === "/fixtures" && req.method === "POST") {
+		let raw;
+		try {
+			raw = await readBody(req);
+		} catch {
+			res.writeHead(400, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ error: "Failed to read request body" }));
+			return true;
+		}
+		let parsed;
+		try {
+			parsed = JSON.parse(raw);
+		} catch {
+			res.writeHead(400, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ error: "Invalid JSON" }));
+			return true;
+		}
+		if (!Array.isArray(parsed.fixtures)) {
+			res.writeHead(400, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ error: "Missing or invalid \"fixtures\" array" }));
+			return true;
+		}
+		const converted = parsed.fixtures.map(entryToFixture);
+		const errors = validateFixtures(converted).filter((i) => i.severity === "error");
+		if (errors.length > 0) {
+			res.writeHead(400, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({
+				error: "Validation failed",
+				details: errors
+			}));
+			return true;
+		}
+		fixtures.push(...converted);
+		res.writeHead(200, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ added: converted.length }));
+		return true;
+	}
+	if (subPath === "/fixtures" && req.method === "DELETE") {
+		fixtures.length = 0;
+		res.writeHead(200, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ cleared: true }));
+		return true;
+	}
+	if (subPath === "/reset" && req.method === "POST") {
+		fixtures.length = 0;
+		journal.clear();
+		res.writeHead(200, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ reset: true }));
+		return true;
+	}
+	if (subPath === "/error" && req.method === "POST") {
+		let raw;
+		try {
+			raw = await readBody(req);
+		} catch {
+			res.writeHead(400, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ error: "Failed to read request body" }));
+			return true;
+		}
+		let parsed;
+		try {
+			parsed = JSON.parse(raw);
+		} catch {
+			res.writeHead(400, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ error: "Invalid JSON" }));
+			return true;
+		}
+		const status = parsed.status ?? 500;
+		const errorBody = parsed.body;
+		const errorFixture = {
+			match: { predicate: () => true },
+			response: {
+				error: {
+					message: errorBody?.message ?? "Injected error",
+					type: errorBody?.type ?? "server_error",
+					code: errorBody?.code
+				},
+				status
+			}
+		};
+		fixtures.unshift(errorFixture);
+		const original = errorFixture.match.predicate;
+		errorFixture.match.predicate = (req) => {
+			const result = original(req);
+			if (result) queueMicrotask(() => {
+				const idx = fixtures.indexOf(errorFixture);
+				if (idx !== -1) fixtures.splice(idx, 1);
+			});
+			return result;
+		};
+		res.writeHead(200, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ queued: true }));
+		return true;
+	}
+	handleNotFound(res, `Unknown control endpoint: ${pathname}`);
+	return true;
+}
+async function handleCompletions(req, res, fixtures, journal, defaults, modelFallback, providerKey) {
+	setCorsHeaders(res);
+	let raw;
+	try {
+		raw = await readBody(req);
+	} catch (err) {
+		const msg = err instanceof Error ? err.message : "Failed to read request body";
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? COMPLETIONS_PATH,
+			headers: flattenHeaders(req.headers),
+			body: null,
+			response: {
+				status: 500,
+				fixture: null
+			}
+		});
+		writeErrorResponse(res, 500, JSON.stringify({ error: {
+			message: `Request body read failed: ${msg}`,
+			type: "server_error"
+		} }));
+		return;
+	}
+	let body;
+	try {
+		body = JSON.parse(raw);
+		if (modelFallback && !body.model) body.model = modelFallback;
+	} catch {
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? COMPLETIONS_PATH,
+			headers: flattenHeaders(req.headers),
+			body: null,
+			response: {
+				status: 400,
+				fixture: null
+			}
+		});
+		writeErrorResponse(res, 400, JSON.stringify({ error: {
+			message: "Malformed JSON",
+			type: "invalid_request_error",
+			code: "invalid_json"
+		} }));
+		return;
+	}
+	const fixture = matchFixture(fixtures, body, journal.fixtureMatchCounts);
+	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures);
+	const method = req.method ?? "POST";
+	const path = req.url ?? COMPLETIONS_PATH;
+	const flatHeaders = flattenHeaders(req.headers);
+	if (applyChaos(res, fixture, defaults.chaos, req.headers, journal, {
+		method,
+		path,
+		headers: flatHeaders,
+		body
+	}, defaults.registry, defaults.logger)) return;
+	if (!fixture) {
+		if (defaults.record && providerKey) {
+			if (await proxyAndRecord(req, res, body, providerKey, req.url ?? COMPLETIONS_PATH, fixtures, defaults, raw)) {
+				journal.add({
+					method: req.method ?? "POST",
+					path: req.url ?? COMPLETIONS_PATH,
+					headers: flattenHeaders(req.headers),
+					body,
+					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) defaults.logger.error(`STRICT: No fixture matched for ${req.method ?? "POST"} ${req.url ?? COMPLETIONS_PATH}`);
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? COMPLETIONS_PATH,
+			headers: flattenHeaders(req.headers),
+			body,
+			response: {
+				status: strictStatus,
+				fixture: null
+			}
+		});
+		writeErrorResponse(res, strictStatus, JSON.stringify({ error: {
+			message: strictMessage,
+			type: "invalid_request_error",
+			code: "no_fixture_match"
+		} }));
+		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 ?? COMPLETIONS_PATH,
+			headers: flattenHeaders(req.headers),
+			body,
+			response: {
+				status,
+				fixture
+			}
+		});
+		writeErrorResponse(res, status, JSON.stringify(response));
+		return;
+	}
+	if (isTextResponse(response)) {
+		const journalEntry = journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? COMPLETIONS_PATH,
+			headers: flattenHeaders(req.headers),
+			body,
+			response: {
+				status: 200,
+				fixture
+			}
+		});
+		if (body.stream !== true) {
+			const completion = buildTextCompletion(response.content, body.model);
+			res.writeHead(200, { "Content-Type": "application/json" });
+			res.end(JSON.stringify(completion));
+		} else {
+			const chunks = buildTextChunks(response.content, body.model, chunkSize);
+			const interruption = createInterruptionSignal(fixture);
+			if (!await writeSSEStream(res, chunks, {
+				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 ?? COMPLETIONS_PATH,
+			headers: flattenHeaders(req.headers),
+			body,
+			response: {
+				status: 200,
+				fixture
+			}
+		});
+		if (body.stream !== true) {
+			const completion = buildToolCallCompletion(response.toolCalls, body.model);
+			res.writeHead(200, { "Content-Type": "application/json" });
+			res.end(JSON.stringify(completion));
+		} else {
+			const chunks = buildToolCallChunks(response.toolCalls, body.model, chunkSize);
+			const interruption = createInterruptionSignal(fixture);
+			if (!await writeSSEStream(res, chunks, {
+				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 ?? COMPLETIONS_PATH,
+		headers: flattenHeaders(req.headers),
+		body,
+		response: {
+			status: 500,
+			fixture
+		}
+	});
+	writeErrorResponse(res, 500, JSON.stringify({ error: {
+		message: "Fixture response did not match any known type",
+		type: "server_error"
+	} }));
+}
+async function createServer(fixtures, options, mounts, serviceFixtures) {
+	const host = options?.host ?? "127.0.0.1";
+	const port = options?.port ?? 0;
+	const logger = new Logger(options?.logLevel ?? "silent");
+	const registry = options?.metrics ? createMetricsRegistry() : void 0;
+	const serverOptions = options ?? {};
+	const defaults = {
+		latency: serverOptions.latency ?? 0,
+		chunkSize: Math.max(1, serverOptions.chunkSize ?? DEFAULT_CHUNK_SIZE),
+		logger,
+		get chaos() {
+			return serverOptions.chaos;
+		},
+		registry,
+		get record() {
+			return serverOptions.record;
+		},
+		get strict() {
+			return serverOptions.strict;
+		}
+	};
+	if (options?.chaos) {
+		const chaosRates = [
+			{
+				name: "dropRate",
+				value: options.chaos.dropRate
+			},
+			{
+				name: "malformedRate",
+				value: options.chaos.malformedRate
+			},
+			{
+				name: "disconnectRate",
+				value: options.chaos.disconnectRate
+			}
+		];
+		for (const { name, value } of chaosRates) if (value !== void 0 && (value < 0 || value > 1)) logger.warn(`Chaos ${name} (${value}) is outside 0-1 range — will be clamped at runtime`);
+	}
+	const journal = new Journal();
+	if (mounts) for (const { handler } of mounts) {
+		if (handler.setJournal) handler.setJournal(journal);
+		if (registry && handler.setRegistry) handler.setRegistry(registry);
+	}
+	if (registry) registry.setGauge("aimock_fixtures_loaded", {}, fixtures.length);
+	const server = http.createServer((req, res) => {
+		handleHttpRequest(req, res).catch((err) => {
+			const msg = err instanceof Error ? err.message : "Internal error";
+			defaults.logger.warn(`Unhandled request error: ${msg}`);
+			if (!res.headersSent) {
+				res.writeHead(500, { "Content-Type": "application/json" });
+				res.end(JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+			}
+		});
+	});
+	async function handleHttpRequest(req, res) {
+		if (req.method === "OPTIONS") {
+			handleOptions(res);
+			return;
+		}
+		const startTime = registry ? process.hrtime.bigint() : 0n;
+		const parsedUrl = new URL(req.url ?? "/", `http://${req.headers.host ?? "localhost"}`);
+		let pathname = parsedUrl.pathname;
+		if (registry) {
+			const rawPathname = pathname;
+			res.on("finish", () => {
+				try {
+					const normalizedPath = normalizePathLabel(rawPathname);
+					const method = req.method ?? "UNKNOWN";
+					const status = String(res.statusCode);
+					registry.incrementCounter("aimock_requests_total", {
+						method,
+						path: normalizedPath,
+						status
+					});
+					const elapsed = Number(process.hrtime.bigint() - startTime) / 1e9;
+					registry.observeHistogram("aimock_request_duration_seconds", {
+						method,
+						path: normalizedPath
+					}, elapsed);
+				} catch (err) {
+					defaults.logger.warn("metrics instrumentation error", err);
+				}
+			});
+		}
+		if (pathname.startsWith(CONTROL_PREFIX)) {
+			await handleControlAPI(req, res, pathname, fixtures, journal);
+			return;
+		}
+		if (mounts) {
+			for (const { path: mountPath, handler } of mounts) if (pathname === mountPath || pathname.startsWith(mountPath + "/")) {
+				const subPath = pathname.slice(mountPath.length) || "/";
+				if (await handler.handleRequest(req, res, subPath)) return;
+			}
+		}
+		let azureDeploymentId;
+		const azureMatch = pathname.match(AZURE_DEPLOYMENT_RE);
+		if (azureMatch && req.method === "POST") {
+			azureDeploymentId = azureMatch[1];
+			pathname = `/v1/${azureMatch[2]}`;
+		}
+		if (!azureDeploymentId && pathname.startsWith("/openai/")) pathname = pathname.slice(7);
+		if (pathname === HEALTH_PATH && req.method === "GET") {
+			setCorsHeaders(res);
+			if (mounts && mounts.length > 0) {
+				const services = { llm: {
+					status: "ok",
+					fixtures: fixtures.length
+				} };
+				for (const { path: mountPath, handler } of mounts) if (handler.health) {
+					const name = mountPath.replace(/^\//, "");
+					services[name] = handler.health();
+				}
+				res.writeHead(200, { "Content-Type": "application/json" });
+				res.end(JSON.stringify({
+					status: "ok",
+					services
+				}));
+			} else {
+				res.writeHead(200, { "Content-Type": "application/json" });
+				res.end(JSON.stringify({ status: "ok" }));
+			}
+			return;
+		}
+		if (pathname === READY_PATH && req.method === "GET") {
+			setCorsHeaders(res);
+			res.writeHead(200, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ status: "ready" }));
+			return;
+		}
+		if (pathname === "/metrics" && req.method === "GET") {
+			if (!registry) {
+				handleNotFound(res, "Not found");
+				return;
+			}
+			setCorsHeaders(res);
+			res.writeHead(200, { "Content-Type": "text/plain; version=0.0.4; charset=utf-8" });
+			res.end(registry.serialize());
+			return;
+		}
+		if (pathname === MODELS_PATH && req.method === "GET") {
+			setCorsHeaders(res);
+			const modelIds = /* @__PURE__ */ new Set();
+			for (const f of fixtures) if (f.match.model && typeof f.match.model === "string") modelIds.add(f.match.model);
+			const data = (modelIds.size > 0 ? [...modelIds] : DEFAULT_MODELS).map((id) => ({
+				id,
+				object: "model",
+				created: 1686935002,
+				owned_by: "aimock"
+			}));
+			res.writeHead(200, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({
+				object: "list",
+				data
+			}));
+			return;
+		}
+		if (pathname === REQUESTS_PATH) {
+			setCorsHeaders(res);
+			if (req.method === "GET") {
+				const limitParam = parsedUrl.searchParams.get("limit");
+				let opts;
+				if (limitParam) {
+					const limit = parseInt(limitParam, 10);
+					if (Number.isNaN(limit) || limit <= 0) {
+						writeErrorResponse(res, 400, JSON.stringify({ error: {
+							message: `Invalid limit parameter: "${limitParam}"`,
+							type: "invalid_request_error"
+						} }));
+						return;
+					}
+					opts = { limit };
+				}
+				const entries = journal.getAll(opts);
+				res.writeHead(200, { "Content-Type": "application/json" });
+				res.end(JSON.stringify(entries));
+				return;
+			}
+			if (req.method === "DELETE") {
+				journal.clear();
+				res.writeHead(204);
+				res.end();
+				return;
+			}
+			handleNotFound(res, "Not found");
+			return;
+		}
+		if (pathname === RESPONSES_PATH && req.method === "POST") {
+			readBody(req).then((raw) => handleResponses(req, res, raw, fixtures, journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) {
+					try {
+						res.write(`event: error\ndata: ${JSON.stringify({ error: { message: msg } })}\n\n`);
+					} catch (writeErr) {
+						logger.debug("Failed to write error recovery response:", writeErr);
+					}
+					res.end();
+				}
+			});
+			return;
+		}
+		if (pathname === MESSAGES_PATH && req.method === "POST") {
+			readBody(req).then((raw) => handleMessages(req, res, raw, fixtures, journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) {
+					try {
+						res.write(`event: error\ndata: ${JSON.stringify({ error: { message: msg } })}\n\n`);
+					} catch (writeErr) {
+						logger.debug("Failed to write error recovery response:", writeErr);
+					}
+					res.end();
+				}
+			});
+			return;
+		}
+		if (pathname === COHERE_CHAT_PATH && req.method === "POST") {
+			readBody(req).then((raw) => handleCohere(req, res, raw, fixtures, journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) {
+					try {
+						res.write(`event: error\ndata: ${JSON.stringify({ error: { message: msg } })}\n\n`);
+					} catch (writeErr) {
+						logger.debug("Failed to write error recovery response:", writeErr);
+					}
+					res.end();
+				}
+			});
+			return;
+		}
+		if (pathname === EMBEDDINGS_PATH && req.method === "POST") {
+			const deploymentId = azureDeploymentId;
+			readBody(req).then((raw) => {
+				if (deploymentId) try {
+					const parsed = JSON.parse(raw);
+					if (!parsed.model) {
+						parsed.model = deploymentId;
+						return handleEmbeddings(req, res, JSON.stringify(parsed), fixtures, journal, defaults, setCorsHeaders);
+					}
+				} catch {}
+				return handleEmbeddings(req, res, raw, fixtures, journal, defaults, setCorsHeaders);
+			}).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			});
+			return;
+		}
+		const geminiMatch = pathname.match(GEMINI_PATH_RE);
+		if (geminiMatch && req.method === "POST") {
+			const geminiModel = geminiMatch[1];
+			const streaming = geminiMatch[2] === "streamGenerateContent";
+			readBody(req).then((raw) => handleGemini(req, res, raw, geminiModel, streaming, fixtures, journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) {
+					try {
+						res.write(`data: ${JSON.stringify({ error: { message: msg } })}\n\n`);
+					} catch (writeErr) {
+						logger.debug("Failed to write error recovery response:", writeErr);
+					}
+					res.end();
+				}
+			});
+			return;
+		}
+		const vertexMatch = pathname.match(VERTEX_AI_RE);
+		if (vertexMatch && req.method === "POST") {
+			const vertexModel = vertexMatch[1];
+			const streaming = vertexMatch[2] === "streamGenerateContent";
+			readBody(req).then((raw) => handleGemini(req, res, raw, vertexModel, streaming, fixtures, journal, defaults, setCorsHeaders, "vertexai")).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) {
+					try {
+						res.write(`data: ${JSON.stringify({ error: { message: msg } })}\n\n`);
+					} catch (writeErr) {
+						logger.debug("Failed to write error recovery response:", writeErr);
+					}
+					res.end();
+				}
+			});
+			return;
+		}
+		const bedrockMatch = pathname.match(BEDROCK_INVOKE_RE);
+		if (bedrockMatch && req.method === "POST") {
+			const bedrockModelId = bedrockMatch[1];
+			readBody(req).then((raw) => handleBedrock(req, res, raw, bedrockModelId, fixtures, journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			});
+			return;
+		}
+		const bedrockStreamMatch = pathname.match(BEDROCK_STREAM_RE);
+		if (bedrockStreamMatch && req.method === "POST") {
+			const bedrockModelId = bedrockStreamMatch[1];
+			readBody(req).then((raw) => handleBedrockStream(req, res, raw, bedrockModelId, fixtures, journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			});
+			return;
+		}
+		const converseMatch = pathname.match(BEDROCK_CONVERSE_RE);
+		if (converseMatch && req.method === "POST") {
+			const converseModelId = converseMatch[1];
+			readBody(req).then((raw) => handleConverse(req, res, raw, converseModelId, fixtures, journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			});
+			return;
+		}
+		const converseStreamMatch = pathname.match(BEDROCK_CONVERSE_STREAM_RE);
+		if (converseStreamMatch && req.method === "POST") {
+			const converseStreamModelId = converseStreamMatch[1];
+			readBody(req).then((raw) => handleConverseStream(req, res, raw, converseStreamModelId, fixtures, journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			});
+			return;
+		}
+		if (pathname === OLLAMA_CHAT_PATH && req.method === "POST") {
+			readBody(req).then((raw) => handleOllama(req, res, raw, fixtures, journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			});
+			return;
+		}
+		if (pathname === OLLAMA_GENERATE_PATH && req.method === "POST") {
+			readBody(req).then((raw) => handleOllamaGenerate(req, res, raw, fixtures, journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			});
+			return;
+		}
+		if (pathname === OLLAMA_TAGS_PATH && req.method === "GET") {
+			setCorsHeaders(res);
+			const modelIds = /* @__PURE__ */ new Set();
+			for (const f of fixtures) if (f.match.model && typeof f.match.model === "string") modelIds.add(f.match.model);
+			const models = (modelIds.size > 0 ? [...modelIds] : DEFAULT_MODELS).map((name) => ({
+				name,
+				model: name,
+				modified_at: (/* @__PURE__ */ new Date()).toISOString(),
+				size: 0,
+				digest: "",
+				details: {}
+			}));
+			res.writeHead(200, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ models }));
+			return;
+		}
+		if (pathname === SEARCH_PATH && req.method === "POST") {
+			readBody(req).then((raw) => handleSearch(req, res, raw, serviceFixtures?.search ?? [], journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			});
+			return;
+		}
+		if (pathname === RERANK_PATH && req.method === "POST") {
+			readBody(req).then((raw) => handleRerank(req, res, raw, serviceFixtures?.rerank ?? [], journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			});
+			return;
+		}
+		if (pathname === MODERATIONS_PATH && req.method === "POST") {
+			readBody(req).then((raw) => handleModeration(req, res, raw, serviceFixtures?.moderation ?? [], journal, defaults, setCorsHeaders)).catch((err) => {
+				const msg = err instanceof Error ? err.message : "Internal error";
+				if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+					message: msg,
+					type: "server_error"
+				} }));
+				else if (!res.writableEnded) res.destroy();
+			});
+			return;
+		}
+		if (pathname !== COMPLETIONS_PATH) {
+			handleNotFound(res, "Not found");
+			return;
+		}
+		if (req.method !== "POST") {
+			handleNotFound(res, "Not found");
+			return;
+		}
+		handleCompletions(req, res, fixtures, journal, defaults, azureDeploymentId, azureDeploymentId ? "azure" : "openai").catch((err) => {
+			const msg = err instanceof Error ? err.message : "Internal error";
+			if (!res.headersSent) writeErrorResponse(res, 500, JSON.stringify({ error: {
+				message: msg,
+				type: "server_error"
+			} }));
+			else if (!res.writableEnded) {
+				try {
+					res.write(`data: ${JSON.stringify({ error: {
+						message: msg,
+						type: "server_error"
+					} })}\n\n`);
+				} catch (writeErr) {
+					logger.debug("Failed to write error recovery response:", writeErr);
+				}
+				res.end();
+			}
+		});
+	}
+	const activeConnections = /* @__PURE__ */ new Set();
+	server.on("upgrade", (req, socket, head) => {
+		handleUpgradeRequest(req, socket, head).catch((err) => {
+			const msg = err instanceof Error ? err.message : "Internal error";
+			defaults.logger.warn(`Unhandled upgrade error: ${msg}`);
+			if (!socket.destroyed) socket.destroy();
+		});
+	});
+	async function handleUpgradeRequest(req, socket, head) {
+		const parsedUrl = new URL(req.url ?? "/", `http://${req.headers.host ?? "localhost"}`);
+		const pathname = parsedUrl.pathname;
+		if (mounts) {
+			for (const { path: mountPath, handler } of mounts) if ((pathname === mountPath || pathname.startsWith(mountPath + "/")) && handler.handleUpgrade) {
+				const subPath = pathname.slice(mountPath.length) || "/";
+				if (await handler.handleUpgrade(socket, head, subPath)) return;
+			}
+		}
+		if (pathname !== RESPONSES_PATH && pathname !== REALTIME_PATH && pathname !== GEMINI_LIVE_PATH) {
+			socket.write("HTTP/1.1 404 Not Found\r\n\r\n");
+			socket.destroy();
+			return;
+		}
+		if (head.length > 0) socket.unshift(head);
+		let ws;
+		try {
+			ws = upgradeToWebSocket(req, socket);
+		} catch (err) {
+			const msg = err instanceof Error ? err.message : "WebSocket upgrade failed";
+			logger.error(`WebSocket upgrade error: ${msg}`);
+			if (!socket.destroyed) socket.destroy();
+			return;
+		}
+		activeConnections.add(ws);
+		ws.on("error", (err) => {
+			logger.error(`WebSocket error: ${err.message}`);
+			activeConnections.delete(ws);
+		});
+		ws.on("close", () => {
+			activeConnections.delete(ws);
+		});
+		if (pathname === RESPONSES_PATH) handleWebSocketResponses(ws, fixtures, journal, {
+			...defaults,
+			model: "gpt-4"
+		});
+		else if (pathname === REALTIME_PATH) {
+			const model = parsedUrl.searchParams.get("model") ?? "gpt-4o-realtime";
+			handleWebSocketRealtime(ws, fixtures, journal, {
+				...defaults,
+				model
+			});
+		} else if (pathname === GEMINI_LIVE_PATH) handleWebSocketGeminiLive(ws, fixtures, journal, {
+			...defaults,
+			model: "gemini-2.0-flash"
+		});
+	}
+	const originalClose = server.close.bind(server);
+	server.close = function(callback) {
+		for (const ws of activeConnections) ws.close(1001, "Server shutting down");
+		activeConnections.clear();
+		originalClose(callback);
+		return this;
+	};
+	return new Promise((resolve, reject) => {
+		server.on("error", reject);
+		server.listen(port, host, () => {
+			const addr = server.address();
+			if (!addr || typeof addr === "string") {
+				reject(/* @__PURE__ */ new Error("Unexpected address format"));
+				return;
+			}
+			const url = `http://${addr.address}:${addr.port}`;
+			if (mounts) {
+				for (const { path: mountPath, handler } of mounts) if (handler.setBaseUrl) handler.setBaseUrl(url + mountPath);
+			}
+			resolve({
+				server,
+				journal,
+				url,
+				defaults
+			});
+		});
+	});
+}
+
+//#endregion
+export { createServer };
+//# sourceMappingURL=server.js.map