@copilotkit/aimock 1.7.0

package/dist/moderation.js

@@ -0,0 +1,91 @@
+import { flattenHeaders, generateId, matchesPattern } from "./helpers.js";
+
+//#region src/moderation.ts
+const DEFAULT_RESULT = {
+	flagged: false,
+	categories: {
+		sexual: false,
+		hate: false,
+		harassment: false,
+		"self-harm": false,
+		"sexual/minors": false,
+		"hate/threatening": false,
+		"violence/graphic": false,
+		"self-harm/intent": false,
+		"self-harm/instructions": false,
+		"harassment/threatening": false,
+		violence: false
+	},
+	category_scores: {
+		sexual: 0,
+		hate: 0,
+		harassment: 0,
+		"self-harm": 0,
+		"sexual/minors": 0,
+		"hate/threatening": 0,
+		"violence/graphic": 0,
+		"self-harm/intent": 0,
+		"self-harm/instructions": 0,
+		"harassment/threatening": 0,
+		violence: 0
+	}
+};
+async function handleModeration(req, res, raw, fixtures, journal, defaults, setCorsHeaders) {
+	const { logger } = defaults;
+	setCorsHeaders(res);
+	let body;
+	try {
+		body = JSON.parse(raw);
+	} catch {
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "/v1/moderations",
+			headers: flattenHeaders(req.headers),
+			body: null,
+			service: "moderation",
+			response: {
+				status: 400,
+				fixture: null
+			}
+		});
+		res.writeHead(400, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ error: {
+			message: "Malformed JSON",
+			type: "invalid_request_error",
+			code: "invalid_json"
+		} }));
+		return;
+	}
+	const rawInput = body.input ?? "";
+	const inputText = Array.isArray(rawInput) ? rawInput.join(" ") : rawInput;
+	let matchedResult = DEFAULT_RESULT;
+	let matchedFixture = null;
+	for (const fixture of fixtures) if (matchesPattern(inputText, fixture.match)) {
+		matchedFixture = fixture;
+		matchedResult = fixture.result;
+		break;
+	}
+	if (matchedFixture) logger.debug(`Moderation fixture matched for input "${inputText.slice(0, 80)}"`);
+	else logger.debug(`No moderation fixture matched for input "${inputText.slice(0, 80)}" — returning unflagged`);
+	journal.add({
+		method: req.method ?? "POST",
+		path: req.url ?? "/v1/moderations",
+		headers: flattenHeaders(req.headers),
+		body: null,
+		service: "moderation",
+		response: {
+			status: 200,
+			fixture: null
+		}
+	});
+	res.writeHead(200, { "Content-Type": "application/json" });
+	res.end(JSON.stringify({
+		id: generateId("modr"),
+		model: "text-moderation-latest",
+		results: [matchedResult]
+	}));
+}
+
+//#endregion
+export { handleModeration };
+//# sourceMappingURL=moderation.js.map

