@copilotkit/aimock 1.11.0 → 1.13.0

package/dist/video.js

@@ -0,0 +1,195 @@
+import { flattenHeaders, getTestId, isErrorResponse, isVideoResponse } from "./helpers.js";
+import { matchFixture } from "./router.js";
+import { writeErrorResponse } from "./sse-writer.js";
+import { applyChaos } from "./chaos.js";
+import { proxyAndRecord } from "./recorder.js";
+
+//#region src/video.ts
+async function handleVideoCreate(req, res, raw, fixtures, journal, defaults, setCorsHeaders, videoStates) {
+	setCorsHeaders(res);
+	const path = req.url ?? "/v1/videos";
+	const method = req.method ?? "POST";
+	let videoReq;
+	try {
+		videoReq = JSON.parse(raw);
+	} catch {
+		journal.add({
+			method,
+			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 syntheticReq = {
+		model: videoReq.model ?? "sora-2",
+		messages: [{
+			role: "user",
+			content: videoReq.prompt
+		}],
+		_endpointType: "video"
+	};
+	const testId = getTestId(req);
+	const fixture = matchFixture(fixtures, syntheticReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
+	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures, testId);
+	if (applyChaos(res, fixture, defaults.chaos, req.headers, journal, {
+		method,
+		path,
+		headers: flattenHeaders(req.headers),
+		body: syntheticReq
+	}, defaults.registry, defaults.logger)) return;
+	if (!fixture) {
+		if (defaults.record) {
+			if (await proxyAndRecord(req, res, syntheticReq, "openai", req.url ?? "/v1/videos", fixtures, defaults, raw)) {
+				journal.add({
+					method,
+					path,
+					headers: flattenHeaders(req.headers),
+					body: syntheticReq,
+					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";
+		journal.add({
+			method,
+			path,
+			headers: flattenHeaders(req.headers),
+			body: syntheticReq,
+			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;
+	if (isErrorResponse(response)) {
+		const status = response.status ?? 500;
+		journal.add({
+			method,
+			path,
+			headers: flattenHeaders(req.headers),
+			body: syntheticReq,
+			response: {
+				status,
+				fixture
+			}
+		});
+		writeErrorResponse(res, status, JSON.stringify(response));
+		return;
+	}
+	if (!isVideoResponse(response)) {
+		journal.add({
+			method,
+			path,
+			headers: flattenHeaders(req.headers),
+			body: syntheticReq,
+			response: {
+				status: 500,
+				fixture
+			}
+		});
+		writeErrorResponse(res, 500, JSON.stringify({ error: {
+			message: "Fixture response is not a video type",
+			type: "server_error"
+		} }));
+		return;
+	}
+	journal.add({
+		method,
+		path,
+		headers: flattenHeaders(req.headers),
+		body: syntheticReq,
+		response: {
+			status: 200,
+			fixture
+		}
+	});
+	const video = response.video;
+	const created_at = Math.floor(Date.now() / 1e3);
+	const stateKey = `${testId}:${video.id}`;
+	videoStates.set(stateKey, video);
+	if (video.status === "completed") {
+		res.writeHead(200, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({
+			id: video.id,
+			status: video.status,
+			url: video.url,
+			created_at
+		}));
+	} else {
+		res.writeHead(200, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({
+			id: video.id,
+			status: video.status,
+			created_at
+		}));
+	}
+}
+function handleVideoStatus(req, res, videoId, journal, setCorsHeaders, videoStates) {
+	setCorsHeaders(res);
+	const path = req.url ?? `/v1/videos/${videoId}`;
+	const method = req.method ?? "GET";
+	const stateKey = `${getTestId(req)}:${videoId}`;
+	const video = videoStates.get(stateKey);
+	if (!video) {
+		journal.add({
+			method,
+			path,
+			headers: flattenHeaders(req.headers),
+			body: null,
+			response: {
+				status: 404,
+				fixture: null
+			}
+		});
+		writeErrorResponse(res, 404, JSON.stringify({ error: {
+			message: `Video ${videoId} not found`,
+			type: "not_found"
+		} }));
+		return;
+	}
+	journal.add({
+		method,
+		path,
+		headers: flattenHeaders(req.headers),
+		body: null,
+		response: {
+			status: 200,
+			fixture: null
+		}
+	});
+	const created_at = Math.floor(Date.now() / 1e3);
+	const body = {
+		id: video.id,
+		status: video.status,
+		created_at
+	};
+	if (video.url) body.url = video.url;
+	res.writeHead(200, { "Content-Type": "application/json" });
+	res.end(JSON.stringify(body));
+}
+
+//#endregion
+export { handleVideoCreate, handleVideoStatus };
+//# sourceMappingURL=video.js.map

package/dist/video.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"video.js","names":[],"sources":["../src/video.ts"],"sourcesContent":["import type * as http from \"node:http\";\nimport type { ChatCompletionRequest, Fixture, HandlerDefaults, VideoResponse } from \"./types.js\";\nimport { isVideoResponse, isErrorResponse, flattenHeaders, getTestId } from \"./helpers.js\";\nimport { matchFixture } from \"./router.js\";\nimport { writeErrorResponse } from \"./sse-writer.js\";\nimport type { Journal } from \"./journal.js\";\nimport { applyChaos } from \"./chaos.js\";\nimport { proxyAndRecord } from \"./recorder.js\";\n\ninterface VideoRequest {\n  model?: string;\n  prompt: string;\n  [key: string]: unknown;\n}\n\n/** Stored video state for GET status checks. Key: `${testId}:${videoId}` */\nexport type VideoStateMap = Map<string, VideoResponse[\"video\"]>;\n\nexport async function handleVideoCreate(\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  videoStates: VideoStateMap,\n): Promise<void> {\n  setCorsHeaders(res);\n  const path = req.url ?? \"/v1/videos\";\n  const method = req.method ?? \"POST\";\n\n  let videoReq: VideoRequest;\n  try {\n    videoReq = JSON.parse(raw) as VideoRequest;\n  } catch {\n    journal.add({\n      method,\n      path,\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: { message: \"Malformed JSON\", type: \"invalid_request_error\", code: \"invalid_json\" },\n      }),\n    );\n    return;\n  }\n\n  const syntheticReq: ChatCompletionRequest = {\n    model: videoReq.model ?? \"sora-2\",\n    messages: [{ role: \"user\", content: videoReq.prompt }],\n    _endpointType: \"video\",\n  };\n\n  const testId = getTestId(req);\n  const fixture = matchFixture(\n    fixtures,\n    syntheticReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (\n    applyChaos(\n      res,\n      fixture,\n      defaults.chaos,\n      req.headers,\n      journal,\n      { method, path, headers: flattenHeaders(req.headers), body: syntheticReq },\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        syntheticReq,\n        \"openai\",\n        req.url ?? \"/v1/videos\",\n        fixtures,\n        defaults,\n        raw,\n      );\n      if (proxied) {\n        journal.add({\n          method,\n          path,\n          headers: flattenHeaders(req.headers),\n          body: syntheticReq,\n          response: { status: res.statusCode ?? 200, fixture: null },\n        });\n        return;\n      }\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    journal.add({\n      method,\n      path,\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: { status: strictStatus, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      strictStatus,\n      JSON.stringify({\n        error: { message: strictMessage, type: \"invalid_request_error\", code: \"no_fixture_match\" },\n      }),\n    );\n    return;\n  }\n\n  const response = fixture.response;\n\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    journal.add({\n      method,\n      path,\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: { status, fixture },\n    });\n    writeErrorResponse(res, status, JSON.stringify(response));\n    return;\n  }\n\n  if (!isVideoResponse(response)) {\n    journal.add({\n      method,\n      path,\n      headers: flattenHeaders(req.headers),\n      body: syntheticReq,\n      response: { status: 500, fixture },\n    });\n    writeErrorResponse(\n      res,\n      500,\n      JSON.stringify({\n        error: { message: \"Fixture response is not a video type\", type: \"server_error\" },\n      }),\n    );\n    return;\n  }\n\n  journal.add({\n    method,\n    path,\n    headers: flattenHeaders(req.headers),\n    body: syntheticReq,\n    response: { status: 200, fixture },\n  });\n\n  const video = response.video;\n  const created_at = Math.floor(Date.now() / 1000);\n\n  // Store for GET status checks\n  const stateKey = `${testId}:${video.id}`;\n  videoStates.set(stateKey, video);\n\n  if (video.status === \"completed\") {\n    res.writeHead(200, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ id: video.id, status: video.status, url: video.url, created_at }));\n  } else {\n    res.writeHead(200, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ id: video.id, status: video.status, created_at }));\n  }\n}\n\nexport function handleVideoStatus(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  videoId: string,\n  journal: Journal,\n  setCorsHeaders: (res: http.ServerResponse) => void,\n  videoStates: VideoStateMap,\n): void {\n  setCorsHeaders(res);\n  const path = req.url ?? `/v1/videos/${videoId}`;\n  const method = req.method ?? \"GET\";\n\n  const testId = getTestId(req);\n  const stateKey = `${testId}:${videoId}`;\n  const video = videoStates.get(stateKey);\n\n  if (!video) {\n    journal.add({\n      method,\n      path,\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 404, fixture: null },\n    });\n    writeErrorResponse(\n      res,\n      404,\n      JSON.stringify({ error: { message: `Video ${videoId} not found`, type: \"not_found\" } }),\n    );\n    return;\n  }\n\n  journal.add({\n    method,\n    path,\n    headers: flattenHeaders(req.headers),\n    body: null,\n    response: { status: 200, fixture: null },\n  });\n\n  const created_at = Math.floor(Date.now() / 1000);\n  const body: Record<string, unknown> = {\n    id: video.id,\n    status: video.status,\n    created_at,\n  };\n  if (video.url) body.url = video.url;\n\n  res.writeHead(200, { \"Content-Type\": \"application/json\" });\n  res.end(JSON.stringify(body));\n}\n"],"mappings":";;;;;;;AAkBA,eAAsB,kBACpB,KACA,KACA,KACA,UACA,SACA,UACA,gBACA,aACe;AACf,gBAAe,IAAI;CACnB,MAAM,OAAO,IAAI,OAAO;CACxB,MAAM,SAAS,IAAI,UAAU;CAE7B,IAAI;AACJ,KAAI;AACF,aAAW,KAAK,MAAM,IAAI;SACpB;AACN,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GAAE,SAAS;GAAkB,MAAM;GAAyB,MAAM;GAAgB,EAC1F,CAAC,CACH;AACD;;CAGF,MAAM,eAAsC;EAC1C,OAAO,SAAS,SAAS;EACzB,UAAU,CAAC;GAAE,MAAM;GAAQ,SAAS,SAAS;GAAQ,CAAC;EACtD,eAAe;EAChB;CAED,MAAM,SAAS,UAAU,IAAI;CAC7B,MAAM,UAAU,aACd,UACA,cACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;AAED,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KACE,WACE,KACA,SACA,SAAS,OACT,IAAI,SACJ,SACA;EAAE;EAAQ;EAAM,SAAS,eAAe,IAAI,QAAQ;EAAE,MAAM;EAAc,EAC1E,SAAS,UACT,SAAS,OACV,CAED;AAEF,KAAI,CAAC,SAAS;AACZ,MAAI,SAAS,QAWX;OAVgB,MAAM,eACpB,KACA,KACA,cACA,UACA,IAAI,OAAO,cACX,UACA,UACA,IACD,EACY;AACX,YAAQ,IAAI;KACV;KACA;KACA,SAAS,eAAe,IAAI,QAAQ;KACpC,MAAM;KACN,UAAU;MAAE,QAAQ,IAAI,cAAc;MAAK,SAAS;MAAM;KAC3D,CAAC;AACF;;;EAIJ,MAAM,eAAe,SAAS,SAAS,MAAM;EAC7C,MAAM,gBAAgB,SAAS,SAC3B,oCACA;AACJ,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAc,SAAS;IAAM;GAClD,CAAC;AACF,qBACE,KACA,cACA,KAAK,UAAU,EACb,OAAO;GAAE,SAAS;GAAe,MAAM;GAAyB,MAAM;GAAoB,EAC3F,CAAC,CACH;AACD;;CAGF,MAAM,WAAW,QAAQ;AAEzB,KAAI,gBAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAClC,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC;AACF,qBAAmB,KAAK,QAAQ,KAAK,UAAU,SAAS,CAAC;AACzD;;AAGF,KAAI,CAAC,gBAAgB,SAAS,EAAE;AAC9B,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EACb,OAAO;GAAE,SAAS;GAAwC,MAAM;GAAgB,EACjF,CAAC,CACH;AACD;;AAGF,SAAQ,IAAI;EACV;EACA;EACA,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC;CAEF,MAAM,QAAQ,SAAS;CACvB,MAAM,aAAa,KAAK,MAAM,KAAK,KAAK,GAAG,IAAK;CAGhD,MAAM,WAAW,GAAG,OAAO,GAAG,MAAM;AACpC,aAAY,IAAI,UAAU,MAAM;AAEhC,KAAI,MAAM,WAAW,aAAa;AAChC,MAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,MAAI,IAAI,KAAK,UAAU;GAAE,IAAI,MAAM;GAAI,QAAQ,MAAM;GAAQ,KAAK,MAAM;GAAK;GAAY,CAAC,CAAC;QACtF;AACL,MAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,MAAI,IAAI,KAAK,UAAU;GAAE,IAAI,MAAM;GAAI,QAAQ,MAAM;GAAQ;GAAY,CAAC,CAAC;;;AAI/E,SAAgB,kBACd,KACA,KACA,SACA,SACA,gBACA,aACM;AACN,gBAAe,IAAI;CACnB,MAAM,OAAO,IAAI,OAAO,cAAc;CACtC,MAAM,SAAS,IAAI,UAAU;CAG7B,MAAM,WAAW,GADF,UAAU,IAAI,CACF,GAAG;CAC9B,MAAM,QAAQ,YAAY,IAAI,SAAS;AAEvC,KAAI,CAAC,OAAO;AACV,UAAQ,IAAI;GACV;GACA;GACA,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,qBACE,KACA,KACA,KAAK,UAAU,EAAE,OAAO;GAAE,SAAS,SAAS,QAAQ;GAAa,MAAM;GAAa,EAAE,CAAC,CACxF;AACD;;AAGF,SAAQ,IAAI;EACV;EACA;EACA,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK,SAAS;GAAM;EACzC,CAAC;CAEF,MAAM,aAAa,KAAK,MAAM,KAAK,KAAK,GAAG,IAAK;CAChD,MAAM,OAAgC;EACpC,IAAI,MAAM;EACV,QAAQ,MAAM;EACd;EACD;AACD,KAAI,MAAM,IAAK,MAAK,MAAM,MAAM;AAEhC,KAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,KAAI,IAAI,KAAK,UAAU,KAAK,CAAC"}

package/dist/vitest.cjs

@@ -0,0 +1,80 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_runtime = require('./_virtual/_rolldown/runtime.cjs');
+const require_fixture_loader = require('./fixture-loader.cjs');
+const require_llmock = require('./llmock.cjs');
+const require_chunk_hooks = require('./node_modules/.pnpm/@vitest_runner@3.2.4/node_modules/@vitest/runner/dist/chunk-hooks.cjs');
+let node_path = require("node:path");
+let node_fs = require("node:fs");
+
+//#region src/vitest.ts
+/**
+* Vitest integration for aimock.
+*
+* Usage:
+*   import { useAimock } from "@copilotkit/aimock/vitest";
+*
+*   const mock = useAimock({ fixtures: "./fixtures" });
+*
+*   it("responds", async () => {
+*     const res = await fetch(`${mock().url}/v1/chat/completions`, { ... });
+*   });
+*/
+/**
+* Start an aimock server for the duration of the test suite.
+*
+* - `beforeAll`: starts the server and optionally loads fixtures
+* - `beforeEach`: resets fixture match counts (not fixtures themselves)
+* - `afterAll`: stops the server
+*
+* Returns a getter function — call it inside tests to access the handle.
+*/
+function useAimock(options = {}) {
+	let handle = null;
+	require_chunk_hooks.beforeAll(async () => {
+		const { fixtures: fixturePath, patchEnv, ...serverOpts } = options;
+		const llm = new require_llmock.LLMock(serverOpts);
+		if (fixturePath) {
+			const loadedFixtures = loadFixtures((0, node_path.resolve)(fixturePath));
+			for (const f of loadedFixtures) llm.addFixture(f);
+		}
+		const url = await llm.start();
+		if (patchEnv !== false) {
+			process.env.OPENAI_BASE_URL = `${url}/v1`;
+			process.env.ANTHROPIC_BASE_URL = `${url}/v1`;
+		}
+		handle = {
+			llm,
+			url
+		};
+	});
+	require_chunk_hooks.beforeEach(() => {
+		if (handle) handle.llm.resetMatchCounts();
+	});
+	require_chunk_hooks.afterAll(async () => {
+		if (handle) {
+			if (options.patchEnv !== false) {
+				delete process.env.OPENAI_BASE_URL;
+				delete process.env.ANTHROPIC_BASE_URL;
+			}
+			await handle.llm.stop();
+			handle = null;
+		}
+	});
+	return () => {
+		if (!handle) throw new Error("useAimock(): server not started — are you calling this inside a test?");
+		return handle;
+	};
+}
+function loadFixtures(fixturePath) {
+	try {
+		if ((0, node_fs.statSync)(fixturePath).isDirectory()) return require_fixture_loader.loadFixturesFromDir(fixturePath);
+		return require_fixture_loader.loadFixtureFile(fixturePath);
+	} catch {
+		return [];
+	}
+}
+
+//#endregion
+exports.LLMock = require_llmock.LLMock;
+exports.useAimock = useAimock;
+//# sourceMappingURL=vitest.cjs.map

package/dist/vitest.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"vitest.cjs","names":["LLMock","loadFixturesFromDir","loadFixtureFile"],"sources":["../src/vitest.ts"],"sourcesContent":["/**\n * Vitest integration for aimock.\n *\n * Usage:\n *   import { useAimock } from \"@copilotkit/aimock/vitest\";\n *\n *   const mock = useAimock({ fixtures: \"./fixtures\" });\n *\n *   it(\"responds\", async () => {\n *     const res = await fetch(`${mock().url}/v1/chat/completions`, { ... });\n *   });\n */\n\nimport { beforeAll, afterAll, beforeEach } from \"vitest\";\nimport { LLMock } from \"./llmock.js\";\nimport { loadFixtureFile, loadFixturesFromDir } from \"./fixture-loader.js\";\nimport type { Fixture, MockServerOptions } from \"./types.js\";\nimport { statSync } from \"node:fs\";\nimport { resolve } from \"node:path\";\n\nexport interface UseAimockOptions extends MockServerOptions {\n  /** Path to fixture file or directory. Loaded automatically on start. */\n  fixtures?: string;\n  /** If true, sets process.env.OPENAI_BASE_URL to the mock URL + /v1. */\n  patchEnv?: boolean;\n}\n\nexport interface AimockHandle {\n  /** The LLMock instance. */\n  readonly llm: LLMock;\n  /** The server URL (e.g., http://127.0.0.1:4010). */\n  readonly url: string;\n}\n\n/**\n * Start an aimock server for the duration of the test suite.\n *\n * - `beforeAll`: starts the server and optionally loads fixtures\n * - `beforeEach`: resets fixture match counts (not fixtures themselves)\n * - `afterAll`: stops the server\n *\n * Returns a getter function — call it inside tests to access the handle.\n */\nexport function useAimock(options: UseAimockOptions = {}): () => AimockHandle {\n  let handle: AimockHandle | null = null;\n\n  beforeAll(async () => {\n    const { fixtures: fixturePath, patchEnv, ...serverOpts } = options;\n    const llm = new LLMock(serverOpts);\n\n    if (fixturePath) {\n      const resolved = resolve(fixturePath);\n      const loadedFixtures = loadFixtures(resolved);\n      for (const f of loadedFixtures) {\n        llm.addFixture(f);\n      }\n    }\n\n    const url = await llm.start();\n\n    if (patchEnv !== false) {\n      process.env.OPENAI_BASE_URL = `${url}/v1`;\n      process.env.ANTHROPIC_BASE_URL = `${url}/v1`;\n    }\n\n    handle = { llm, url };\n  });\n\n  beforeEach(() => {\n    if (handle) {\n      handle.llm.resetMatchCounts();\n    }\n  });\n\n  afterAll(async () => {\n    if (handle) {\n      if (options.patchEnv !== false) {\n        delete process.env.OPENAI_BASE_URL;\n        delete process.env.ANTHROPIC_BASE_URL;\n      }\n      await handle.llm.stop();\n      handle = null;\n    }\n  });\n\n  return () => {\n    if (!handle) {\n      throw new Error(\"useAimock(): server not started — are you calling this inside a test?\");\n    }\n    return handle;\n  };\n}\n\nfunction loadFixtures(fixturePath: string): Fixture[] {\n  try {\n    const stat = statSync(fixturePath);\n    if (stat.isDirectory()) {\n      return loadFixturesFromDir(fixturePath);\n    }\n    return loadFixtureFile(fixturePath);\n  } catch {\n    return [];\n  }\n}\n\nexport { LLMock } from \"./llmock.js\";\nexport type { MockServerOptions, Fixture } from \"./types.js\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2CA,SAAgB,UAAU,UAA4B,EAAE,EAAsB;CAC5E,IAAI,SAA8B;AAElC,+BAAU,YAAY;EACpB,MAAM,EAAE,UAAU,aAAa,UAAU,GAAG,eAAe;EAC3D,MAAM,MAAM,IAAIA,sBAAO,WAAW;AAElC,MAAI,aAAa;GAEf,MAAM,iBAAiB,oCADE,YAAY,CACQ;AAC7C,QAAK,MAAM,KAAK,eACd,KAAI,WAAW,EAAE;;EAIrB,MAAM,MAAM,MAAM,IAAI,OAAO;AAE7B,MAAI,aAAa,OAAO;AACtB,WAAQ,IAAI,kBAAkB,GAAG,IAAI;AACrC,WAAQ,IAAI,qBAAqB,GAAG,IAAI;;AAG1C,WAAS;GAAE;GAAK;GAAK;GACrB;AAEF,sCAAiB;AACf,MAAI,OACF,QAAO,IAAI,kBAAkB;GAE/B;AAEF,8BAAS,YAAY;AACnB,MAAI,QAAQ;AACV,OAAI,QAAQ,aAAa,OAAO;AAC9B,WAAO,QAAQ,IAAI;AACnB,WAAO,QAAQ,IAAI;;AAErB,SAAM,OAAO,IAAI,MAAM;AACvB,YAAS;;GAEX;AAEF,cAAa;AACX,MAAI,CAAC,OACH,OAAM,IAAI,MAAM,wEAAwE;AAE1F,SAAO;;;AAIX,SAAS,aAAa,aAAgC;AACpD,KAAI;AAEF,4BADsB,YAAY,CACzB,aAAa,CACpB,QAAOC,2CAAoB,YAAY;AAEzC,SAAOC,uCAAgB,YAAY;SAC7B;AACN,SAAO,EAAE"}

package/dist/vitest.d.cts

@@ -0,0 +1,30 @@
+import { Fixture, MockServerOptions } from "./types.cjs";
+import { LLMock } from "./llmock.cjs";
+
+//#region src/vitest.d.ts
+
+interface UseAimockOptions extends MockServerOptions {
+  /** Path to fixture file or directory. Loaded automatically on start. */
+  fixtures?: string;
+  /** If true, sets process.env.OPENAI_BASE_URL to the mock URL + /v1. */
+  patchEnv?: boolean;
+}
+interface AimockHandle {
+  /** The LLMock instance. */
+  readonly llm: LLMock;
+  /** The server URL (e.g., http://127.0.0.1:4010). */
+  readonly url: string;
+}
+/**
+ * Start an aimock server for the duration of the test suite.
+ *
+ * - `beforeAll`: starts the server and optionally loads fixtures
+ * - `beforeEach`: resets fixture match counts (not fixtures themselves)
+ * - `afterAll`: stops the server
+ *
+ * Returns a getter function — call it inside tests to access the handle.
+ */
+declare function useAimock(options?: UseAimockOptions): () => AimockHandle;
+//#endregion
+export { AimockHandle, type Fixture, LLMock, type MockServerOptions, UseAimockOptions, useAimock };
+//# sourceMappingURL=vitest.d.cts.map

package/dist/vitest.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vitest.d.cts","names":[],"sources":["../src/vitest.ts"],"sourcesContent":[],"mappings":";;;;;UAoBiB,gBAAA,SAAyB;;;;;;UAOzB,YAAA;;gBAED;;;;;;;;;;;;;iBAcA,SAAA,WAAmB,yBAA8B"}

package/dist/vitest.d.ts

@@ -0,0 +1,30 @@
+import { Fixture, MockServerOptions } from "./types.js";
+import { LLMock } from "./llmock.js";
+
+//#region src/vitest.d.ts
+
+interface UseAimockOptions extends MockServerOptions {
+  /** Path to fixture file or directory. Loaded automatically on start. */
+  fixtures?: string;
+  /** If true, sets process.env.OPENAI_BASE_URL to the mock URL + /v1. */
+  patchEnv?: boolean;
+}
+interface AimockHandle {
+  /** The LLMock instance. */
+  readonly llm: LLMock;
+  /** The server URL (e.g., http://127.0.0.1:4010). */
+  readonly url: string;
+}
+/**
+ * Start an aimock server for the duration of the test suite.
+ *
+ * - `beforeAll`: starts the server and optionally loads fixtures
+ * - `beforeEach`: resets fixture match counts (not fixtures themselves)
+ * - `afterAll`: stops the server
+ *
+ * Returns a getter function — call it inside tests to access the handle.
+ */
+declare function useAimock(options?: UseAimockOptions): () => AimockHandle;
+//#endregion
+export { AimockHandle, type Fixture, LLMock, type MockServerOptions, UseAimockOptions, useAimock };
+//# sourceMappingURL=vitest.d.ts.map

package/dist/vitest.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vitest.d.ts","names":[],"sources":["../src/vitest.ts"],"sourcesContent":[],"mappings":";;;;;UAoBiB,gBAAA,SAAyB;;;;;;UAOzB,YAAA;;gBAED;;;;;;;;;;;;;iBAcA,SAAA,WAAmB,yBAA8B"}

package/dist/vitest.js

@@ -0,0 +1,77 @@
+import { loadFixtureFile, loadFixturesFromDir } from "./fixture-loader.js";
+import { LLMock } from "./llmock.js";
+import { afterAll, beforeAll, beforeEach } from "./node_modules/.pnpm/@vitest_runner@3.2.4/node_modules/@vitest/runner/dist/chunk-hooks.js";
+import { resolve } from "node:path";
+import { statSync } from "node:fs";
+
+//#region src/vitest.ts
+/**
+* Vitest integration for aimock.
+*
+* Usage:
+*   import { useAimock } from "@copilotkit/aimock/vitest";
+*
+*   const mock = useAimock({ fixtures: "./fixtures" });
+*
+*   it("responds", async () => {
+*     const res = await fetch(`${mock().url}/v1/chat/completions`, { ... });
+*   });
+*/
+/**
+* Start an aimock server for the duration of the test suite.
+*
+* - `beforeAll`: starts the server and optionally loads fixtures
+* - `beforeEach`: resets fixture match counts (not fixtures themselves)
+* - `afterAll`: stops the server
+*
+* Returns a getter function — call it inside tests to access the handle.
+*/
+function useAimock(options = {}) {
+	let handle = null;
+	beforeAll(async () => {
+		const { fixtures: fixturePath, patchEnv, ...serverOpts } = options;
+		const llm = new LLMock(serverOpts);
+		if (fixturePath) {
+			const loadedFixtures = loadFixtures(resolve(fixturePath));
+			for (const f of loadedFixtures) llm.addFixture(f);
+		}
+		const url = await llm.start();
+		if (patchEnv !== false) {
+			process.env.OPENAI_BASE_URL = `${url}/v1`;
+			process.env.ANTHROPIC_BASE_URL = `${url}/v1`;
+		}
+		handle = {
+			llm,
+			url
+		};
+	});
+	beforeEach(() => {
+		if (handle) handle.llm.resetMatchCounts();
+	});
+	afterAll(async () => {
+		if (handle) {
+			if (options.patchEnv !== false) {
+				delete process.env.OPENAI_BASE_URL;
+				delete process.env.ANTHROPIC_BASE_URL;
+			}
+			await handle.llm.stop();
+			handle = null;
+		}
+	});
+	return () => {
+		if (!handle) throw new Error("useAimock(): server not started — are you calling this inside a test?");
+		return handle;
+	};
+}
+function loadFixtures(fixturePath) {
+	try {
+		if (statSync(fixturePath).isDirectory()) return loadFixturesFromDir(fixturePath);
+		return loadFixtureFile(fixturePath);
+	} catch {
+		return [];
+	}
+}
+
+//#endregion
+export { LLMock, useAimock };
+//# sourceMappingURL=vitest.js.map

package/dist/vitest.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vitest.js","names":[],"sources":["../src/vitest.ts"],"sourcesContent":["/**\n * Vitest integration for aimock.\n *\n * Usage:\n *   import { useAimock } from \"@copilotkit/aimock/vitest\";\n *\n *   const mock = useAimock({ fixtures: \"./fixtures\" });\n *\n *   it(\"responds\", async () => {\n *     const res = await fetch(`${mock().url}/v1/chat/completions`, { ... });\n *   });\n */\n\nimport { beforeAll, afterAll, beforeEach } from \"vitest\";\nimport { LLMock } from \"./llmock.js\";\nimport { loadFixtureFile, loadFixturesFromDir } from \"./fixture-loader.js\";\nimport type { Fixture, MockServerOptions } from \"./types.js\";\nimport { statSync } from \"node:fs\";\nimport { resolve } from \"node:path\";\n\nexport interface UseAimockOptions extends MockServerOptions {\n  /** Path to fixture file or directory. Loaded automatically on start. */\n  fixtures?: string;\n  /** If true, sets process.env.OPENAI_BASE_URL to the mock URL + /v1. */\n  patchEnv?: boolean;\n}\n\nexport interface AimockHandle {\n  /** The LLMock instance. */\n  readonly llm: LLMock;\n  /** The server URL (e.g., http://127.0.0.1:4010). */\n  readonly url: string;\n}\n\n/**\n * Start an aimock server for the duration of the test suite.\n *\n * - `beforeAll`: starts the server and optionally loads fixtures\n * - `beforeEach`: resets fixture match counts (not fixtures themselves)\n * - `afterAll`: stops the server\n *\n * Returns a getter function — call it inside tests to access the handle.\n */\nexport function useAimock(options: UseAimockOptions = {}): () => AimockHandle {\n  let handle: AimockHandle | null = null;\n\n  beforeAll(async () => {\n    const { fixtures: fixturePath, patchEnv, ...serverOpts } = options;\n    const llm = new LLMock(serverOpts);\n\n    if (fixturePath) {\n      const resolved = resolve(fixturePath);\n      const loadedFixtures = loadFixtures(resolved);\n      for (const f of loadedFixtures) {\n        llm.addFixture(f);\n      }\n    }\n\n    const url = await llm.start();\n\n    if (patchEnv !== false) {\n      process.env.OPENAI_BASE_URL = `${url}/v1`;\n      process.env.ANTHROPIC_BASE_URL = `${url}/v1`;\n    }\n\n    handle = { llm, url };\n  });\n\n  beforeEach(() => {\n    if (handle) {\n      handle.llm.resetMatchCounts();\n    }\n  });\n\n  afterAll(async () => {\n    if (handle) {\n      if (options.patchEnv !== false) {\n        delete process.env.OPENAI_BASE_URL;\n        delete process.env.ANTHROPIC_BASE_URL;\n      }\n      await handle.llm.stop();\n      handle = null;\n    }\n  });\n\n  return () => {\n    if (!handle) {\n      throw new Error(\"useAimock(): server not started — are you calling this inside a test?\");\n    }\n    return handle;\n  };\n}\n\nfunction loadFixtures(fixturePath: string): Fixture[] {\n  try {\n    const stat = statSync(fixturePath);\n    if (stat.isDirectory()) {\n      return loadFixturesFromDir(fixturePath);\n    }\n    return loadFixtureFile(fixturePath);\n  } catch {\n    return [];\n  }\n}\n\nexport { LLMock } from \"./llmock.js\";\nexport type { MockServerOptions, Fixture } from \"./types.js\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2CA,SAAgB,UAAU,UAA4B,EAAE,EAAsB;CAC5E,IAAI,SAA8B;AAElC,WAAU,YAAY;EACpB,MAAM,EAAE,UAAU,aAAa,UAAU,GAAG,eAAe;EAC3D,MAAM,MAAM,IAAI,OAAO,WAAW;AAElC,MAAI,aAAa;GAEf,MAAM,iBAAiB,aADN,QAAQ,YAAY,CACQ;AAC7C,QAAK,MAAM,KAAK,eACd,KAAI,WAAW,EAAE;;EAIrB,MAAM,MAAM,MAAM,IAAI,OAAO;AAE7B,MAAI,aAAa,OAAO;AACtB,WAAQ,IAAI,kBAAkB,GAAG,IAAI;AACrC,WAAQ,IAAI,qBAAqB,GAAG,IAAI;;AAG1C,WAAS;GAAE;GAAK;GAAK;GACrB;AAEF,kBAAiB;AACf,MAAI,OACF,QAAO,IAAI,kBAAkB;GAE/B;AAEF,UAAS,YAAY;AACnB,MAAI,QAAQ;AACV,OAAI,QAAQ,aAAa,OAAO;AAC9B,WAAO,QAAQ,IAAI;AACnB,WAAO,QAAQ,IAAI;;AAErB,SAAM,OAAO,IAAI,MAAM;AACvB,YAAS;;GAEX;AAEF,cAAa;AACX,MAAI,CAAC,OACH,OAAM,IAAI,MAAM,wEAAwE;AAE1F,SAAO;;;AAIX,SAAS,aAAa,aAAgC;AACpD,KAAI;AAEF,MADa,SAAS,YAAY,CACzB,aAAa,CACpB,QAAO,oBAAoB,YAAY;AAEzC,SAAO,gBAAgB,YAAY;SAC7B;AACN,SAAO,EAAE"}

package/fixtures/examples/a2a/a2a-config.json

@@ -0,0 +1,42 @@
+{
+  "a2a": {
+    "agents": [
+      {
+        "name": "research-agent",
+        "description": "An agent that researches topics and returns summaries",
+        "version": "1.0.0",
+        "skills": [
+          {
+            "id": "web-research",
+            "name": "Web Research",
+            "description": "Search the web and summarize findings",
+            "tags": ["research", "search"]
+          }
+        ],
+        "capabilities": { "streaming": true },
+        "messages": [
+          {
+            "pattern": "research",
+            "parts": [{ "text": "Here is a summary of my research findings on the topic." }]
+          }
+        ],
+        "streamingTasks": [
+          {
+            "pattern": "deep-research",
+            "events": [
+              { "type": "status", "state": "TASK_STATE_WORKING" },
+              {
+                "type": "artifact",
+                "name": "research-report",
+                "parts": [{ "text": "## Research Report\n\nFindings from deep research..." }],
+                "lastChunk": true
+              },
+              { "type": "status", "state": "TASK_STATE_COMPLETED" }
+            ],
+            "delayMs": 100
+          }
+        ]
+      }
+    ]
+  }
+}

package/fixtures/examples/agui/agui-text-response.json

@@ -0,0 +1,35 @@
+{
+  "agui": {
+    "fixtures": [
+      {
+        "match": { "message": "hello" },
+        "text": "Hi! How can I help you today?"
+      },
+      {
+        "match": { "message": "search" },
+        "events": [
+          { "type": "RUN_STARTED", "threadId": "t1", "runId": "r1" },
+          { "type": "TOOL_CALL_START", "toolCallId": "tc1", "toolCallName": "web_search" },
+          {
+            "type": "TOOL_CALL_ARGS",
+            "toolCallId": "tc1",
+            "delta": "{\"query\": \"latest news\"}"
+          },
+          { "type": "TOOL_CALL_END", "toolCallId": "tc1" },
+          {
+            "type": "TEXT_MESSAGE_START",
+            "messageId": "m1",
+            "role": "assistant"
+          },
+          {
+            "type": "TEXT_MESSAGE_CONTENT",
+            "messageId": "m1",
+            "delta": "Here are the latest results..."
+          },
+          { "type": "TEXT_MESSAGE_END", "messageId": "m1" },
+          { "type": "RUN_FINISHED", "threadId": "t1", "runId": "r1" }
+        ]
+      }
+    ]
+  }
+}

package/fixtures/examples/chaos/chaos-config.json

@@ -0,0 +1,10 @@
+{
+  "llm": {
+    "fixtures": "fixtures/example-greeting.json",
+    "chaos": {
+      "dropRate": 0.1,
+      "malformedRate": 0.05,
+      "disconnectRate": 0.02
+    }
+  }
+}

package/fixtures/examples/full-suite.json

@@ -0,0 +1,116 @@
+{
+  "port": 4000,
+  "host": "127.0.0.1",
+  "metrics": true,
+  "strict": true,
+
+  "llm": {
+    "fixtures": "fixtures/example-greeting.json",
+    "chaos": {
+      "dropRate": 0.01,
+      "malformedRate": 0.005,
+      "disconnectRate": 0.002
+    }
+  },
+
+  "mcp": {
+    "serverInfo": { "name": "full-suite-mcp", "version": "1.0.0" },
+    "tools": [
+      {
+        "name": "search",
+        "description": "Search the knowledge base",
+        "inputSchema": {
+          "type": "object",
+          "properties": {
+            "query": { "type": "string" },
+            "limit": { "type": "number" }
+          },
+          "required": ["query"]
+        },
+        "result": "Found 3 results for your query."
+      }
+    ],
+    "resources": [
+      {
+        "uri": "file:///config",
+        "name": "Configuration",
+        "mimeType": "application/json",
+        "text": "{\"version\": \"1.0\", \"environment\": \"test\"}"
+      }
+    ],
+    "prompts": [
+      {
+        "name": "summarize",
+        "description": "Summarize a document",
+        "arguments": [{ "name": "text", "description": "The text to summarize", "required": true }],
+        "result": {
+          "messages": [
+            {
+              "role": "assistant",
+              "content": { "type": "text", "text": "Here is a summary of the provided text." }
+            }
+          ]
+        }
+      }
+    ]
+  },
+
+  "a2a": {
+    "agents": [
+      {
+        "name": "assistant",
+        "description": "A general-purpose assistant agent",
+        "version": "1.0.0",
+        "skills": [{ "id": "qa", "name": "Q&A", "description": "Answer questions" }],
+        "capabilities": { "streaming": true },
+        "messages": [
+          {
+            "pattern": ".*",
+            "parts": [{ "text": "I can help you with that." }]
+          }
+        ]
+      }
+    ]
+  },
+
+  "agui": {
+    "fixtures": [
+      {
+        "match": { "message": "hello" },
+        "text": "Hello from the full-suite mock!"
+      },
+      {
+        "match": { "toolName": "get_data" },
+        "events": [
+          { "type": "RUN_STARTED", "threadId": "t1", "runId": "r1" },
+          { "type": "TOOL_CALL_START", "toolCallId": "tc1", "toolCallName": "get_data" },
+          { "type": "TOOL_CALL_ARGS", "toolCallId": "tc1", "delta": "{}" },
+          { "type": "TOOL_CALL_END", "toolCallId": "tc1" },
+          { "type": "RUN_FINISHED", "threadId": "t1", "runId": "r1" }
+        ]
+      }
+    ]
+  },
+
+  "vector": {
+    "collections": [
+      {
+        "name": "knowledge-base",
+        "dimension": 384,
+        "queryResults": [
+          {
+            "id": "kb-001",
+            "score": 0.97,
+            "metadata": { "source": "docs", "title": "Quick Start Guide" }
+          }
+        ]
+      }
+    ]
+  },
+
+  "services": {
+    "search": true,
+    "rerank": true,
+    "moderate": true
+  }
+}