@copilotkit/aimock 1.15.0 → 1.15.1

package/dist/video.js

@@ -5,6 +5,49 @@ import { applyChaos } from "./chaos.js";
 import { proxyAndRecord } from "./recorder.js";
 
 //#region src/video.ts
+const VIDEO_STATE_MAX_ENTRIES = 1e4;
+const VIDEO_STATE_TTL_MS = 36e5;
+/**
+* A Map wrapper for video state that enforces a maximum size and per-entry TTL.
+* Entries older than VIDEO_STATE_TTL_MS are lazily evicted on `get`.
+* When the map exceeds VIDEO_STATE_MAX_ENTRIES on `set`, the oldest entries
+* are removed to stay within bounds.
+*/
+var VideoStateMap = class {
+	entries = /* @__PURE__ */ new Map();
+	get(key) {
+		const entry = this.entries.get(key);
+		if (!entry) return void 0;
+		if (Date.now() - entry.createdAt > VIDEO_STATE_TTL_MS) {
+			this.entries.delete(key);
+			return;
+		}
+		return entry.video;
+	}
+	set(key, video) {
+		this.entries.set(key, {
+			video,
+			createdAt: Date.now()
+		});
+		if (this.entries.size > VIDEO_STATE_MAX_ENTRIES) {
+			const excess = this.entries.size - VIDEO_STATE_MAX_ENTRIES;
+			const iter = this.entries.keys();
+			for (let i = 0; i < excess; i++) {
+				const next = iter.next();
+				if (!next.done) this.entries.delete(next.value);
+			}
+		}
+	}
+	delete(key) {
+		return this.entries.delete(key);
+	}
+	clear() {
+		this.entries.clear();
+	}
+	get size() {
+		return this.entries.size;
+	}
+};
 async function handleVideoCreate(req, res, raw, fixtures, journal, defaults, setCorsHeaders, videoStates) {
 	setCorsHeaders(res);
 	const path = req.url ?? "/v1/videos";
@@ -30,6 +73,23 @@ async function handleVideoCreate(req, res, raw, fixtures, journal, defaults, set
 		} }));
 		return;
 	}