package/dist/moderation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"moderation.js","names":[],"sources":["../src/moderation.ts"],"sourcesContent":["/**\n * Moderation API support for LLMock.\n *\n * Handles POST /v1/moderations requests (OpenAI-compatible). Matches\n * fixtures by comparing the request `input` field against registered\n * patterns. First match wins; no match returns a default unflagged result.\n */\n\nimport type * as http from \"node:http\";\nimport { flattenHeaders, generateId, matchesPattern } from \"./helpers.js\";\nimport type { Journal } from \"./journal.js\";\nimport type { Logger } from \"./logger.js\";\n\n// ─── Moderation types ─────────────────────────────────────────────────────\n\nexport interface ModerationResult {\n  flagged: boolean;\n  categories: Record<string, boolean>;\n  category_scores?: Record<string, number>;\n}\n\nexport interface ModerationFixture {\n  match: string | RegExp;\n  result: ModerationResult;\n}\n\n// ─── Default unflagged result ─────────────────────────────────────────────\n\nconst DEFAULT_RESULT: ModerationResult = {\n  flagged: false,\n  categories: {\n    sexual: false,\n    hate: false,\n    harassment: false,\n    \"self-harm\": false,\n    \"sexual/minors\": false,\n    \"hate/threatening\": false,\n    \"violence/graphic\": false,\n    \"self-harm/intent\": false,\n    \"self-harm/instructions\": false,\n    \"harassment/threatening\": false,\n    violence: false,\n  },\n  category_scores: {\n    sexual: 0,\n    hate: 0,\n    harassment: 0,\n    \"self-harm\": 0,\n    \"sexual/minors\": 0,\n    \"hate/threatening\": 0,\n    \"violence/graphic\": 0,\n    \"self-harm/intent\": 0,\n    \"self-harm/instructions\": 0,\n    \"harassment/threatening\": 0,\n    violence: 0,\n  },\n};\n\n// ─── Request handler ──────────────────────────────────────────────────────\n\nexport async function handleModeration(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  raw: string,\n  fixtures: ModerationFixture[],\n  journal: Journal,\n  defaults: { logger: Logger },\n  setCorsHeaders: (res: http.ServerResponse) => void,\n): Promise<void> {\n  const { logger } = defaults;\n  setCorsHeaders(res);\n\n  let body: { input?: string | string[] };\n  try {\n    body = JSON.parse(raw) as { input?: string | string[] };\n  } catch {\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"/v1/moderations\",\n      headers: flattenHeaders(req.headers),\n      body: null,\n      service: \"moderation\",\n      response: { status: 400, fixture: null },\n    });\n    res.writeHead(400, { \"Content-Type\": \"application/json\" });\n    res.end(\n      JSON.stringify({\n        error: {\n          message: \"Malformed JSON\",\n          type: \"invalid_request_error\",\n          code: \"invalid_json\",\n        },\n      }),\n    );\n    return;\n  }\n\n  // Normalize input to a single string for matching\n  const rawInput = body.input ?? \"\";\n  const inputText = Array.isArray(rawInput) ? rawInput.join(\" \") : rawInput;\n\n  // Find first matching fixture\n  let matchedResult: ModerationResult = DEFAULT_RESULT;\n  let matchedFixture: ModerationFixture | null = null;\n\n  for (const fixture of fixtures) {\n    if (matchesPattern(inputText, fixture.match)) {\n      matchedFixture = fixture;\n      matchedResult = fixture.result;\n      break;\n    }\n  }\n\n  if (matchedFixture) {\n    logger.debug(`Moderation fixture matched for input \"${inputText.slice(0, 80)}\"`);\n  } else {\n    logger.debug(\n      `No moderation fixture matched for input \"${inputText.slice(0, 80)}\" — returning unflagged`,\n    );\n  }\n\n  journal.add({\n    method: req.method ?? \"POST\",\n    path: req.url ?? \"/v1/moderations\",\n    headers: flattenHeaders(req.headers),\n    body: null,\n    service: \"moderation\",\n    response: { status: 200, fixture: null },\n  });\n\n  res.writeHead(200, { \"Content-Type\": \"application/json\" });\n  res.end(\n    JSON.stringify({\n      id: generateId(\"modr\"),\n      model: \"text-moderation-latest\",\n      results: [matchedResult],\n    }),\n  );\n}\n"],"mappings":";;;AA4BA,MAAM,iBAAmC;CACvC,SAAS;CACT,YAAY;EACV,QAAQ;EACR,MAAM;EACN,YAAY;EACZ,aAAa;EACb,iBAAiB;EACjB,oBAAoB;EACpB,oBAAoB;EACpB,oBAAoB;EACpB,0BAA0B;EAC1B,0BAA0B;EAC1B,UAAU;EACX;CACD,iBAAiB;EACf,QAAQ;EACR,MAAM;EACN,YAAY;EACZ,aAAa;EACb,iBAAiB;EACjB,oBAAoB;EACpB,oBAAoB;EACpB,oBAAoB;EACpB,0BAA0B;EAC1B,0BAA0B;EAC1B,UAAU;EACX;CACF;AAID,eAAsB,iBACpB,KACA,KACA,KACA,UACA,SACA,UACA,gBACe;CACf,MAAM,EAAE,WAAW;AACnB,gBAAe,IAAI;CAEnB,IAAI;AACJ,KAAI;AACF,SAAO,KAAK,MAAM,IAAI;SAChB;AACN,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,SAAS;GACT,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM;GACzC,CAAC;AACF,MAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,MAAI,IACF,KAAK,UAAU,EACb,OAAO;GACL,SAAS;GACT,MAAM;GACN,MAAM;GACP,EACF,CAAC,CACH;AACD;;CAIF,MAAM,WAAW,KAAK,SAAS;CAC/B,MAAM,YAAY,MAAM,QAAQ,SAAS,GAAG,SAAS,KAAK,IAAI,GAAG;CAGjE,IAAI,gBAAkC;CACtC,IAAI,iBAA2C;AAE/C,MAAK,MAAM,WAAW,SACpB,KAAI,eAAe,WAAW,QAAQ,MAAM,EAAE;AAC5C,mBAAiB;AACjB,kBAAgB,QAAQ;AACxB;;AAIJ,KAAI,eACF,QAAO,MAAM,yCAAyC,UAAU,MAAM,GAAG,GAAG,CAAC,GAAG;KAEhF,QAAO,MACL,4CAA4C,UAAU,MAAM,GAAG,GAAG,CAAC,yBACpE;AAGH,SAAQ,IAAI;EACV,QAAQ,IAAI,UAAU;EACtB,MAAM,IAAI,OAAO;EACjB,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,SAAS;EACT,UAAU;GAAE,QAAQ;GAAK,SAAS;GAAM;EACzC,CAAC;AAEF,KAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,KAAI,IACF,KAAK,UAAU;EACb,IAAI,WAAW,OAAO;EACtB,OAAO;EACP,SAAS,CAAC,cAAc;EACzB,CAAC,CACH"}

package/dist/ndjson-writer.cjs

@@ -0,0 +1,31 @@
+const require_sse_writer = require('./sse-writer.cjs');
+
+//#region src/ndjson-writer.ts
+async function writeNDJSONStream(res, chunks, options) {
+	const opts = options ?? {};
+	const latency = opts.latency ?? 0;
+	const profile = opts.streamingProfile;
+	const signal = opts.signal;
+	const onChunkSent = opts.onChunkSent;
+	if (res.writableEnded) return true;
+	res.setHeader("Content-Type", "application/x-ndjson");
+	res.setHeader("Cache-Control", "no-cache");
+	res.setHeader("Connection", "keep-alive");
+	let chunkIndex = 0;
+	for (const chunk of chunks) {
+		const chunkDelay = require_sse_writer.calculateDelay(chunkIndex, profile, latency);
+		if (chunkDelay > 0) await require_sse_writer.delay(chunkDelay, signal);
+		if (signal?.aborted) return false;
+		if (res.writableEnded) return true;
+		res.write(JSON.stringify(chunk) + "\n");
+		onChunkSent?.();
+		if (signal?.aborted) return false;
+		chunkIndex++;
+	}
+	if (!res.writableEnded) res.end();
+	return true;
+}
+
+//#endregion
+exports.writeNDJSONStream = writeNDJSONStream;
+//# sourceMappingURL=ndjson-writer.cjs.map

package/dist/ndjson-writer.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"ndjson-writer.cjs","names":["calculateDelay","delay"],"sources":["../src/ndjson-writer.ts"],"sourcesContent":["/**\n * NDJSON streaming writer for Ollama endpoints.\n *\n * Mirrors writeSSEStream from sse-writer.ts but writes newline-delimited JSON\n * (one JSON object per line) instead of SSE events.\n */\n\nimport type * as http from \"node:http\";\nimport type { StreamingProfile } from \"./types.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\n\nexport interface NDJSONStreamOptions {\n  latency?: number;\n  streamingProfile?: StreamingProfile;\n  signal?: AbortSignal;\n  onChunkSent?: () => void;\n}\n\nexport async function writeNDJSONStream(\n  res: http.ServerResponse,\n  chunks: object[],\n  options?: NDJSONStreamOptions,\n): Promise<boolean> {\n  const opts = options ?? {};\n  const latency = opts.latency ?? 0;\n  const profile = opts.streamingProfile;\n  const signal = opts.signal;\n  const onChunkSent = opts.onChunkSent;\n\n  if (res.writableEnded) return true;\n  res.setHeader(\"Content-Type\", \"application/x-ndjson\");\n  res.setHeader(\"Cache-Control\", \"no-cache\");\n  res.setHeader(\"Connection\", \"keep-alive\");\n\n  let chunkIndex = 0;\n  for (const chunk of chunks) {\n    const chunkDelay = calculateDelay(chunkIndex, profile, latency);\n    if (chunkDelay > 0) {\n      await delay(chunkDelay, signal);\n    }\n    if (signal?.aborted) return false;\n    if (res.writableEnded) return true;\n    res.write(JSON.stringify(chunk) + \"\\n\");\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n    chunkIndex++;\n  }\n\n  if (!res.writableEnded) {\n    res.end();\n  }\n  return true;\n}\n"],"mappings":";;;AAkBA,eAAsB,kBACpB,KACA,QACA,SACkB;CAClB,MAAM,OAAO,WAAW,EAAE;CAC1B,MAAM,UAAU,KAAK,WAAW;CAChC,MAAM,UAAU,KAAK;CACrB,MAAM,SAAS,KAAK;CACpB,MAAM,cAAc,KAAK;AAEzB,KAAI,IAAI,cAAe,QAAO;AAC9B,KAAI,UAAU,gBAAgB,uBAAuB;AACrD,KAAI,UAAU,iBAAiB,WAAW;AAC1C,KAAI,UAAU,cAAc,aAAa;CAEzC,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,aAAaA,kCAAe,YAAY,SAAS,QAAQ;AAC/D,MAAI,aAAa,EACf,OAAMC,yBAAM,YAAY,OAAO;AAEjC,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,IAAI,cAAe,QAAO;AAC9B,MAAI,MAAM,KAAK,UAAU,MAAM,GAAG,KAAK;AACvC,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;AAC5B;;AAGF,KAAI,CAAC,IAAI,cACP,KAAI,KAAK;AAEX,QAAO"}