+	if (!videoReq.prompt) {
+		journal.add({
+			method,
+			path,
+			headers: flattenHeaders(req.headers),
+			body: null,
+			response: {
+				status: 400,
+				fixture: null
+			}
+		});
+		writeErrorResponse(res, 400, JSON.stringify({ error: {
+			message: "Missing required parameter: 'prompt'",
+			type: "invalid_request_error"
+		} }));
+		return;
+	}
 	const syntheticReq = {
 		model: videoReq.model ?? "sora-2",
 		messages: [{
@@ -46,10 +106,10 @@ async function handleVideoCreate(req, res, raw, fixtures, journal, defaults, set
 		path,
 		headers: flattenHeaders(req.headers),
 		body: syntheticReq
-	}, fixture ? "fixture" : "proxy", defaults.registry, defaults.logger)) return;
+	}, defaults.registry, defaults.logger)) return;
 	if (!fixture) {
 		if (defaults.record) {
-			if (await proxyAndRecord(req, res, syntheticReq, "openai", req.url ?? "/v1/videos", fixtures, defaults, raw) !== "not_configured") {
+			if (await proxyAndRecord(req, res, syntheticReq, "openai", req.url ?? "/v1/videos", fixtures, defaults, raw)) {
 				journal.add({
 					method,
 					path,
@@ -57,7 +117,8 @@ async function handleVideoCreate(req, res, raw, fixtures, journal, defaults, set
 					body: syntheticReq,
 					response: {
 						status: res.statusCode ?? 200,
-						fixture: null
+						fixture: null,
+						source: "proxy"
 					}
 				});
 				return;
@@ -146,16 +207,10 @@ async function handleVideoCreate(req, res, raw, fixtures, journal, defaults, set
 		}));
 	}
 }
-function handleVideoStatus(req, res, videoId, journal, defaults, setCorsHeaders, videoStates) {
+function handleVideoStatus(req, res, videoId, journal, setCorsHeaders, videoStates) {
 	setCorsHeaders(res);
 	const path = req.url ?? `/v1/videos/${videoId}`;
 	const method = req.method ?? "GET";
-	if (applyChaos(res, null, defaults.chaos, req.headers, journal, {
-		method,
-		path,
-		headers: flattenHeaders(req.headers),
-		body: null
-	}, "fixture", defaults.registry, defaults.logger)) return;
 	const stateKey = `${getTestId(req)}:${videoId}`;
 	const video = videoStates.get(stateKey);
 	if (!video) {
@@ -197,5 +252,5 @@ function handleVideoStatus(req, res, videoId, journal, defaults, setCorsHeaders,
 }
 
 //#endregion
-export { handleVideoCreate, handleVideoStatus };
+export { VideoStateMap, handleVideoCreate, handleVideoStatus };
 //# sourceMappingURL=video.js.map

package/dist/video.js.map

@@ -1 +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      fixture ? \"fixture\" : \"proxy\",\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\n\n  if (!fixture) {\n    if (defaults.record) {\n      const outcome = await proxyAndRecord(\n        req,\n        res,\n        syntheticReq,\n        \"openai\",\n        req.url ?? \"/v1/videos\",\n        fixtures,\n        defaults,\n        raw,\n      );\n      if (outcome !== \"not_configured\") {\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  defaults: HandlerDefaults,\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  if (\n    applyChaos(\n      res,\n      null,\n      defaults.chaos,\n      req.headers,\n      journal,\n      { method, path, headers: flattenHeaders(req.headers), body: null },\n      \"fixture\",\n      defaults.registry,\n      defaults.logger,\n    )\n  )\n    return;\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,UAAU,YAAY,SACtB,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,KACe,kBAAkB;AAChC,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,UACA,gBACA,aACM;AACN,gBAAe,IAAI;CACnB,MAAM,OAAO,IAAI,OAAO,cAAc;CACtC,MAAM,SAAS,IAAI,UAAU;AAE7B,KACE,WACE,KACA,MACA,SAAS,OACT,IAAI,SACJ,SACA;EAAE;EAAQ;EAAM,SAAS,eAAe,IAAI,QAAQ;EAAE,MAAM;EAAM,EAClE,WACA,SAAS,UACT,SAAS,OACV,CAED;CAGF,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"}
+{"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// ─── VideoStateMap with TTL and size bound ────────────────────────────────\n\nconst VIDEO_STATE_MAX_ENTRIES = 10_000;\nconst VIDEO_STATE_TTL_MS = 3_600_000; // 1 hour\n\ninterface VideoStateEntry {\n  video: VideoResponse[\"video\"];\n  createdAt: number;\n}\n\n/**\n * A Map wrapper for video state that enforces a maximum size and per-entry TTL.\n * Entries older than VIDEO_STATE_TTL_MS are lazily evicted on `get`.\n * When the map exceeds VIDEO_STATE_MAX_ENTRIES on `set`, the oldest entries\n * are removed to stay within bounds.\n */\nexport class VideoStateMap {\n  private readonly entries = new Map<string, VideoStateEntry>();\n\n  get(key: string): VideoResponse[\"video\"] | undefined {\n    const entry = this.entries.get(key);\n    if (!entry) return undefined;\n    if (Date.now() - entry.createdAt > VIDEO_STATE_TTL_MS) {\n      this.entries.delete(key);\n      return undefined;\n    }\n    return entry.video;\n  }\n\n  set(key: string, video: VideoResponse[\"video\"]): void {\n    this.entries.set(key, { video, createdAt: Date.now() });\n    // Evict oldest entries if over capacity\n    if (this.entries.size > VIDEO_STATE_MAX_ENTRIES) {\n      const excess = this.entries.size - VIDEO_STATE_MAX_ENTRIES;\n      const iter = this.entries.keys();\n      for (let i = 0; i < excess; i++) {\n        const next = iter.next();\n        if (!next.done) this.entries.delete(next.value);\n      }\n    }\n  }\n\n  delete(key: string): boolean {\n    return this.entries.delete(key);\n  }\n\n  clear(): void {\n    this.entries.clear();\n  }\n\n  get size(): number {\n    return this.entries.size;\n  }\n}\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  if (!videoReq.prompt) {\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: \"Missing required parameter: 'prompt'\", type: \"invalid_request_error\" },\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, source: \"proxy\" },\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":";;;;;;;AAiBA,MAAM,0BAA0B;AAChC,MAAM,qBAAqB;;;;;;;AAa3B,IAAa,gBAAb,MAA2B;CACzB,AAAiB,0BAAU,IAAI,KAA8B;CAE7D,IAAI,KAAiD;EACnD,MAAM,QAAQ,KAAK,QAAQ,IAAI,IAAI;AACnC,MAAI,CAAC,MAAO,QAAO;AACnB,MAAI,KAAK,KAAK,GAAG,MAAM,YAAY,oBAAoB;AACrD,QAAK,QAAQ,OAAO,IAAI;AACxB;;AAEF,SAAO,MAAM;;CAGf,IAAI,KAAa,OAAqC;AACpD,OAAK,QAAQ,IAAI,KAAK;GAAE;GAAO,WAAW,KAAK,KAAK;GAAE,CAAC;AAEvD,MAAI,KAAK,QAAQ,OAAO,yBAAyB;GAC/C,MAAM,SAAS,KAAK,QAAQ,OAAO;GACnC,MAAM,OAAO,KAAK,QAAQ,MAAM;AAChC,QAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,KAAK;IAC/B,MAAM,OAAO,KAAK,MAAM;AACxB,QAAI,CAAC,KAAK,KAAM,MAAK,QAAQ,OAAO,KAAK,MAAM;;;;CAKrD,OAAO,KAAsB;AAC3B,SAAO,KAAK,QAAQ,OAAO,IAAI;;CAGjC,QAAc;AACZ,OAAK,QAAQ,OAAO;;CAGtB,IAAI,OAAe;AACjB,SAAO,KAAK,QAAQ;;;AAIxB,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;;AAGF,KAAI,CAAC,SAAS,QAAQ;AACpB,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;GAAwC,MAAM;GAAyB,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,QAAQ;MAAS;KAC5E,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

@@ -30,6 +30,8 @@ let vitest = require("vitest");
 */
 function useAimock(options = {}) {
 	let handle = null;
+	let origOpenaiUrl;
+	let origAnthropicUrl;
 	(0, vitest.beforeAll)(async () => {
 		const { fixtures: fixturePath, patchEnv, ...serverOpts } = options;
 		const llm = new require_llmock.LLMock(serverOpts);
@@ -39,6 +41,8 @@ function useAimock(options = {}) {
 		}
 		const url = await llm.start();
 		if (patchEnv !== false) {
+			origOpenaiUrl = process.env.OPENAI_BASE_URL;
+			origAnthropicUrl = process.env.ANTHROPIC_BASE_URL;
 			process.env.OPENAI_BASE_URL = `${url}/v1`;
 			process.env.ANTHROPIC_BASE_URL = `${url}/v1`;
 		}
@@ -53,8 +57,10 @@ function useAimock(options = {}) {
 	(0, vitest.afterAll)(async () => {
 		if (handle) {
 			if (options.patchEnv !== false) {
-				delete process.env.OPENAI_BASE_URL;
-				delete process.env.ANTHROPIC_BASE_URL;
+				if (origOpenaiUrl !== void 0) process.env.OPENAI_BASE_URL = origOpenaiUrl;
+				else delete process.env.OPENAI_BASE_URL;
+				if (origAnthropicUrl !== void 0) process.env.ANTHROPIC_BASE_URL = origAnthropicUrl;
+				else delete process.env.ANTHROPIC_BASE_URL;
 			}
 			await handle.llm.stop();
 			handle = null;
@@ -69,7 +75,8 @@ function loadFixtures(fixturePath) {
 	try {
 		if ((0, node_fs.statSync)(fixturePath).isDirectory()) return require_fixture_loader.loadFixturesFromDir(fixturePath);
 		return require_fixture_loader.loadFixtureFile(fixturePath);
-	} catch {
+	} catch (err) {
+		console.warn(`[aimock] Failed to load fixtures from ${fixturePath}: ${err instanceof Error ? err.message : String(err)}`);
 		return [];
 	}
 }

package/dist/vitest.cjs.map

@@ -1 +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,uBAAU,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,8BAAiB;AACf,MAAI,OACF,QAAO,IAAI,kBAAkB;GAE/B;AAEF,sBAAS,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"}
+{"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  let origOpenaiUrl: string | undefined;\n  let origAnthropicUrl: string | undefined;\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      origOpenaiUrl = process.env.OPENAI_BASE_URL;\n      origAnthropicUrl = process.env.ANTHROPIC_BASE_URL;\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        if (origOpenaiUrl !== undefined) process.env.OPENAI_BASE_URL = origOpenaiUrl;\n        else delete process.env.OPENAI_BASE_URL;\n        if (origAnthropicUrl !== undefined) process.env.ANTHROPIC_BASE_URL = origAnthropicUrl;\n        else 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 (err) {\n    console.warn(\n      `[aimock] Failed to load fixtures from ${fixturePath}: ${err instanceof Error ? err.message : String(err)}`,\n    );\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;CAClC,IAAI;CACJ,IAAI;AAEJ,uBAAU,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,mBAAgB,QAAQ,IAAI;AAC5B,sBAAmB,QAAQ,IAAI;AAC/B,WAAQ,IAAI,kBAAkB,GAAG,IAAI;AACrC,WAAQ,IAAI,qBAAqB,GAAG,IAAI;;AAG1C,WAAS;GAAE;GAAK;GAAK;GACrB;AAEF,8BAAiB;AACf,MAAI,OACF,QAAO,IAAI,kBAAkB;GAE/B;AAEF,sBAAS,YAAY;AACnB,MAAI,QAAQ;AACV,OAAI,QAAQ,aAAa,OAAO;AAC9B,QAAI,kBAAkB,OAAW,SAAQ,IAAI,kBAAkB;QAC1D,QAAO,QAAQ,IAAI;AACxB,QAAI,qBAAqB,OAAW,SAAQ,IAAI,qBAAqB;QAChE,QAAO,QAAQ,IAAI;;AAE1B,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;UAC5B,KAAK;AACZ,UAAQ,KACN,yCAAyC,YAAY,IAAI,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,GAC1G;AACD,SAAO,EAAE"}

package/dist/vitest.js

@@ -28,6 +28,8 @@ import { afterAll, beforeAll, beforeEach } from "vitest";
 */
 function useAimock(options = {}) {
 	let handle = null;
+	let origOpenaiUrl;
+	let origAnthropicUrl;
 	beforeAll(async () => {
 		const { fixtures: fixturePath, patchEnv, ...serverOpts } = options;
 		const llm = new LLMock(serverOpts);
@@ -37,6 +39,8 @@ function useAimock(options = {}) {
 		}
 		const url = await llm.start();
 		if (patchEnv !== false) {
+			origOpenaiUrl = process.env.OPENAI_BASE_URL;
+			origAnthropicUrl = process.env.ANTHROPIC_BASE_URL;
 			process.env.OPENAI_BASE_URL = `${url}/v1`;
 			process.env.ANTHROPIC_BASE_URL = `${url}/v1`;
 		}
@@ -51,8 +55,10 @@ function useAimock(options = {}) {
 	afterAll(async () => {
 		if (handle) {
 			if (options.patchEnv !== false) {
-				delete process.env.OPENAI_BASE_URL;
-				delete process.env.ANTHROPIC_BASE_URL;
+				if (origOpenaiUrl !== void 0) process.env.OPENAI_BASE_URL = origOpenaiUrl;
+				else delete process.env.OPENAI_BASE_URL;
+				if (origAnthropicUrl !== void 0) process.env.ANTHROPIC_BASE_URL = origAnthropicUrl;
+				else delete process.env.ANTHROPIC_BASE_URL;
 			}
 			await handle.llm.stop();
 			handle = null;
@@ -67,7 +73,8 @@ function loadFixtures(fixturePath) {
 	try {
 		if (statSync(fixturePath).isDirectory()) return loadFixturesFromDir(fixturePath);
 		return loadFixtureFile(fixturePath);
-	} catch {
+	} catch (err) {
+		console.warn(`[aimock] Failed to load fixtures from ${fixturePath}: ${err instanceof Error ? err.message : String(err)}`);
 		return [];
 	}
 }

package/dist/vitest.js.map

@@ -1 +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"}
+{"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  let origOpenaiUrl: string | undefined;\n  let origAnthropicUrl: string | undefined;\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      origOpenaiUrl = process.env.OPENAI_BASE_URL;\n      origAnthropicUrl = process.env.ANTHROPIC_BASE_URL;\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        if (origOpenaiUrl !== undefined) process.env.OPENAI_BASE_URL = origOpenaiUrl;\n        else delete process.env.OPENAI_BASE_URL;\n        if (origAnthropicUrl !== undefined) process.env.ANTHROPIC_BASE_URL = origAnthropicUrl;\n        else 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 (err) {\n    console.warn(\n      `[aimock] Failed to load fixtures from ${fixturePath}: ${err instanceof Error ? err.message : String(err)}`,\n    );\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;CAClC,IAAI;CACJ,IAAI;AAEJ,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,mBAAgB,QAAQ,IAAI;AAC5B,sBAAmB,QAAQ,IAAI;AAC/B,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,QAAI,kBAAkB,OAAW,SAAQ,IAAI,kBAAkB;QAC1D,QAAO,QAAQ,IAAI;AACxB,QAAI,qBAAqB,OAAW,SAAQ,IAAI,qBAAqB;QAChE,QAAO,QAAQ,IAAI;;AAE1B,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;UAC5B,KAAK;AACZ,UAAQ,KACN,yCAAyC,YAAY,IAAI,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,GAC1G;AACD,SAAO,EAAE"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkit/aimock",
-  "version": "1.15.0",
+  "version": "1.15.1",
   "description": "Mock infrastructure for AI application testing — LLM APIs, image generation, text-to-speech, transcription, video generation, MCP tools, A2A agents, AG-UI event streams, vector databases, search, rerank, and moderation. One package, one port, zero dependencies.",
   "license": "MIT",
   "keywords": [