package/dist/ndjson-writer.d.cts

@@ -0,0 +1,17 @@
+import { StreamingProfile } from "./types.cjs";
+import * as http from "node:http";
+
+//#region src/ndjson-writer.d.ts
+
+interface NDJSONStreamOptions {
+  latency?: number;
+  streamingProfile?: StreamingProfile;
+  signal?: AbortSignal;
+  onChunkSent?: () => void;
+}
+declare function writeNDJSONStream(res: http.ServerResponse, chunks: object[], options?: NDJSONStreamOptions): Promise<boolean>;
+//# sourceMappingURL=ndjson-writer.d.ts.map
+
+//#endregion
+export { NDJSONStreamOptions, writeNDJSONStream };
+//# sourceMappingURL=ndjson-writer.d.cts.map

package/dist/ndjson-writer.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ndjson-writer.d.cts","names":[],"sources":["../src/ndjson-writer.ts"],"sourcesContent":[],"mappings":";;;;;AAcsB,UAHL,mBAAA,CAGK;EAIA,OAAA,CAAA,EAAA,MAAA;EAAiB,gBAAA,CAAA,EALlB,gBAKkB;QAChC,CAAA,EALI,WAKC;aAEA,CAAA,EAAA,GAAA,GAAA,IAAA;;AACF,iBAJY,iBAAA,CAIZ,GAAA,EAHH,IAAA,CAAK,cAGF,EAAA,MAAA,EAAA,MAAA,EAAA,EAAA,OAAA,CAAA,EADE,mBACF,CAAA,EAAP,OAAO,CAAA,OAAA,CAAA"}

package/dist/ndjson-writer.d.ts

@@ -0,0 +1,17 @@
+import { StreamingProfile } from "./types.js";
+import * as http from "node:http";
+
+//#region src/ndjson-writer.d.ts
+
+interface NDJSONStreamOptions {
+  latency?: number;
+  streamingProfile?: StreamingProfile;
+  signal?: AbortSignal;
+  onChunkSent?: () => void;
+}
+declare function writeNDJSONStream(res: http.ServerResponse, chunks: object[], options?: NDJSONStreamOptions): Promise<boolean>;
+//# sourceMappingURL=ndjson-writer.d.ts.map
+
+//#endregion
+export { NDJSONStreamOptions, writeNDJSONStream };
+//# sourceMappingURL=ndjson-writer.d.ts.map

package/dist/ndjson-writer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ndjson-writer.d.ts","names":[],"sources":["../src/ndjson-writer.ts"],"sourcesContent":[],"mappings":";;;;;AAcsB,UAHL,mBAAA,CAGK;EAIA,OAAA,CAAA,EAAA,MAAA;EAAiB,gBAAA,CAAA,EALlB,gBAKkB;QAChC,CAAA,EALI,WAKC;aAEA,CAAA,EAAA,GAAA,GAAA,IAAA;;AACF,iBAJY,iBAAA,CAIZ,GAAA,EAHH,IAAA,CAAK,cAGF,EAAA,MAAA,EAAA,MAAA,EAAA,EAAA,OAAA,CAAA,EADE,mBACF,CAAA,EAAP,OAAO,CAAA,OAAA,CAAA"}

package/dist/ndjson-writer.js

@@ -0,0 +1,31 @@
+import { calculateDelay, delay } from "./sse-writer.js";
+
+//#region src/ndjson-writer.ts
+async function writeNDJSONStream(res, chunks, options) {
+	const opts = options ?? {};
+	const latency = opts.latency ?? 0;
+	const profile = opts.streamingProfile;
+	const signal = opts.signal;
+	const onChunkSent = opts.onChunkSent;
+	if (res.writableEnded) return true;
+	res.setHeader("Content-Type", "application/x-ndjson");
+	res.setHeader("Cache-Control", "no-cache");
+	res.setHeader("Connection", "keep-alive");
+	let chunkIndex = 0;
+	for (const chunk of chunks) {
+		const chunkDelay = calculateDelay(chunkIndex, profile, latency);
+		if (chunkDelay > 0) await delay(chunkDelay, signal);
+		if (signal?.aborted) return false;
+		if (res.writableEnded) return true;
+		res.write(JSON.stringify(chunk) + "\n");
+		onChunkSent?.();
+		if (signal?.aborted) return false;
+		chunkIndex++;
+	}
+	if (!res.writableEnded) res.end();
+	return true;
+}
+
+//#endregion
+export { writeNDJSONStream };
+//# sourceMappingURL=ndjson-writer.js.map

package/dist/ndjson-writer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ndjson-writer.js","names":[],"sources":["../src/ndjson-writer.ts"],"sourcesContent":["/**\n * NDJSON streaming writer for Ollama endpoints.\n *\n * Mirrors writeSSEStream from sse-writer.ts but writes newline-delimited JSON\n * (one JSON object per line) instead of SSE events.\n */\n\nimport type * as http from \"node:http\";\nimport type { StreamingProfile } from \"./types.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\n\nexport interface NDJSONStreamOptions {\n  latency?: number;\n  streamingProfile?: StreamingProfile;\n  signal?: AbortSignal;\n  onChunkSent?: () => void;\n}\n\nexport async function writeNDJSONStream(\n  res: http.ServerResponse,\n  chunks: object[],\n  options?: NDJSONStreamOptions,\n): Promise<boolean> {\n  const opts = options ?? {};\n  const latency = opts.latency ?? 0;\n  const profile = opts.streamingProfile;\n  const signal = opts.signal;\n  const onChunkSent = opts.onChunkSent;\n\n  if (res.writableEnded) return true;\n  res.setHeader(\"Content-Type\", \"application/x-ndjson\");\n  res.setHeader(\"Cache-Control\", \"no-cache\");\n  res.setHeader(\"Connection\", \"keep-alive\");\n\n  let chunkIndex = 0;\n  for (const chunk of chunks) {\n    const chunkDelay = calculateDelay(chunkIndex, profile, latency);\n    if (chunkDelay > 0) {\n      await delay(chunkDelay, signal);\n    }\n    if (signal?.aborted) return false;\n    if (res.writableEnded) return true;\n    res.write(JSON.stringify(chunk) + \"\\n\");\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n    chunkIndex++;\n  }\n\n  if (!res.writableEnded) {\n    res.end();\n  }\n  return true;\n}\n"],"mappings":";;;AAkBA,eAAsB,kBACpB,KACA,QACA,SACkB;CAClB,MAAM,OAAO,WAAW,EAAE;CAC1B,MAAM,UAAU,KAAK,WAAW;CAChC,MAAM,UAAU,KAAK;CACrB,MAAM,SAAS,KAAK;CACpB,MAAM,cAAc,KAAK;AAEzB,KAAI,IAAI,cAAe,QAAO;AAC9B,KAAI,UAAU,gBAAgB,uBAAuB;AACrD,KAAI,UAAU,iBAAiB,WAAW;AAC1C,KAAI,UAAU,cAAc,aAAa;CAEzC,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,aAAa,eAAe,YAAY,SAAS,QAAQ;AAC/D,MAAI,aAAa,EACf,OAAM,MAAM,YAAY,OAAO;AAEjC,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,IAAI,cAAe,QAAO;AAC9B,MAAI,MAAM,KAAK,UAAU,MAAM,GAAG,KAAK;AACvC,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;AAC5B;;AAGF,KAAI,CAAC,IAAI,cACP,KAAI,KAAK;AAEX,QAAO"}