@copilotkit/aimock 1.45.1 → 1.46.0

package/dist/config-loader.d.ts

@@ -1,4 +1,4 @@
-import { ApiKeyAuthConfig, ChaosConfig, McpRecordConfig, MisbehaviorConfig, MisbehaviorFaultId, RecordConfig, ResponsesToolsMode } from "./types.js";
+import { AimockPolicy, ApiKeyAuthConfig, ChaosConfig, McpRecordConfig, MisbehaviorConfig, MisbehaviorFaultId, RecordConfig, ResponsesToolsMode } from "./types.js";
 import { A2AAgentDefinition, A2AArtifact, A2APart, A2AStreamEvent } from "./a2a-types.js";
 import { AGUIEvent } from "./agui-types.js";
 import { LLMock } from "./llmock.js";
@@ -115,6 +115,8 @@ interface AimockConfig {
     strictToolArguments?: boolean;
     /** See `MockServerOptions.responsesTools`. */
     responsesTools?: ResponsesToolsMode;
+    /** See `MockServerOptions.policy`. Independent of `strict`. */
+    policy?: AimockPolicy;
   };
   mcp?: MCPConfig;
   a2a?: A2AConfig;
@@ -144,7 +146,11 @@ interface StartFromConfigOverrides {
   strictToolArguments?: boolean;
   /** `aimock --responses-tools`: overrides `llm.responsesTools`. */
   responsesTools?: ResponsesToolsMode;
+  /** `aimock --policy`: overrides individual keys from `llm.policy`. */
+  policy?: AimockPolicy;
 }
+/** Both CLI entry points use the shared policy parser and treat failures as usage errors. */
+
 declare function startFromConfig(config: AimockConfig, overrides?: StartFromConfigOverrides): Promise<{
   llmock: LLMock;
   url: string;

package/dist/config-loader.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"config-loader.d.ts","names":[],"sources":["../src/config-loader.ts"],"sourcesContent":[],"mappings":";;;;;;;;UA4BiB,aAAA,SAAsB;;AAAvC;AAIiB,UAAA,iBAAA,CAAiB;EASjB,GAAA,EAAA,MAAA;EAAgB,IAAA,EAAA,MAAA;UAEnB,CAAA,EAAA,MAAA;aAF2B,CAAA,EAAA,MAAA;EAAmB,IAAA,CAAA,EAAA,MAAA;EAM3C,IAAA,CAAA,EAAA,MAAS;;AAGhB,UATO,eAAA,SAAwB,mBAS/B,CAAA;QACI,CAAA,EAAA;IACF,QAAA,EATE,KASF,CAAA;MAAe,IAAA,EAAA,MAAA;MAGV,OAAA,EAAA;QAAgB,IAAA,EAAA,MAAA;QAEvB,IAAA,EAAA,MAAA;MACI,CAAA;IACH,CAAA,CAAA;EAAc,CAAA;AAIzB;AAAgC,UAhBf,SAAA,CAgBe;MACnB,CAAA,EAAA,MAAA;YACH,CAAA,EAAA;IACS,IAAA,EAAA,MAAA;IAHqB,OAAA,EAAA,MAAA;EAAkB,CAAA;EAMzC,KAAA,CAAA,EAnBP,aAmBgB,EAAA;EAKT,SAAA,CAAA,EAvBH,iBAuBoB,EAAA;EAOjB,OAAA,CAAA,EA7BL,eA+BC,EAAA;AAGb;AAAuC,UA/BtB,gBAAA,CA+BsB;SAMxB,EAAA,MAAA;OAHH,CAAA,EAhCF,OAgCE,EAAA;WAKK,CAAA,EApCH,WAoCG,EAAA;EAAW,MAAA,CAAA,EAnCjB,cAmCiB,EAAA;EAGX,OAAA,CAAA,EAAA,MAAY;AAM7B;AAAoC,UAxCnB,cAAA,SAAuB,kBAwCJ,CAAA;UAEJ,CAAA,EAzCnB,gBAyCmB,EAAA;OAAxB,CAAA,EAxCE,gBAwCF,EAAA;gBAFoC,CAAA,EArCzB,gBAqCyB,EAAA;;AAK3B,UAvCA,SAAA,CAuCY;EAAA,IAAA,CAAA,EAAA,MAAA;QACpB,CAAA,EAtCE,cAsCF,EAAA;;AAaS,UAhDD,iBAAA,CAgDC;OAAoB,EAAA;IAGzB,OAAA,CAAA,EAAA,MAAA;IASQ,UAAA,CAAA,EAAA,MAAA;IAEb,QAAA,CAAA,EAAA,MAAA;IACA,QAAA,CAAA,EAAA,MAAA;;MAEG,CAAA,EAAA,MAAA;EAAY,MAAA,CAAA,EA9DZ,SA8DY,EAAA;EAgBP,OAAA,CAAA,EAAA,MAAU;AAgB1B;AAWsB,UArGL,UAAA,CAqGoB;EAAA,IAAA,CAAA,EAAA,MAAA;UAC3B,CAAA,EApGG,iBAoGH,EAAA;;AAEW,UAnGJ,sBAAA,CAmGI;MAAlB,EAAA,MAAA;EAAO,SAAA,EAAA,MAAA;YAhGE;;;eAGG;;iBAEE;;UAGA,YAAA;;gBAED;;;UAIC,kBAAA,SAA2B;;QAEpC,wBAAwB;;UAGf,YAAA;SACR;;;;;;;YAOG;;;;;;kBAMM,oBAAoB;;;aAGzB;;;;;;;;;qBASQ;;QAEb;QACA;SACC;WACE;;;;;;;;;;;iBAgBK,UAAA,sBAAgC;;;;;UAgB/B,wBAAA;;;;;;;;mBAQE;;iBAGG,eAAA,SACZ,0BACI,2BACX;UAAkB"}
+{"version":3,"file":"config-loader.d.ts","names":[],"sources":["../src/config-loader.ts"],"sourcesContent":[],"mappings":";;;;;;;;UA8BiB,aAAA,SAAsB;;AAAvC;AAIiB,UAAA,iBAAA,CAAiB;EASjB,GAAA,EAAA,MAAA;EAAgB,IAAA,EAAA,MAAA;UAEnB,CAAA,EAAA,MAAA;aAF2B,CAAA,EAAA,MAAA;EAAmB,IAAA,CAAA,EAAA,MAAA;EAM3C,IAAA,CAAA,EAAA,MAAS;;AAGhB,UATO,eAAA,SAAwB,mBAS/B,CAAA;QACI,CAAA,EAAA;IACF,QAAA,EATE,KASF,CAAA;MAAe,IAAA,EAAA,MAAA;MAGV,OAAA,EAAA;QAAgB,IAAA,EAAA,MAAA;QAEvB,IAAA,EAAA,MAAA;MACI,CAAA;IACH,CAAA,CAAA;EAAc,CAAA;AAIzB;AAAgC,UAhBf,SAAA,CAgBe;MACnB,CAAA,EAAA,MAAA;YACH,CAAA,EAAA;IACS,IAAA,EAAA,MAAA;IAHqB,OAAA,EAAA,MAAA;EAAkB,CAAA;EAMzC,KAAA,CAAA,EAnBP,aAmBgB,EAEf;EAGM,SAAA,CAAA,EAvBH,iBAuBoB,EAGvB;EAIM,OAAA,CAAA,EA7BL,eA+BC,EAAA;AAGb;AAAuC,UA/BtB,gBAAA,CA+BsB;SAMxB,EAAA,MAAA;OAHH,CAAA,EAhCF,OAgCE,EAAA;WAKK,CAAA,EApCH,WAoCG,EAAA;EAAW,MAAA,CAAA,EAnCjB,cAmCiB,EAAA;EAGX,OAAA,CAAA,EAAA,MAAY;AAM7B;AAAoC,UAxCnB,cAAA,SAAuB,kBAwCJ,CAAA;UAEJ,CAAA,EAzCnB,gBAyCmB,EAAA;OAAxB,CAAA,EAxCE,gBAwCF,EAAA;gBAFoC,CAAA,EArCzB,gBAqCyB,EAAA;;AAK3B,UAvCA,SAAA,CAuCY;EAAA,IAAA,CAAA,EAAA,MAAA;QACpB,CAAA,EAtCE,cAsCF,EAAA;;AAaS,UAhDD,iBAAA,CAgDC;OAAoB,EAAA;IAGzB,OAAA,CAAA,EAAA,MAAA;IASQ,UAAA,CAAA,EAAA,MAAA;IAER,QAAA,CAAA,EAAA,MAAA;IAEL,QAAA,CAAA,EAAA,MAAA;;MAEC,CAAA,EAAA,MAAA;QACE,CAAA,EAhEA,SAgEA,EAAA;EAAY,OAAA,CAAA,EAAA,MAAA;AAgBvB;AAgBiB,UA5FA,UAAA,CA4FA;EAAwB,IAAA,CAAA,EAAA,MAAA;UAQtB,CAAA,EAlGN,iBAkGM,EAAA;;AAEI,UAjGN,sBAAA,CAiGM;EA0BD,IAAA,EAAA,MAAA;EAAe,SAAA,EAAA,MAAA;SAC3B,CAAA,EAzHE,KAyHF,CAAA;IACI,EAAA,EAAA,MAAA;IACO,MAAA,EAAA,MAAA,EAAA;IAAlB,QAAA,CAAA,EAxHY,MAwHZ,CAAA,MAAA,EAAA,OAAA,CAAA;EAAO,CAAA,CAAA;iBAtHO;;UAGA,YAAA;;gBAED;;;UAIC,kBAAA,SAA2B;;QAEpC,wBAAwB;;UAGf,YAAA;SACR;;;;;;;YAOG;;;;;;kBAMM,oBAAoB;;;aAGzB;;;;;;;;;qBASQ;;aAER;;QAEL;QACA;SACC;WACE;;;;;;;;;;;iBAgBK,UAAA,sBAAgC;;;;;UAgB/B,wBAAA;;;;;;;;mBAQE;;WAER;;;;iBA0BW,eAAA,SACZ,0BACI,2BACX;UAAkB"}

package/dist/config-loader.js

@@ -1,4 +1,5 @@
 import { resolveInboundAuth, selectInboundAuthSource } from "./api-key-auth.js";
+import { assignPolicyEntry, parsePolicy, parsePolicyEntry } from "./response/policy.js";
 import { A2AMock } from "./a2a-mock.js";
 import { buildTextResponse } from "./agui-handler.js";
 import { parseMisbehavior } from "./misbehavior.js";
@@ -27,8 +28,33 @@ function loadConfig(configPath) {
 function hasMcpRecordMount(mcp) {
 	return typeof mcp === "object" && mcp !== null && !Array.isArray(mcp) && Object.keys(mcp).length > 0;
 }
+/** Both CLI entry points use the shared policy parser and treat failures as usage errors. */
+function parsePolicyFlags(entries) {
+	const policy = {};
+	for (const entry of entries) {
+		const separator = entry.indexOf("=");
+		if (separator < 0) return {
+			ok: false,
+			error: `Invalid --policy ${JSON.stringify(entry)}: expected key=value.`
+		};
+		const parsed = parsePolicyEntry(entry.slice(0, separator), entry.slice(separator + 1));
+		if (!parsed.ok) return {
+			ok: false,
+			error: `Invalid --policy ${JSON.stringify(entry)}: ${parsed.reason} Use key=value.`
+		};
+		assignPolicyEntry(policy, parsed);
+	}
+	return {
+		ok: true,
+		policy
+	};
+}
 async function startFromConfig(config, overrides) {
 	const logger = new Logger("info");
+	const policy = {
+		...parsePolicy(config.llm?.policy, (message) => logger.warn(message)),
+		...parsePolicy(overrides?.policy, (message) => logger.warn(message))
+	};
 	const booleanOptIn = (key) => {
 		const value = config.llm?.[key];
 		if (value === void 0 || typeof value === "boolean") return value;
@@ -73,7 +99,8 @@ async function startFromConfig(config, overrides) {
 		metrics: config.metrics,
 		strict: config.strict,
 		strictToolArguments: overrides?.strictToolArguments ?? configStrictToolArguments,
-		responsesTools: overrides?.responsesTools ?? responsesTools
+		responsesTools: overrides?.responsesTools ?? responsesTools,
+		policy
 	}, resolvedAuth);
 	const configMounts = /* @__PURE__ */ new Map();
 	if (config.llm?.fixtures) {
@@ -205,5 +232,5 @@ async function startFromConfig(config, overrides) {
 }
 
 //#endregion
-export { loadConfig, startFromConfig };
+export { loadConfig, parsePolicyFlags, startFromConfig };
 //# sourceMappingURL=config-loader.js.map

package/dist/config-loader.js.map

@@ -1 +1 @@
-{"version":3,"file":"config-loader.js","names":[],"sources":["../src/config-loader.ts"],"sourcesContent":["import * as fs from \"node:fs\";\nimport * as path from \"node:path\";\nimport { LLMock, createLLMockWithResolvedAuth } from \"./llmock.js\";\nimport { resolveInboundAuth, selectInboundAuthSource } from \"./api-key-auth.js\";\nimport { MCPMock } from \"./mcp-mock.js\";\nimport { configToolHandlers } from \"./mcp-handler.js\";\nimport { A2AMock } from \"./a2a-mock.js\";\nimport { AGUIMock } from \"./agui-mock.js\";\nimport type {\n  ApiKeyAuthConfig,\n  ChaosConfig,\n  McpRecordConfig,\n  MisbehaviorConfig,\n  MisbehaviorFaultId,\n  Mountable,\n  RecordConfig,\n  ResponsesToolsMode,\n} from \"./types.js\";\nimport { parseMisbehavior } from \"./misbehavior.js\";\nimport type { MCPToolDefinition, MCPPromptDefinition } from \"./mcp-types.js\";\nimport type { A2AAgentDefinition, A2APart, A2AArtifact, A2AStreamEvent } from \"./a2a-types.js\";\nimport type { AGUIEvent } from \"./agui-types.js\";\nimport { buildTextResponse } from \"./agui-handler.js\";\nimport { VectorMock } from \"./vector-mock.js\";\nimport type { QueryResult } from \"./vector-types.js\";\nimport { Logger } from \"./logger.js\";\nimport { wireMcpRecording } from \"./mcp-recorder.js\";\n\nexport interface MCPConfigTool extends MCPToolDefinition {\n  result?: string;\n}\n\nexport interface MCPConfigResource {\n  uri: string;\n  name: string;\n  mimeType?: string;\n  description?: string;\n  text?: string;\n  blob?: string;\n}\n\nexport interface MCPConfigPrompt extends MCPPromptDefinition {\n  result?: {\n    messages: Array<{ role: string; content: { type: string; text: string } }>;\n  };\n}\n\nexport interface MCPConfig {\n  path?: string;\n  serverInfo?: { name: string; version: string };\n  tools?: MCPConfigTool[];\n  resources?: MCPConfigResource[];\n  prompts?: MCPConfigPrompt[];\n}\n\nexport interface A2AConfigPattern {\n  pattern: string;\n  parts?: A2APart[];\n  artifacts?: A2AArtifact[];\n  events?: A2AStreamEvent[];\n  delayMs?: number;\n}\n\nexport interface A2AConfigAgent extends A2AAgentDefinition {\n  messages?: A2AConfigPattern[];\n  tasks?: A2AConfigPattern[];\n  streamingTasks?: A2AConfigPattern[];\n}\n\nexport interface A2AConfig {\n  path?: string;\n  agents?: A2AConfigAgent[];\n}\n\nexport interface AGUIConfigFixture {\n  match: { message?: string; toolCallId?: string; toolName?: string; stateKey?: string };\n  text?: string; // shorthand: uses buildTextResponse\n  events?: AGUIEvent[]; // raw events\n  delayMs?: number;\n}\n\nexport interface AGUIConfig {\n  path?: string; // mount path, default \"/agui\"\n  fixtures?: AGUIConfigFixture[];\n}\n\nexport interface VectorConfigCollection {\n  name: string;\n  dimension: number;\n  vectors?: Array<{\n    id: string;\n    values: number[];\n    metadata?: Record<string, unknown>;\n  }>;\n  queryResults?: QueryResult[];\n}\n\nexport interface VectorConfig {\n  path?: string;\n  collections?: VectorConfigCollection[];\n}\n\n/** `llm.record` in a config file: the LLM `RecordConfig`, plus MCP record mounts (C10). */\nexport interface AimockRecordConfig extends RecordConfig {\n  /** MR1: mount path → upstream URL, or a full MCP record config. */\n  mcp?: Record<string, string | McpRecordConfig>;\n}\n\nexport interface AimockConfig {\n  auth?: ApiKeyAuthConfig;\n  llm?: {\n    fixtures?: string;\n    latency?: number;\n    chunkSize?: number;\n    replaySpeed?: number;\n    logLevel?: \"silent\" | \"warn\" | \"info\" | \"debug\";\n    chaos?: ChaosConfig;\n    /**\n     * Server-wide misbehavior defaults. Read only when misbehavior is enabled\n     * (`llm.enableMisbehavior: true` or `aimock --misbehavior`); otherwise the\n     * key is ignored with a warning, as 1.44.0 ignored it.\n     */\n    misbehavior?: MisbehaviorConfig | MisbehaviorFaultId;\n    /** See `MockServerOptions.enableMisbehavior`. */\n    enableMisbehavior?: boolean;\n    record?: AimockRecordConfig;\n    /**\n     * Turns on `llm.record.mcp`. Without it the key is ignored with a warning,\n     * as 1.44.0 ignored it.\n     */\n    enableMcpRecording?: boolean;\n    /** See `MockServerOptions.strictToolArguments`. */\n    strictToolArguments?: boolean;\n    /** See `MockServerOptions.responsesTools`. */\n    responsesTools?: ResponsesToolsMode;\n  };\n  mcp?: MCPConfig;\n  a2a?: A2AConfig;\n  agui?: AGUIConfig;\n  vector?: VectorConfig;\n  services?: { search?: boolean; rerank?: boolean; moderate?: boolean };\n  metrics?: boolean;\n  strict?: boolean;\n  port?: number;\n  host?: string;\n}\n\n/** C10: the LLM part of `llm.record`. `mcp` is MCP-only; no `providers` means no LLM recording. */\nfunction llmRecordOf(record: AimockRecordConfig | undefined): RecordConfig | undefined {\n  if (!record) return undefined;\n  const { mcp: _mcp, ...llm } = record;\n  void _mcp; // repo pattern for an intentionally unused binding (byteplus-video.ts:1379-1380)\n  return llm.providers === undefined ? undefined : llm;\n}\n\nexport function loadConfig(configPath: string): AimockConfig {\n  const raw = fs.readFileSync(configPath, \"utf-8\");\n  return JSON.parse(raw) as AimockConfig;\n}\n\n/** C3: `llm.record.mcp` is an object with at least one mount. */\nfunction hasMcpRecordMount(mcp: unknown): boolean {\n  return (\n    typeof mcp === \"object\" && mcp !== null && !Array.isArray(mcp) && Object.keys(mcp).length > 0\n  );\n}\n\n/**\n * Command-line overrides for `startFromConfig`. The opt-in flags win over the\n * matching `llm.*` config keys.\n */\nexport interface StartFromConfigOverrides {\n  port?: number;\n  host?: string;\n  /** `aimock --misbehavior`: overrides `llm.enableMisbehavior`. */\n  enableMisbehavior?: boolean;\n  /** `aimock --strict-tool-arguments`: overrides `llm.strictToolArguments`. */\n  strictToolArguments?: boolean;\n  /** `aimock --responses-tools`: overrides `llm.responsesTools`. */\n  responsesTools?: ResponsesToolsMode;\n}\n\nexport async function startFromConfig(\n  config: AimockConfig,\n  overrides?: StartFromConfigOverrides,\n): Promise<{ llmock: LLMock; url: string }> {\n  const logger = new Logger(\"info\");\n  // 1.44.0 ignored these keys, so a non-boolean value (such as the string\n  // \"true\") is ignored with a warning, as `llm.misbehavior` is, and the\n  // option stays off unless a CLI flag turns it on (`llm.enableMcpRecording`\n  // has no CLI flag).\n  const booleanOptIn = (\n    key: \"enableMisbehavior\" | \"enableMcpRecording\" | \"strictToolArguments\",\n  ): boolean | undefined => {\n    const value: unknown = config.llm?.[key];\n    if (value === undefined || typeof value === \"boolean\") return value;\n    logger.warn(\n      `Ignoring llm.${key} because it must be true or false, got ${JSON.stringify(value)}.`,\n    );\n    return undefined;\n  };\n  const configEnableMisbehavior = booleanOptIn(\"enableMisbehavior\");\n  const configStrictToolArguments = booleanOptIn(\"strictToolArguments\");\n  // MCP recording also needs an explicit opt-in. 1.44.0 ignored\n  // `llm.record.mcp`, so without `llm.enableMcpRecording: true` the key is\n  // ignored (with a warning): nothing is mounted or validated and the log level\n  // is unchanged. With the opt-in, an invalid value fails startup.\n  const enableMcpRecording = booleanOptIn(\"enableMcpRecording\") === true;\n  const mcpRecord = enableMcpRecording ? config.llm?.record?.mcp : undefined;\n  // Misbehavior needs an explicit opt-in that a 1.44.0 config cannot contain.\n  // 1.44.0 ignored `llm.misbehavior`, so without the opt-in the key is ignored\n  // (with a warning) and the server serves exactly what 1.44.0 served. With the\n  // opt-in, an invalid value fails startup like any other enabled-mode error.\n  const enableMisbehavior = (overrides?.enableMisbehavior ?? configEnableMisbehavior) === true;\n  let misbehavior: MisbehaviorConfig | undefined;\n  if (config.llm?.misbehavior !== undefined) {\n    if (!enableMisbehavior) {\n      logger.warn(\n        \"Ignoring llm.misbehavior because misbehavior is not enabled. Set llm.enableMisbehavior: true or pass --misbehavior to use it.\",\n      );\n    } else {\n      const parsed = parseMisbehavior(config.llm.misbehavior, \"llm.misbehavior\");\n      if (!parsed.ok) {\n        throw new TypeError(`${parsed.issue.rule}: ${parsed.issue.path}: ${parsed.issue.message}`);\n      }\n      misbehavior = parsed.config;\n    }\n  }\n  if (config.llm?.record?.mcp !== undefined && !enableMcpRecording) {\n    logger.warn(\n      \"Ignoring llm.record.mcp because MCP recording is not enabled. Set llm.enableMcpRecording: true to use it.\",\n    );\n  }\n\n  // 1.44.0 ignored `llm.responsesTools`, so an invalid value is ignored with a\n  // warning (as `llm.misbehavior` is) and the server runs in \"legacy\" mode,\n  // unless `--responses-tools` sets the mode. The warning names the mode that\n  // actually runs.\n  let responsesTools = config.llm?.responsesTools;\n  if (\n    responsesTools !== undefined &&\n    responsesTools !== \"legacy\" &&\n    responsesTools !== \"extended\"\n  ) {\n    const effective =\n      overrides?.responsesTools !== undefined\n        ? `\"${overrides.responsesTools}\" from --responses-tools`\n        : '\"legacy\"';\n    logger.warn(\n      `Ignoring llm.responsesTools because it must be \"legacy\" or \"extended\", got ${JSON.stringify(responsesTools)}. Using ${effective}.`,\n    );\n    responsesTools = undefined;\n  }\n\n  // A non-positive replaySpeed fails calculateDelay's `speed > 0` check and applies the\n  // full delay rather than none. Mirrors the fixture-level guard in fixture-loader.ts.\n  let replaySpeed = config.llm?.replaySpeed;\n  if (replaySpeed != null && (!Number.isFinite(replaySpeed) || replaySpeed <= 0)) {\n    logger.warn(`llm.replaySpeed must be a positive number, got ${replaySpeed}. Ignoring.`);\n    replaySpeed = undefined;\n  }\n\n  // Load fixtures if specified\n  const resolvedAuth = resolveInboundAuth(selectInboundAuthSource(config.auth));\n  const llmock = createLLMockWithResolvedAuth(\n    {\n      port: overrides?.port ?? config.port ?? 0,\n      host: overrides?.host ?? config.host ?? \"127.0.0.1\",\n      latency: config.llm?.latency,\n      chunkSize: config.llm?.chunkSize,\n      replaySpeed,\n      // llm.record.mcp reports skipped or failed recordings only as MCP-RECORD: warnings,\n      // so an enabled config with at least one mount defaults to \"warn\" rather than the silent\n      // default. An empty llm.record.mcp records nothing. An explicit logLevel wins.\n      logLevel: config.llm?.logLevel ?? (hasMcpRecordMount(mcpRecord) ? \"warn\" : undefined),\n      chaos: config.llm?.chaos,\n      misbehavior,\n      ...(enableMisbehavior ? { enableMisbehavior: true } : {}),\n      record: llmRecordOf(config.llm?.record),\n      metrics: config.metrics,\n      strict: config.strict,\n      strictToolArguments: overrides?.strictToolArguments ?? configStrictToolArguments,\n      responsesTools: overrides?.responsesTools ?? responsesTools,\n    },\n    resolvedAuth,\n  );\n\n  // What this function mounted, by path (MR1: llm.record.mcp records through an MCP mount).\n  const configMounts = new Map<string, Mountable>();\n\n  if (config.llm?.fixtures) {\n    const fixturePath = path.resolve(config.llm.fixtures);\n    const stat = fs.statSync(fixturePath);\n    if (stat.isDirectory()) {\n      llmock.loadFixtureDir(fixturePath);\n    } else {\n      llmock.loadFixtureFile(fixturePath);\n    }\n  }\n\n  // MCP\n  if (config.mcp) {\n    const mcpConfig = config.mcp;\n    const mcp = new MCPMock({\n      serverInfo: mcpConfig.serverInfo,\n    });\n\n    if (mcpConfig.tools) {\n      for (const tool of mcpConfig.tools) {\n        const { result, ...def } = tool;\n        mcp.addTool(def);\n        if (result !== undefined) {\n          const answer = () => result;\n          configToolHandlers.add(answer);\n          mcp.onToolCall(def.name, answer);\n        }\n      }\n    }\n\n    if (mcpConfig.resources) {\n      for (const res of mcpConfig.resources) {\n        mcp.addResource(\n          { uri: res.uri, name: res.name, mimeType: res.mimeType, description: res.description },\n          res.text !== undefined || res.blob !== undefined\n            ? { text: res.text, blob: res.blob, mimeType: res.mimeType }\n            : undefined,\n        );\n      }\n    }\n\n    if (mcpConfig.prompts) {\n      for (const prompt of mcpConfig.prompts) {\n        const { result, ...def } = prompt;\n        if (result) {\n          mcp.addPrompt(def, () => result as import(\"./mcp-types.js\").MCPPromptResult);\n        } else {\n          mcp.addPrompt(def);\n        }\n      }\n    }\n\n    const mcpPath = mcpConfig.path ?? \"/mcp\";\n    llmock.mount(mcpPath, mcp);\n    configMounts.set(mcpPath, mcp);\n    logger.info(`MCPMock mounted at ${mcpPath}`);\n  }\n\n  // A2A\n  if (config.a2a) {\n    const a2aConfig = config.a2a;\n    const a2a = new A2AMock();\n\n    if (a2aConfig.agents) {\n      for (const agentConfig of a2aConfig.agents) {\n        const { messages, tasks, streamingTasks, ...def } = agentConfig;\n        a2a.registerAgent(def);\n\n        if (messages) {\n          for (const m of messages) {\n            a2a.onMessage(def.name, m.pattern, m.parts ?? [{ text: \"\" }]);\n          }\n        }\n\n        if (tasks) {\n          for (const t of tasks) {\n            a2a.onTask(def.name, t.pattern, t.artifacts ?? []);\n          }\n        }\n\n        if (streamingTasks) {\n          for (const s of streamingTasks) {\n            a2a.onStreamingTask(def.name, s.pattern, s.events ?? [], s.delayMs);\n          }\n        }\n      }\n    }\n\n    const a2aPath = a2aConfig.path ?? \"/a2a\";\n    llmock.mount(a2aPath, a2a);\n    configMounts.set(a2aPath, a2a);\n    logger.info(`A2AMock mounted at ${a2aPath}`);\n  }\n\n  // AG-UI\n  if (config.agui) {\n    const aguiConfig = config.agui;\n    const agui = new AGUIMock();\n\n    if (aguiConfig.fixtures) {\n      for (const f of aguiConfig.fixtures) {\n        if (f.match.toolCallId && f.text && !f.events) {\n          logger.warn(\n            `AG-UI fixture uses text shorthand with toolCallId — text shorthand ignores toolCallId matching; use events[] instead (match: ${JSON.stringify(f.match)})`,\n          );\n        }\n        if (f.text) {\n          if (f.match.message !== undefined) {\n            agui.onMessage(f.match.message, f.text, { delayMs: f.delayMs });\n          } else {\n            // No message pattern — register via addFixture so it only matches\n            // on other criteria (toolCallId, toolName, stateKey) instead of\n            // becoming a catch-all that matches every request.\n            agui.addFixture({\n              match: {\n                toolCallId: f.match.toolCallId,\n                toolName: f.match.toolName,\n                stateKey: f.match.stateKey,\n              },\n              events: buildTextResponse(f.text),\n              delayMs: f.delayMs,\n            });\n          }\n        } else if (f.events) {\n          agui.addFixture({\n            match: {\n              message: f.match.message,\n              toolCallId: f.match.toolCallId,\n              toolName: f.match.toolName,\n              stateKey: f.match.stateKey,\n            },\n            events: f.events,\n            delayMs: f.delayMs,\n          });\n        } else {\n          logger.warn(\n            `AG-UI fixture has neither text nor events — it will be skipped (match: ${JSON.stringify(f.match)})`,\n          );\n        }\n      }\n    }\n\n    const aguiPath = aguiConfig.path ?? \"/agui\";\n    llmock.mount(aguiPath, agui);\n    configMounts.set(aguiPath, agui);\n    logger.info(`AGUIMock mounted at ${aguiPath}`);\n  }\n\n  // Vector\n  if (config.vector) {\n    const vectorConfig = config.vector;\n    const vector = new VectorMock();\n\n    if (vectorConfig.collections) {\n      for (const col of vectorConfig.collections) {\n        vector.addCollection(col.name, { dimension: col.dimension });\n\n        if (col.vectors && col.vectors.length > 0) {\n          vector.upsert(col.name, col.vectors);\n        }\n\n        if (col.queryResults) {\n          vector.onQuery(col.name, col.queryResults);\n        }\n      }\n    }\n\n    const vectorPath = vectorConfig.path ?? \"/vector\";\n    llmock.mount(vectorPath, vector);\n    configMounts.set(vectorPath, vector);\n    logger.info(`VectorMock mounted at ${vectorPath}`);\n  }\n\n  // MR1: llm.record.mcp — record mounts (auto-mount an MCPMock where none is).\n  // Read only with llm.enableMcpRecording: true. Any value that is present is then\n  // validated, so null, false, 0 and \"\" fail at start too.\n  if (mcpRecord !== undefined) {\n    wireMcpRecording(llmock, mcpRecord, {\n      mounted: configMounts,\n      mcpMock: MCPMock,\n      fixtures: config.llm?.fixtures,\n      fixturePath: config.llm?.record?.fixturePath,\n      strict: config.strict,\n      env: process.env,\n    });\n  }\n\n  // Services — configure default catch-all responses\n  if (config.services) {\n    if (config.services.search) {\n      llmock.onSearch(/.*/, []);\n      logger.info(\"Search service enabled with default empty results\");\n    }\n    if (config.services.rerank) {\n      llmock.onRerank(/.*/, []);\n      logger.info(\"Rerank service enabled with default empty results\");\n    }\n    if (config.services.moderate) {\n      llmock.onModerate(/.*/, { flagged: false, categories: {} });\n      logger.info(\"Moderation service enabled with default unflagged results\");\n    }\n  }\n\n  const url = await llmock.start();\n  return { llmock, url };\n}\n"],"mappings":";;;;;;;;;;;;;;;;AAoJA,SAAS,YAAY,QAAkE;AACrF,KAAI,CAAC,OAAQ,QAAO;CACpB,MAAM,EAAE,KAAK,MAAM,GAAG,QAAQ;AAE9B,QAAO,IAAI,cAAc,SAAY,SAAY;;AAGnD,SAAgB,WAAW,YAAkC;CAC3D,MAAM,MAAM,GAAG,aAAa,YAAY,QAAQ;AAChD,QAAO,KAAK,MAAM,IAAI;;;AAIxB,SAAS,kBAAkB,KAAuB;AAChD,QACE,OAAO,QAAQ,YAAY,QAAQ,QAAQ,CAAC,MAAM,QAAQ,IAAI,IAAI,OAAO,KAAK,IAAI,CAAC,SAAS;;AAmBhG,eAAsB,gBACpB,QACA,WAC0C;CAC1C,MAAM,SAAS,IAAI,OAAO,OAAO;CAKjC,MAAM,gBACJ,QACwB;EACxB,MAAM,QAAiB,OAAO,MAAM;AACpC,MAAI,UAAU,UAAa,OAAO,UAAU,UAAW,QAAO;AAC9D,SAAO,KACL,gBAAgB,IAAI,yCAAyC,KAAK,UAAU,MAAM,CAAC,GACpF;;CAGH,MAAM,0BAA0B,aAAa,oBAAoB;CACjE,MAAM,4BAA4B,aAAa,sBAAsB;CAKrE,MAAM,qBAAqB,aAAa,qBAAqB,KAAK;CAClE,MAAM,YAAY,qBAAqB,OAAO,KAAK,QAAQ,MAAM;CAKjE,MAAM,qBAAqB,WAAW,qBAAqB,6BAA6B;CACxF,IAAI;AACJ,KAAI,OAAO,KAAK,gBAAgB,OAC9B,KAAI,CAAC,kBACH,QAAO,KACL,gIACD;MACI;EACL,MAAM,SAAS,iBAAiB,OAAO,IAAI,aAAa,kBAAkB;AAC1E,MAAI,CAAC,OAAO,GACV,OAAM,IAAI,UAAU,GAAG,OAAO,MAAM,KAAK,IAAI,OAAO,MAAM,KAAK,IAAI,OAAO,MAAM,UAAU;AAE5F,gBAAc,OAAO;;AAGzB,KAAI,OAAO,KAAK,QAAQ,QAAQ,UAAa,CAAC,mBAC5C,QAAO,KACL,4GACD;CAOH,IAAI,iBAAiB,OAAO,KAAK;AACjC,KACE,mBAAmB,UACnB,mBAAmB,YACnB,mBAAmB,YACnB;EACA,MAAM,YACJ,WAAW,mBAAmB,SAC1B,IAAI,UAAU,eAAe,4BAC7B;AACN,SAAO,KACL,8EAA8E,KAAK,UAAU,eAAe,CAAC,UAAU,UAAU,GAClI;AACD,mBAAiB;;CAKnB,IAAI,cAAc,OAAO,KAAK;AAC9B,KAAI,eAAe,SAAS,CAAC,OAAO,SAAS,YAAY,IAAI,eAAe,IAAI;AAC9E,SAAO,KAAK,kDAAkD,YAAY,aAAa;AACvF,gBAAc;;CAIhB,MAAM,eAAe,mBAAmB,wBAAwB,OAAO,KAAK,CAAC;CAC7E,MAAM,SAAS,6BACb;EACE,MAAM,WAAW,QAAQ,OAAO,QAAQ;EACxC,MAAM,WAAW,QAAQ,OAAO,QAAQ;EACxC,SAAS,OAAO,KAAK;EACrB,WAAW,OAAO,KAAK;EACvB;EAIA,UAAU,OAAO,KAAK,aAAa,kBAAkB,UAAU,GAAG,SAAS;EAC3E,OAAO,OAAO,KAAK;EACnB;EACA,GAAI,oBAAoB,EAAE,mBAAmB,MAAM,GAAG,EAAE;EACxD,QAAQ,YAAY,OAAO,KAAK,OAAO;EACvC,SAAS,OAAO;EAChB,QAAQ,OAAO;EACf,qBAAqB,WAAW,uBAAuB;EACvD,gBAAgB,WAAW,kBAAkB;EAC9C,EACD,aACD;CAGD,MAAM,+BAAe,IAAI,KAAwB;AAEjD,KAAI,OAAO,KAAK,UAAU;EACxB,MAAM,cAAc,KAAK,QAAQ,OAAO,IAAI,SAAS;AAErD,MADa,GAAG,SAAS,YAAY,CAC5B,aAAa,CACpB,QAAO,eAAe,YAAY;MAElC,QAAO,gBAAgB,YAAY;;AAKvC,KAAI,OAAO,KAAK;EACd,MAAM,YAAY,OAAO;EACzB,MAAM,MAAM,IAAI,QAAQ,EACtB,YAAY,UAAU,YACvB,CAAC;AAEF,MAAI,UAAU,MACZ,MAAK,MAAM,QAAQ,UAAU,OAAO;GAClC,MAAM,EAAE,QAAQ,GAAG,QAAQ;AAC3B,OAAI,QAAQ,IAAI;AAChB,OAAI,WAAW,QAAW;IACxB,MAAM,eAAe;AACrB,uBAAmB,IAAI,OAAO;AAC9B,QAAI,WAAW,IAAI,MAAM,OAAO;;;AAKtC,MAAI,UAAU,UACZ,MAAK,MAAM,OAAO,UAAU,UAC1B,KAAI,YACF;GAAE,KAAK,IAAI;GAAK,MAAM,IAAI;GAAM,UAAU,IAAI;GAAU,aAAa,IAAI;GAAa,EACtF,IAAI,SAAS,UAAa,IAAI,SAAS,SACnC;GAAE,MAAM,IAAI;GAAM,MAAM,IAAI;GAAM,UAAU,IAAI;GAAU,GAC1D,OACL;AAIL,MAAI,UAAU,QACZ,MAAK,MAAM,UAAU,UAAU,SAAS;GACtC,MAAM,EAAE,QAAQ,GAAG,QAAQ;AAC3B,OAAI,OACF,KAAI,UAAU,WAAW,OAAmD;OAE5E,KAAI,UAAU,IAAI;;EAKxB,MAAM,UAAU,UAAU,QAAQ;AAClC,SAAO,MAAM,SAAS,IAAI;AAC1B,eAAa,IAAI,SAAS,IAAI;AAC9B,SAAO,KAAK,sBAAsB,UAAU;;AAI9C,KAAI,OAAO,KAAK;EACd,MAAM,YAAY,OAAO;EACzB,MAAM,MAAM,IAAI,SAAS;AAEzB,MAAI,UAAU,OACZ,MAAK,MAAM,eAAe,UAAU,QAAQ;GAC1C,MAAM,EAAE,UAAU,OAAO,gBAAgB,GAAG,QAAQ;AACpD,OAAI,cAAc,IAAI;AAEtB,OAAI,SACF,MAAK,MAAM,KAAK,SACd,KAAI,UAAU,IAAI,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC;AAIjE,OAAI,MACF,MAAK,MAAM,KAAK,MACd,KAAI,OAAO,IAAI,MAAM,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;AAItD,OAAI,eACF,MAAK,MAAM,KAAK,eACd,KAAI,gBAAgB,IAAI,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ;;EAM3E,MAAM,UAAU,UAAU,QAAQ;AAClC,SAAO,MAAM,SAAS,IAAI;AAC1B,eAAa,IAAI,SAAS,IAAI;AAC9B,SAAO,KAAK,sBAAsB,UAAU;;AAI9C,KAAI,OAAO,MAAM;EACf,MAAM,aAAa,OAAO;EAC1B,MAAM,OAAO,IAAI,UAAU;AAE3B,MAAI,WAAW,SACb,MAAK,MAAM,KAAK,WAAW,UAAU;AACnC,OAAI,EAAE,MAAM,cAAc,EAAE,QAAQ,CAAC,EAAE,OACrC,QAAO,KACL,gIAAgI,KAAK,UAAU,EAAE,MAAM,CAAC,GACzJ;AAEH,OAAI,EAAE,KACJ,KAAI,EAAE,MAAM,YAAY,OACtB,MAAK,UAAU,EAAE,MAAM,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC;OAK/D,MAAK,WAAW;IACd,OAAO;KACL,YAAY,EAAE,MAAM;KACpB,UAAU,EAAE,MAAM;KAClB,UAAU,EAAE,MAAM;KACnB;IACD,QAAQ,kBAAkB,EAAE,KAAK;IACjC,SAAS,EAAE;IACZ,CAAC;YAEK,EAAE,OACX,MAAK,WAAW;IACd,OAAO;KACL,SAAS,EAAE,MAAM;KACjB,YAAY,EAAE,MAAM;KACpB,UAAU,EAAE,MAAM;KAClB,UAAU,EAAE,MAAM;KACnB;IACD,QAAQ,EAAE;IACV,SAAS,EAAE;IACZ,CAAC;OAEF,QAAO,KACL,0EAA0E,KAAK,UAAU,EAAE,MAAM,CAAC,GACnG;;EAKP,MAAM,WAAW,WAAW,QAAQ;AACpC,SAAO,MAAM,UAAU,KAAK;AAC5B,eAAa,IAAI,UAAU,KAAK;AAChC,SAAO,KAAK,uBAAuB,WAAW;;AAIhD,KAAI,OAAO,QAAQ;EACjB,MAAM,eAAe,OAAO;EAC5B,MAAM,SAAS,IAAI,YAAY;AAE/B,MAAI,aAAa,YACf,MAAK,MAAM,OAAO,aAAa,aAAa;AAC1C,UAAO,cAAc,IAAI,MAAM,EAAE,WAAW,IAAI,WAAW,CAAC;AAE5D,OAAI,IAAI,WAAW,IAAI,QAAQ,SAAS,EACtC,QAAO,OAAO,IAAI,MAAM,IAAI,QAAQ;AAGtC,OAAI,IAAI,aACN,QAAO,QAAQ,IAAI,MAAM,IAAI,aAAa;;EAKhD,MAAM,aAAa,aAAa,QAAQ;AACxC,SAAO,MAAM,YAAY,OAAO;AAChC,eAAa,IAAI,YAAY,OAAO;AACpC,SAAO,KAAK,yBAAyB,aAAa;;AAMpD,KAAI,cAAc,OAChB,kBAAiB,QAAQ,WAAW;EAClC,SAAS;EACT,SAAS;EACT,UAAU,OAAO,KAAK;EACtB,aAAa,OAAO,KAAK,QAAQ;EACjC,QAAQ,OAAO;EACf,KAAK,QAAQ;EACd,CAAC;AAIJ,KAAI,OAAO,UAAU;AACnB,MAAI,OAAO,SAAS,QAAQ;AAC1B,UAAO,SAAS,MAAM,EAAE,CAAC;AACzB,UAAO,KAAK,oDAAoD;;AAElE,MAAI,OAAO,SAAS,QAAQ;AAC1B,UAAO,SAAS,MAAM,EAAE,CAAC;AACzB,UAAO,KAAK,oDAAoD;;AAElE,MAAI,OAAO,SAAS,UAAU;AAC5B,UAAO,WAAW,MAAM;IAAE,SAAS;IAAO,YAAY,EAAE;IAAE,CAAC;AAC3D,UAAO,KAAK,4DAA4D;;;AAK5E,QAAO;EAAE;EAAQ,KADL,MAAM,OAAO,OAAO;EACV"}
+{"version":3,"file":"config-loader.js","names":[],"sources":["../src/config-loader.ts"],"sourcesContent":["import * as fs from \"node:fs\";\nimport * as path from \"node:path\";\nimport { LLMock, createLLMockWithResolvedAuth } from \"./llmock.js\";\nimport { resolveInboundAuth, selectInboundAuthSource } from \"./api-key-auth.js\";\nimport { MCPMock } from \"./mcp-mock.js\";\nimport { configToolHandlers } from \"./mcp-handler.js\";\nimport { A2AMock } from \"./a2a-mock.js\";\nimport { AGUIMock } from \"./agui-mock.js\";\nimport type {\n  AimockPolicy,\n  ApiKeyAuthConfig,\n  ChaosConfig,\n  McpRecordConfig,\n  MisbehaviorConfig,\n  MisbehaviorFaultId,\n  Mountable,\n  RecordConfig,\n  ResponsesToolsMode,\n} from \"./types.js\";\nimport { parseMisbehavior } from \"./misbehavior.js\";\nimport type { MCPToolDefinition, MCPPromptDefinition } from \"./mcp-types.js\";\nimport type { A2AAgentDefinition, A2APart, A2AArtifact, A2AStreamEvent } from \"./a2a-types.js\";\nimport type { AGUIEvent } from \"./agui-types.js\";\nimport { buildTextResponse } from \"./agui-handler.js\";\nimport { VectorMock } from \"./vector-mock.js\";\nimport type { QueryResult } from \"./vector-types.js\";\nimport { Logger } from \"./logger.js\";\nimport { wireMcpRecording } from \"./mcp-recorder.js\";\nimport { assignPolicyEntry, parsePolicy, parsePolicyEntry } from \"./response/policy.js\";\n\nexport interface MCPConfigTool extends MCPToolDefinition {\n  result?: string;\n}\n\nexport interface MCPConfigResource {\n  uri: string;\n  name: string;\n  mimeType?: string;\n  description?: string;\n  text?: string;\n  blob?: string;\n}\n\nexport interface MCPConfigPrompt extends MCPPromptDefinition {\n  result?: {\n    messages: Array<{ role: string; content: { type: string; text: string } }>;\n  };\n}\n\nexport interface MCPConfig {\n  path?: string;\n  serverInfo?: { name: string; version: string };\n  tools?: MCPConfigTool[];\n  resources?: MCPConfigResource[];\n  prompts?: MCPConfigPrompt[];\n}\n\nexport interface A2AConfigPattern {\n  pattern: string;\n  parts?: A2APart[];\n  artifacts?: A2AArtifact[];\n  events?: A2AStreamEvent[];\n  delayMs?: number;\n}\n\nexport interface A2AConfigAgent extends A2AAgentDefinition {\n  messages?: A2AConfigPattern[];\n  tasks?: A2AConfigPattern[];\n  streamingTasks?: A2AConfigPattern[];\n}\n\nexport interface A2AConfig {\n  path?: string;\n  agents?: A2AConfigAgent[];\n}\n\nexport interface AGUIConfigFixture {\n  match: { message?: string; toolCallId?: string; toolName?: string; stateKey?: string };\n  text?: string; // shorthand: uses buildTextResponse\n  events?: AGUIEvent[]; // raw events\n  delayMs?: number;\n}\n\nexport interface AGUIConfig {\n  path?: string; // mount path, default \"/agui\"\n  fixtures?: AGUIConfigFixture[];\n}\n\nexport interface VectorConfigCollection {\n  name: string;\n  dimension: number;\n  vectors?: Array<{\n    id: string;\n    values: number[];\n    metadata?: Record<string, unknown>;\n  }>;\n  queryResults?: QueryResult[];\n}\n\nexport interface VectorConfig {\n  path?: string;\n  collections?: VectorConfigCollection[];\n}\n\n/** `llm.record` in a config file: the LLM `RecordConfig`, plus MCP record mounts (C10). */\nexport interface AimockRecordConfig extends RecordConfig {\n  /** MR1: mount path → upstream URL, or a full MCP record config. */\n  mcp?: Record<string, string | McpRecordConfig>;\n}\n\nexport interface AimockConfig {\n  auth?: ApiKeyAuthConfig;\n  llm?: {\n    fixtures?: string;\n    latency?: number;\n    chunkSize?: number;\n    replaySpeed?: number;\n    logLevel?: \"silent\" | \"warn\" | \"info\" | \"debug\";\n    chaos?: ChaosConfig;\n    /**\n     * Server-wide misbehavior defaults. Read only when misbehavior is enabled\n     * (`llm.enableMisbehavior: true` or `aimock --misbehavior`); otherwise the\n     * key is ignored with a warning, as 1.44.0 ignored it.\n     */\n    misbehavior?: MisbehaviorConfig | MisbehaviorFaultId;\n    /** See `MockServerOptions.enableMisbehavior`. */\n    enableMisbehavior?: boolean;\n    record?: AimockRecordConfig;\n    /**\n     * Turns on `llm.record.mcp`. Without it the key is ignored with a warning,\n     * as 1.44.0 ignored it.\n     */\n    enableMcpRecording?: boolean;\n    /** See `MockServerOptions.strictToolArguments`. */\n    strictToolArguments?: boolean;\n    /** See `MockServerOptions.responsesTools`. */\n    responsesTools?: ResponsesToolsMode;\n    /** See `MockServerOptions.policy`. Independent of `strict`. */\n    policy?: AimockPolicy;\n  };\n  mcp?: MCPConfig;\n  a2a?: A2AConfig;\n  agui?: AGUIConfig;\n  vector?: VectorConfig;\n  services?: { search?: boolean; rerank?: boolean; moderate?: boolean };\n  metrics?: boolean;\n  strict?: boolean;\n  port?: number;\n  host?: string;\n}\n\n/** C10: the LLM part of `llm.record`. `mcp` is MCP-only; no `providers` means no LLM recording. */\nfunction llmRecordOf(record: AimockRecordConfig | undefined): RecordConfig | undefined {\n  if (!record) return undefined;\n  const { mcp: _mcp, ...llm } = record;\n  void _mcp; // repo pattern for an intentionally unused binding (byteplus-video.ts:1379-1380)\n  return llm.providers === undefined ? undefined : llm;\n}\n\nexport function loadConfig(configPath: string): AimockConfig {\n  const raw = fs.readFileSync(configPath, \"utf-8\");\n  return JSON.parse(raw) as AimockConfig;\n}\n\n/** C3: `llm.record.mcp` is an object with at least one mount. */\nfunction hasMcpRecordMount(mcp: unknown): boolean {\n  return (\n    typeof mcp === \"object\" && mcp !== null && !Array.isArray(mcp) && Object.keys(mcp).length > 0\n  );\n}\n\n/**\n * Command-line overrides for `startFromConfig`. The opt-in flags win over the\n * matching `llm.*` config keys.\n */\nexport interface StartFromConfigOverrides {\n  port?: number;\n  host?: string;\n  /** `aimock --misbehavior`: overrides `llm.enableMisbehavior`. */\n  enableMisbehavior?: boolean;\n  /** `aimock --strict-tool-arguments`: overrides `llm.strictToolArguments`. */\n  strictToolArguments?: boolean;\n  /** `aimock --responses-tools`: overrides `llm.responsesTools`. */\n  responsesTools?: ResponsesToolsMode;\n  /** `aimock --policy`: overrides individual keys from `llm.policy`. */\n  policy?: AimockPolicy;\n}\n\n/** Both CLI entry points use the shared policy parser and treat failures as usage errors. */\nexport function parsePolicyFlags(entries: string[]) {\n  const policy: AimockPolicy = {};\n  for (const entry of entries) {\n    const separator = entry.indexOf(\"=\");\n    if (separator < 0) {\n      return {\n        ok: false,\n        error: `Invalid --policy ${JSON.stringify(entry)}: expected key=value.`,\n      } as const;\n    }\n    const parsed = parsePolicyEntry(entry.slice(0, separator), entry.slice(separator + 1));\n    if (!parsed.ok) {\n      return {\n        ok: false,\n        error: `Invalid --policy ${JSON.stringify(entry)}: ${parsed.reason} Use key=value.`,\n      } as const;\n    }\n    assignPolicyEntry(policy, parsed);\n  }\n  return { ok: true, policy } as const;\n}\n\nexport async function startFromConfig(\n  config: AimockConfig,\n  overrides?: StartFromConfigOverrides,\n): Promise<{ llmock: LLMock; url: string }> {\n  const logger = new Logger(\"info\");\n  const policy = {\n    ...parsePolicy(config.llm?.policy, (message) => logger.warn(message)),\n    ...parsePolicy(overrides?.policy, (message) => logger.warn(message)),\n  };\n  // 1.44.0 ignored these keys, so a non-boolean value (such as the string\n  // \"true\") is ignored with a warning, as `llm.misbehavior` is, and the\n  // option stays off unless a CLI flag turns it on (`llm.enableMcpRecording`\n  // has no CLI flag).\n  const booleanOptIn = (\n    key: \"enableMisbehavior\" | \"enableMcpRecording\" | \"strictToolArguments\",\n  ): boolean | undefined => {\n    const value: unknown = config.llm?.[key];\n    if (value === undefined || typeof value === \"boolean\") return value;\n    logger.warn(\n      `Ignoring llm.${key} because it must be true or false, got ${JSON.stringify(value)}.`,\n    );\n    return undefined;\n  };\n  const configEnableMisbehavior = booleanOptIn(\"enableMisbehavior\");\n  const configStrictToolArguments = booleanOptIn(\"strictToolArguments\");\n  // MCP recording also needs an explicit opt-in. 1.44.0 ignored\n  // `llm.record.mcp`, so without `llm.enableMcpRecording: true` the key is\n  // ignored (with a warning): nothing is mounted or validated and the log level\n  // is unchanged. With the opt-in, an invalid value fails startup.\n  const enableMcpRecording = booleanOptIn(\"enableMcpRecording\") === true;\n  const mcpRecord = enableMcpRecording ? config.llm?.record?.mcp : undefined;\n  // Misbehavior needs an explicit opt-in that a 1.44.0 config cannot contain.\n  // 1.44.0 ignored `llm.misbehavior`, so without the opt-in the key is ignored\n  // (with a warning) and the server serves exactly what 1.44.0 served. With the\n  // opt-in, an invalid value fails startup like any other enabled-mode error.\n  const enableMisbehavior = (overrides?.enableMisbehavior ?? configEnableMisbehavior) === true;\n  let misbehavior: MisbehaviorConfig | undefined;\n  if (config.llm?.misbehavior !== undefined) {\n    if (!enableMisbehavior) {\n      logger.warn(\n        \"Ignoring llm.misbehavior because misbehavior is not enabled. Set llm.enableMisbehavior: true or pass --misbehavior to use it.\",\n      );\n    } else {\n      const parsed = parseMisbehavior(config.llm.misbehavior, \"llm.misbehavior\");\n      if (!parsed.ok) {\n        throw new TypeError(`${parsed.issue.rule}: ${parsed.issue.path}: ${parsed.issue.message}`);\n      }\n      misbehavior = parsed.config;\n    }\n  }\n  if (config.llm?.record?.mcp !== undefined && !enableMcpRecording) {\n    logger.warn(\n      \"Ignoring llm.record.mcp because MCP recording is not enabled. Set llm.enableMcpRecording: true to use it.\",\n    );\n  }\n\n  // 1.44.0 ignored `llm.responsesTools`, so an invalid value is ignored with a\n  // warning (as `llm.misbehavior` is) and the server runs in \"legacy\" mode,\n  // unless `--responses-tools` sets the mode. The warning names the mode that\n  // actually runs.\n  let responsesTools = config.llm?.responsesTools;\n  if (\n    responsesTools !== undefined &&\n    responsesTools !== \"legacy\" &&\n    responsesTools !== \"extended\"\n  ) {\n    const effective =\n      overrides?.responsesTools !== undefined\n        ? `\"${overrides.responsesTools}\" from --responses-tools`\n        : '\"legacy\"';\n    logger.warn(\n      `Ignoring llm.responsesTools because it must be \"legacy\" or \"extended\", got ${JSON.stringify(responsesTools)}. Using ${effective}.`,\n    );\n    responsesTools = undefined;\n  }\n\n  // A non-positive replaySpeed fails calculateDelay's `speed > 0` check and applies the\n  // full delay rather than none. Mirrors the fixture-level guard in fixture-loader.ts.\n  let replaySpeed = config.llm?.replaySpeed;\n  if (replaySpeed != null && (!Number.isFinite(replaySpeed) || replaySpeed <= 0)) {\n    logger.warn(`llm.replaySpeed must be a positive number, got ${replaySpeed}. Ignoring.`);\n    replaySpeed = undefined;\n  }\n\n  // Load fixtures if specified\n  const resolvedAuth = resolveInboundAuth(selectInboundAuthSource(config.auth));\n  const llmock = createLLMockWithResolvedAuth(\n    {\n      port: overrides?.port ?? config.port ?? 0,\n      host: overrides?.host ?? config.host ?? \"127.0.0.1\",\n      latency: config.llm?.latency,\n      chunkSize: config.llm?.chunkSize,\n      replaySpeed,\n      // llm.record.mcp reports skipped or failed recordings only as MCP-RECORD: warnings,\n      // so an enabled config with at least one mount defaults to \"warn\" rather than the silent\n      // default. An empty llm.record.mcp records nothing. An explicit logLevel wins.\n      logLevel: config.llm?.logLevel ?? (hasMcpRecordMount(mcpRecord) ? \"warn\" : undefined),\n      chaos: config.llm?.chaos,\n      misbehavior,\n      ...(enableMisbehavior ? { enableMisbehavior: true } : {}),\n      record: llmRecordOf(config.llm?.record),\n      metrics: config.metrics,\n      strict: config.strict,\n      strictToolArguments: overrides?.strictToolArguments ?? configStrictToolArguments,\n      responsesTools: overrides?.responsesTools ?? responsesTools,\n      policy,\n    },\n    resolvedAuth,\n  );\n\n  // What this function mounted, by path (MR1: llm.record.mcp records through an MCP mount).\n  const configMounts = new Map<string, Mountable>();\n\n  if (config.llm?.fixtures) {\n    const fixturePath = path.resolve(config.llm.fixtures);\n    const stat = fs.statSync(fixturePath);\n    if (stat.isDirectory()) {\n      llmock.loadFixtureDir(fixturePath);\n    } else {\n      llmock.loadFixtureFile(fixturePath);\n    }\n  }\n\n  // MCP\n  if (config.mcp) {\n    const mcpConfig = config.mcp;\n    const mcp = new MCPMock({\n      serverInfo: mcpConfig.serverInfo,\n    });\n\n    if (mcpConfig.tools) {\n      for (const tool of mcpConfig.tools) {\n        const { result, ...def } = tool;\n        mcp.addTool(def);\n        if (result !== undefined) {\n          const answer = () => result;\n          configToolHandlers.add(answer);\n          mcp.onToolCall(def.name, answer);\n        }\n      }\n    }\n\n    if (mcpConfig.resources) {\n      for (const res of mcpConfig.resources) {\n        mcp.addResource(\n          { uri: res.uri, name: res.name, mimeType: res.mimeType, description: res.description },\n          res.text !== undefined || res.blob !== undefined\n            ? { text: res.text, blob: res.blob, mimeType: res.mimeType }\n            : undefined,\n        );\n      }\n    }\n\n    if (mcpConfig.prompts) {\n      for (const prompt of mcpConfig.prompts) {\n        const { result, ...def } = prompt;\n        if (result) {\n          mcp.addPrompt(def, () => result as import(\"./mcp-types.js\").MCPPromptResult);\n        } else {\n          mcp.addPrompt(def);\n        }\n      }\n    }\n\n    const mcpPath = mcpConfig.path ?? \"/mcp\";\n    llmock.mount(mcpPath, mcp);\n    configMounts.set(mcpPath, mcp);\n    logger.info(`MCPMock mounted at ${mcpPath}`);\n  }\n\n  // A2A\n  if (config.a2a) {\n    const a2aConfig = config.a2a;\n    const a2a = new A2AMock();\n\n    if (a2aConfig.agents) {\n      for (const agentConfig of a2aConfig.agents) {\n        const { messages, tasks, streamingTasks, ...def } = agentConfig;\n        a2a.registerAgent(def);\n\n        if (messages) {\n          for (const m of messages) {\n            a2a.onMessage(def.name, m.pattern, m.parts ?? [{ text: \"\" }]);\n          }\n        }\n\n        if (tasks) {\n          for (const t of tasks) {\n            a2a.onTask(def.name, t.pattern, t.artifacts ?? []);\n          }\n        }\n\n        if (streamingTasks) {\n          for (const s of streamingTasks) {\n            a2a.onStreamingTask(def.name, s.pattern, s.events ?? [], s.delayMs);\n          }\n        }\n      }\n    }\n\n    const a2aPath = a2aConfig.path ?? \"/a2a\";\n    llmock.mount(a2aPath, a2a);\n    configMounts.set(a2aPath, a2a);\n    logger.info(`A2AMock mounted at ${a2aPath}`);\n  }\n\n  // AG-UI\n  if (config.agui) {\n    const aguiConfig = config.agui;\n    const agui = new AGUIMock();\n\n    if (aguiConfig.fixtures) {\n      for (const f of aguiConfig.fixtures) {\n        if (f.match.toolCallId && f.text && !f.events) {\n          logger.warn(\n            `AG-UI fixture uses text shorthand with toolCallId — text shorthand ignores toolCallId matching; use events[] instead (match: ${JSON.stringify(f.match)})`,\n          );\n        }\n        if (f.text) {\n          if (f.match.message !== undefined) {\n            agui.onMessage(f.match.message, f.text, { delayMs: f.delayMs });\n          } else {\n            // No message pattern — register via addFixture so it only matches\n            // on other criteria (toolCallId, toolName, stateKey) instead of\n            // becoming a catch-all that matches every request.\n            agui.addFixture({\n              match: {\n                toolCallId: f.match.toolCallId,\n                toolName: f.match.toolName,\n                stateKey: f.match.stateKey,\n              },\n              events: buildTextResponse(f.text),\n              delayMs: f.delayMs,\n            });\n          }\n        } else if (f.events) {\n          agui.addFixture({\n            match: {\n              message: f.match.message,\n              toolCallId: f.match.toolCallId,\n              toolName: f.match.toolName,\n              stateKey: f.match.stateKey,\n            },\n            events: f.events,\n            delayMs: f.delayMs,\n          });\n        } else {\n          logger.warn(\n            `AG-UI fixture has neither text nor events — it will be skipped (match: ${JSON.stringify(f.match)})`,\n          );\n        }\n      }\n    }\n\n    const aguiPath = aguiConfig.path ?? \"/agui\";\n    llmock.mount(aguiPath, agui);\n    configMounts.set(aguiPath, agui);\n    logger.info(`AGUIMock mounted at ${aguiPath}`);\n  }\n\n  // Vector\n  if (config.vector) {\n    const vectorConfig = config.vector;\n    const vector = new VectorMock();\n\n    if (vectorConfig.collections) {\n      for (const col of vectorConfig.collections) {\n        vector.addCollection(col.name, { dimension: col.dimension });\n\n        if (col.vectors && col.vectors.length > 0) {\n          vector.upsert(col.name, col.vectors);\n        }\n\n        if (col.queryResults) {\n          vector.onQuery(col.name, col.queryResults);\n        }\n      }\n    }\n\n    const vectorPath = vectorConfig.path ?? \"/vector\";\n    llmock.mount(vectorPath, vector);\n    configMounts.set(vectorPath, vector);\n    logger.info(`VectorMock mounted at ${vectorPath}`);\n  }\n\n  // MR1: llm.record.mcp — record mounts (auto-mount an MCPMock where none is).\n  // Read only with llm.enableMcpRecording: true. Any value that is present is then\n  // validated, so null, false, 0 and \"\" fail at start too.\n  if (mcpRecord !== undefined) {\n    wireMcpRecording(llmock, mcpRecord, {\n      mounted: configMounts,\n      mcpMock: MCPMock,\n      fixtures: config.llm?.fixtures,\n      fixturePath: config.llm?.record?.fixturePath,\n      strict: config.strict,\n      env: process.env,\n    });\n  }\n\n  // Services — configure default catch-all responses\n  if (config.services) {\n    if (config.services.search) {\n      llmock.onSearch(/.*/, []);\n      logger.info(\"Search service enabled with default empty results\");\n    }\n    if (config.services.rerank) {\n      llmock.onRerank(/.*/, []);\n      logger.info(\"Rerank service enabled with default empty results\");\n    }\n    if (config.services.moderate) {\n      llmock.onModerate(/.*/, { flagged: false, categories: {} });\n      logger.info(\"Moderation service enabled with default unflagged results\");\n    }\n  }\n\n  const url = await llmock.start();\n  return { llmock, url };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAwJA,SAAS,YAAY,QAAkE;AACrF,KAAI,CAAC,OAAQ,QAAO;CACpB,MAAM,EAAE,KAAK,MAAM,GAAG,QAAQ;AAE9B,QAAO,IAAI,cAAc,SAAY,SAAY;;AAGnD,SAAgB,WAAW,YAAkC;CAC3D,MAAM,MAAM,GAAG,aAAa,YAAY,QAAQ;AAChD,QAAO,KAAK,MAAM,IAAI;;;AAIxB,SAAS,kBAAkB,KAAuB;AAChD,QACE,OAAO,QAAQ,YAAY,QAAQ,QAAQ,CAAC,MAAM,QAAQ,IAAI,IAAI,OAAO,KAAK,IAAI,CAAC,SAAS;;;AAsBhG,SAAgB,iBAAiB,SAAmB;CAClD,MAAM,SAAuB,EAAE;AAC/B,MAAK,MAAM,SAAS,SAAS;EAC3B,MAAM,YAAY,MAAM,QAAQ,IAAI;AACpC,MAAI,YAAY,EACd,QAAO;GACL,IAAI;GACJ,OAAO,oBAAoB,KAAK,UAAU,MAAM,CAAC;GAClD;EAEH,MAAM,SAAS,iBAAiB,MAAM,MAAM,GAAG,UAAU,EAAE,MAAM,MAAM,YAAY,EAAE,CAAC;AACtF,MAAI,CAAC,OAAO,GACV,QAAO;GACL,IAAI;GACJ,OAAO,oBAAoB,KAAK,UAAU,MAAM,CAAC,IAAI,OAAO,OAAO;GACpE;AAEH,oBAAkB,QAAQ,OAAO;;AAEnC,QAAO;EAAE,IAAI;EAAM;EAAQ;;AAG7B,eAAsB,gBACpB,QACA,WAC0C;CAC1C,MAAM,SAAS,IAAI,OAAO,OAAO;CACjC,MAAM,SAAS;EACb,GAAG,YAAY,OAAO,KAAK,SAAS,YAAY,OAAO,KAAK,QAAQ,CAAC;EACrE,GAAG,YAAY,WAAW,SAAS,YAAY,OAAO,KAAK,QAAQ,CAAC;EACrE;CAKD,MAAM,gBACJ,QACwB;EACxB,MAAM,QAAiB,OAAO,MAAM;AACpC,MAAI,UAAU,UAAa,OAAO,UAAU,UAAW,QAAO;AAC9D,SAAO,KACL,gBAAgB,IAAI,yCAAyC,KAAK,UAAU,MAAM,CAAC,GACpF;;CAGH,MAAM,0BAA0B,aAAa,oBAAoB;CACjE,MAAM,4BAA4B,aAAa,sBAAsB;CAKrE,MAAM,qBAAqB,aAAa,qBAAqB,KAAK;CAClE,MAAM,YAAY,qBAAqB,OAAO,KAAK,QAAQ,MAAM;CAKjE,MAAM,qBAAqB,WAAW,qBAAqB,6BAA6B;CACxF,IAAI;AACJ,KAAI,OAAO,KAAK,gBAAgB,OAC9B,KAAI,CAAC,kBACH,QAAO,KACL,gIACD;MACI;EACL,MAAM,SAAS,iBAAiB,OAAO,IAAI,aAAa,kBAAkB;AAC1E,MAAI,CAAC,OAAO,GACV,OAAM,IAAI,UAAU,GAAG,OAAO,MAAM,KAAK,IAAI,OAAO,MAAM,KAAK,IAAI,OAAO,MAAM,UAAU;AAE5F,gBAAc,OAAO;;AAGzB,KAAI,OAAO,KAAK,QAAQ,QAAQ,UAAa,CAAC,mBAC5C,QAAO,KACL,4GACD;CAOH,IAAI,iBAAiB,OAAO,KAAK;AACjC,KACE,mBAAmB,UACnB,mBAAmB,YACnB,mBAAmB,YACnB;EACA,MAAM,YACJ,WAAW,mBAAmB,SAC1B,IAAI,UAAU,eAAe,4BAC7B;AACN,SAAO,KACL,8EAA8E,KAAK,UAAU,eAAe,CAAC,UAAU,UAAU,GAClI;AACD,mBAAiB;;CAKnB,IAAI,cAAc,OAAO,KAAK;AAC9B,KAAI,eAAe,SAAS,CAAC,OAAO,SAAS,YAAY,IAAI,eAAe,IAAI;AAC9E,SAAO,KAAK,kDAAkD,YAAY,aAAa;AACvF,gBAAc;;CAIhB,MAAM,eAAe,mBAAmB,wBAAwB,OAAO,KAAK,CAAC;CAC7E,MAAM,SAAS,6BACb;EACE,MAAM,WAAW,QAAQ,OAAO,QAAQ;EACxC,MAAM,WAAW,QAAQ,OAAO,QAAQ;EACxC,SAAS,OAAO,KAAK;EACrB,WAAW,OAAO,KAAK;EACvB;EAIA,UAAU,OAAO,KAAK,aAAa,kBAAkB,UAAU,GAAG,SAAS;EAC3E,OAAO,OAAO,KAAK;EACnB;EACA,GAAI,oBAAoB,EAAE,mBAAmB,MAAM,GAAG,EAAE;EACxD,QAAQ,YAAY,OAAO,KAAK,OAAO;EACvC,SAAS,OAAO;EAChB,QAAQ,OAAO;EACf,qBAAqB,WAAW,uBAAuB;EACvD,gBAAgB,WAAW,kBAAkB;EAC7C;EACD,EACD,aACD;CAGD,MAAM,+BAAe,IAAI,KAAwB;AAEjD,KAAI,OAAO,KAAK,UAAU;EACxB,MAAM,cAAc,KAAK,QAAQ,OAAO,IAAI,SAAS;AAErD,MADa,GAAG,SAAS,YAAY,CAC5B,aAAa,CACpB,QAAO,eAAe,YAAY;MAElC,QAAO,gBAAgB,YAAY;;AAKvC,KAAI,OAAO,KAAK;EACd,MAAM,YAAY,OAAO;EACzB,MAAM,MAAM,IAAI,QAAQ,EACtB,YAAY,UAAU,YACvB,CAAC;AAEF,MAAI,UAAU,MACZ,MAAK,MAAM,QAAQ,UAAU,OAAO;GAClC,MAAM,EAAE,QAAQ,GAAG,QAAQ;AAC3B,OAAI,QAAQ,IAAI;AAChB,OAAI,WAAW,QAAW;IACxB,MAAM,eAAe;AACrB,uBAAmB,IAAI,OAAO;AAC9B,QAAI,WAAW,IAAI,MAAM,OAAO;;;AAKtC,MAAI,UAAU,UACZ,MAAK,MAAM,OAAO,UAAU,UAC1B,KAAI,YACF;GAAE,KAAK,IAAI;GAAK,MAAM,IAAI;GAAM,UAAU,IAAI;GAAU,aAAa,IAAI;GAAa,EACtF,IAAI,SAAS,UAAa,IAAI,SAAS,SACnC;GAAE,MAAM,IAAI;GAAM,MAAM,IAAI;GAAM,UAAU,IAAI;GAAU,GAC1D,OACL;AAIL,MAAI,UAAU,QACZ,MAAK,MAAM,UAAU,UAAU,SAAS;GACtC,MAAM,EAAE,QAAQ,GAAG,QAAQ;AAC3B,OAAI,OACF,KAAI,UAAU,WAAW,OAAmD;OAE5E,KAAI,UAAU,IAAI;;EAKxB,MAAM,UAAU,UAAU,QAAQ;AAClC,SAAO,MAAM,SAAS,IAAI;AAC1B,eAAa,IAAI,SAAS,IAAI;AAC9B,SAAO,KAAK,sBAAsB,UAAU;;AAI9C,KAAI,OAAO,KAAK;EACd,MAAM,YAAY,OAAO;EACzB,MAAM,MAAM,IAAI,SAAS;AAEzB,MAAI,UAAU,OACZ,MAAK,MAAM,eAAe,UAAU,QAAQ;GAC1C,MAAM,EAAE,UAAU,OAAO,gBAAgB,GAAG,QAAQ;AACpD,OAAI,cAAc,IAAI;AAEtB,OAAI,SACF,MAAK,MAAM,KAAK,SACd,KAAI,UAAU,IAAI,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC;AAIjE,OAAI,MACF,MAAK,MAAM,KAAK,MACd,KAAI,OAAO,IAAI,MAAM,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;AAItD,OAAI,eACF,MAAK,MAAM,KAAK,eACd,KAAI,gBAAgB,IAAI,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ;;EAM3E,MAAM,UAAU,UAAU,QAAQ;AAClC,SAAO,MAAM,SAAS,IAAI;AAC1B,eAAa,IAAI,SAAS,IAAI;AAC9B,SAAO,KAAK,sBAAsB,UAAU;;AAI9C,KAAI,OAAO,MAAM;EACf,MAAM,aAAa,OAAO;EAC1B,MAAM,OAAO,IAAI,UAAU;AAE3B,MAAI,WAAW,SACb,MAAK,MAAM,KAAK,WAAW,UAAU;AACnC,OAAI,EAAE,MAAM,cAAc,EAAE,QAAQ,CAAC,EAAE,OACrC,QAAO,KACL,gIAAgI,KAAK,UAAU,EAAE,MAAM,CAAC,GACzJ;AAEH,OAAI,EAAE,KACJ,KAAI,EAAE,MAAM,YAAY,OACtB,MAAK,UAAU,EAAE,MAAM,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC;OAK/D,MAAK,WAAW;IACd,OAAO;KACL,YAAY,EAAE,MAAM;KACpB,UAAU,EAAE,MAAM;KAClB,UAAU,EAAE,MAAM;KACnB;IACD,QAAQ,kBAAkB,EAAE,KAAK;IACjC,SAAS,EAAE;IACZ,CAAC;YAEK,EAAE,OACX,MAAK,WAAW;IACd,OAAO;KACL,SAAS,EAAE,MAAM;KACjB,YAAY,EAAE,MAAM;KACpB,UAAU,EAAE,MAAM;KAClB,UAAU,EAAE,MAAM;KACnB;IACD,QAAQ,EAAE;IACV,SAAS,EAAE;IACZ,CAAC;OAEF,QAAO,KACL,0EAA0E,KAAK,UAAU,EAAE,MAAM,CAAC,GACnG;;EAKP,MAAM,WAAW,WAAW,QAAQ;AACpC,SAAO,MAAM,UAAU,KAAK;AAC5B,eAAa,IAAI,UAAU,KAAK;AAChC,SAAO,KAAK,uBAAuB,WAAW;;AAIhD,KAAI,OAAO,QAAQ;EACjB,MAAM,eAAe,OAAO;EAC5B,MAAM,SAAS,IAAI,YAAY;AAE/B,MAAI,aAAa,YACf,MAAK,MAAM,OAAO,aAAa,aAAa;AAC1C,UAAO,cAAc,IAAI,MAAM,EAAE,WAAW,IAAI,WAAW,CAAC;AAE5D,OAAI,IAAI,WAAW,IAAI,QAAQ,SAAS,EACtC,QAAO,OAAO,IAAI,MAAM,IAAI,QAAQ;AAGtC,OAAI,IAAI,aACN,QAAO,QAAQ,IAAI,MAAM,IAAI,aAAa;;EAKhD,MAAM,aAAa,aAAa,QAAQ;AACxC,SAAO,MAAM,YAAY,OAAO;AAChC,eAAa,IAAI,YAAY,OAAO;AACpC,SAAO,KAAK,yBAAyB,aAAa;;AAMpD,KAAI,cAAc,OAChB,kBAAiB,QAAQ,WAAW;EAClC,SAAS;EACT,SAAS;EACT,UAAU,OAAO,KAAK;EACtB,aAAa,OAAO,KAAK,QAAQ;EACjC,QAAQ,OAAO;EACf,KAAK,QAAQ;EACd,CAAC;AAIJ,KAAI,OAAO,UAAU;AACnB,MAAI,OAAO,SAAS,QAAQ;AAC1B,UAAO,SAAS,MAAM,EAAE,CAAC;AACzB,UAAO,KAAK,oDAAoD;;AAElE,MAAI,OAAO,SAAS,QAAQ;AAC1B,UAAO,SAAS,MAAM,EAAE,CAAC;AACzB,UAAO,KAAK,oDAAoD;;AAElE,MAAI,OAAO,SAAS,UAAU;AAC5B,UAAO,WAAW,MAAM;IAAE,SAAS;IAAO,YAAY,EAAE;IAAE,CAAC;AAC3D,UAAO,KAAK,4DAA4D;;;AAK5E,QAAO;EAAE;EAAQ,KADL,MAAM,OAAO,OAAO;EACV"}

package/dist/count-tokens.cjs

@@ -0,0 +1,146 @@
+const require_estimate = require('./response/usage/estimate.cjs');
+const require_profiles = require('./response/profiles.cjs');
+const require_errors = require('./response/errors.cjs');
+const require_policy = require('./response/policy.cjs');
+const require_helpers = require('./helpers.cjs');
+const require_messages = require('./messages.cjs');
+const require_gemini = require('./gemini.cjs');
+
+//#region src/count-tokens.ts
+function toolsError(tools, google) {
+	if (tools === void 0) return;
+	if (!Array.isArray(tools)) return "tools must be an array";
+	for (const tool of tools) {
+		if (!require_helpers.isJsonObject(tool)) return "tools entries must be objects";
+		if (google && tool.functionDeclarations !== void 0) {
+			if (!Array.isArray(tool.functionDeclarations)) return "functionDeclarations must be an array";
+			if (tool.functionDeclarations.some((declaration) => !require_helpers.isJsonObject(declaration))) return "functionDeclarations entries must be objects";
+		}
+	}
+}
+function hasObjectText(blocks, google, includeThought = false) {
+	return blocks.some((block) => require_helpers.isJsonObject(block) && (google ? includeThought || !block.thought : block.type === "text") && block.text !== null && typeof block.text === "object");
+}
+function contentError(contents, google) {
+	if (!Array.isArray(contents)) return google ? "contents must be an array" : "messages must be an array";
+	for (const content of contents) {
+		if (!require_helpers.isJsonObject(content)) return google ? "contents entries must be objects" : "messages entries must be objects";
+		const parts = google ? content.parts : content.content;
+		if (google) {
+			if (parts !== void 0 && !Array.isArray(parts)) return "parts must be an array";
+		} else if (typeof parts !== "string" && !Array.isArray(parts)) return "message content must be a string or an array";
+		if (Array.isArray(parts) && parts.some((part) => !require_helpers.isJsonObject(part))) return "content blocks must be objects";
+		const role = google ? content.role ?? "user" : content.role;
+		const consumed = google ? role === "user" || role === "model" : role === "user" || role === "assistant";
+		if (Array.isArray(parts) && consumed) {
+			if (hasObjectText(parts, google)) return "content text must not be an object or array";
+			if (!google && role === "user" && parts.some((block) => require_helpers.isJsonObject(block) && block.type === "tool_result" && Array.isArray(block.content) && hasObjectText(block.content, false))) return "tool_result content text must not be an object or array";
+		}
+	}
+}
+/** Counts always use the existing conversion and estimator; no fixture or egress path. */
+function handleCountTokens(req, res, raw, google, model, journal, defaults, setCorsHeaders) {
+	setCorsHeaders(res);
+	const reject = (message, requestId) => {
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "?",
+			headers: require_helpers.flattenHeaders(req.headers),
+			body: null,
+			response: {
+				status: 400,
+				fixture: null,
+				source: "internal"
+			}
+		});
+		const rendered = google ? require_errors.renderError("gemini", "fixture", {
+			message,
+			bodyCode: 400,
+			status: "INVALID_ARGUMENT"
+		}) : require_errors.renderError("messages", "thinking", {
+			message,
+			requestId
+		});
+		res.writeHead(400, {
+			"Content-Type": "application/json",
+			...rendered.headers
+		});
+		res.end(rendered.body);
+	};
+	const policy = require_policy.resolvePolicy(defaults, req.headers);
+	const missingHeader = !google && policy.requiredHeaders === "enforce" ? require_profiles.missingRequiredHeader("messages", req.headers) : void 0;
+	if (missingHeader) {
+		reject(`${missingHeader}: header is required`, require_helpers.resolveRequestId(req.headers).id);
+		return;
+	}
+	let body;
+	try {
+		body = JSON.parse(raw);
+	} catch (error) {
+		reject(`Malformed JSON body: ${error instanceof Error ? error.message : "unknown"}`);
+		return;
+	}
+	if (google && require_helpers.isJsonObject(body) && body.generateContentRequest !== void 0) body = body.generateContentRequest;
+	if (!require_helpers.isJsonObject(body)) {
+		reject("Request body must be an object");
+		return;
+	}
+	const invalidContent = contentError(google ? body.contents : body.messages, google);
+	const invalidTools = toolsError(body.tools, google);
+	const invalidBody = invalidContent ?? invalidTools;
+	if (invalidBody) {
+		reject(invalidBody);
+		return;
+	}
+	if (!google && body.system !== void 0) {
+		if (typeof body.system !== "string" && !Array.isArray(body.system)) {
+			reject("system must be a string or an array");
+			return;
+		}
+		if (Array.isArray(body.system) && body.system.some((block) => !require_helpers.isJsonObject(block))) {
+			reject("system blocks must be objects");
+			return;
+		}
+		if (Array.isArray(body.system) && hasObjectText(body.system, false)) {
+			reject("system text must not be an object or array");
+			return;
+		}
+	}
+	if (google && body.systemInstruction !== void 0 && body.systemInstruction !== null) {
+		if (!require_helpers.isJsonObject(body.systemInstruction) || body.systemInstruction.parts !== void 0 && body.systemInstruction.parts !== null && !Array.isArray(body.systemInstruction.parts)) {
+			reject("systemInstruction must be an object with an array of parts");
+			return;
+		}
+		if (Array.isArray(body.systemInstruction.parts) && body.systemInstruction.parts.some((part) => !require_helpers.isJsonObject(part))) {
+			reject("systemInstruction parts must be objects");
+			return;
+		}
+		if (Array.isArray(body.systemInstruction.parts) && hasObjectText(body.systemInstruction.parts, true, true)) {
+			reject("systemInstruction text must not be an object or array");
+			return;
+		}
+	}
+	const converted = google ? require_gemini.geminiToCompletionRequest(body, model, false) : require_messages.claudeToCompletionRequest({
+		...body,
+		model: typeof body.model === "string" ? body.model : "",
+		max_tokens: 0
+	});
+	const count = require_estimate.estimatePromptTokens(converted.messages) + (converted.tools === void 0 ? 0 : require_estimate.estimateTokens(JSON.stringify(converted.tools)));
+	journal.add({
+		method: req.method ?? "POST",
+		path: req.url ?? "?",
+		headers: require_helpers.flattenHeaders(req.headers),
+		body: converted,
+		response: {
+			status: 200,
+			fixture: null,
+			source: "internal"
+		}
+	});
+	res.writeHead(200, { "Content-Type": "application/json" });
+	res.end(JSON.stringify(google ? { totalTokens: count } : { input_tokens: count }));
+}
+
+//#endregion
+exports.handleCountTokens = handleCountTokens;
+//# sourceMappingURL=count-tokens.cjs.map

package/dist/count-tokens.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"count-tokens.cjs","names":["isJsonObject","flattenHeaders","renderError","resolvePolicy","missingRequiredHeader","resolveRequestId","geminiToCompletionRequest","claudeToCompletionRequest","estimatePromptTokens","estimateTokens"],"sources":["../src/count-tokens.ts"],"sourcesContent":["import type * as http from \"node:http\";\nimport { claudeToCompletionRequest } from \"./messages.js\";\nimport { geminiToCompletionRequest } from \"./gemini.js\";\nimport { flattenHeaders, isJsonObject, resolveRequestId } from \"./helpers.js\";\nimport type { Journal } from \"./journal.js\";\nimport type { ChatCompletionRequest, HandlerDefaults } from \"./types.js\";\nimport { renderError } from \"./response/errors.js\";\nimport { resolvePolicy } from \"./response/policy.js\";\nimport { missingRequiredHeader } from \"./response/profiles.js\";\nimport { estimatePromptTokens, estimateTokens } from \"./response/usage/estimate.js\";\n\ntype ClaudeRequest = Parameters<typeof claudeToCompletionRequest>[0];\ntype GeminiRequest = Parameters<typeof geminiToCompletionRequest>[0];\n\nfunction toolsError(tools: unknown, google: boolean): string | undefined {\n  if (tools === undefined) return;\n  if (!Array.isArray(tools)) return \"tools must be an array\";\n  for (const tool of tools) {\n    if (!isJsonObject(tool)) return \"tools entries must be objects\";\n    if (google && tool.functionDeclarations !== undefined) {\n      if (!Array.isArray(tool.functionDeclarations)) return \"functionDeclarations must be an array\";\n      if (tool.functionDeclarations.some((declaration: unknown) => !isJsonObject(declaration)))\n        return \"functionDeclarations entries must be objects\";\n    }\n  }\n}\n\nfunction hasObjectText(blocks: unknown[], google: boolean, includeThought = false): boolean {\n  return blocks.some(\n    (block) =>\n      isJsonObject(block) &&\n      (google ? includeThought || !block.thought : block.type === \"text\") &&\n      block.text !== null &&\n      typeof block.text === \"object\",\n  );\n}\n\nfunction contentError(contents: unknown, google: boolean): string | undefined {\n  if (!Array.isArray(contents))\n    return google ? \"contents must be an array\" : \"messages must be an array\";\n  for (const content of contents) {\n    if (!isJsonObject(content))\n      return google ? \"contents entries must be objects\" : \"messages entries must be objects\";\n    const parts = google ? content.parts : content.content;\n    if (google) {\n      if (parts !== undefined && !Array.isArray(parts)) return \"parts must be an array\";\n    } else if (typeof parts !== \"string\" && !Array.isArray(parts)) {\n      return \"message content must be a string or an array\";\n    }\n    if (Array.isArray(parts) && parts.some((part: unknown) => !isJsonObject(part)))\n      return \"content blocks must be objects\";\n    const role = google ? (content.role ?? \"user\") : content.role;\n    const consumed = google\n      ? role === \"user\" || role === \"model\"\n      : role === \"user\" || role === \"assistant\";\n    if (Array.isArray(parts) && consumed) {\n      if (hasObjectText(parts, google)) return \"content text must not be an object or array\";\n      if (\n        !google &&\n        role === \"user\" &&\n        parts.some(\n          (block) =>\n            isJsonObject(block) &&\n            block.type === \"tool_result\" &&\n            Array.isArray(block.content) &&\n            hasObjectText(block.content, false),\n        )\n      )\n        return \"tool_result content text must not be an object or array\";\n    }\n  }\n}\n\n/** Counts always use the existing conversion and estimator; no fixture or egress path. */\nexport function handleCountTokens(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  raw: string,\n  google: boolean,\n  model: string,\n  journal: Journal,\n  defaults: HandlerDefaults,\n  setCorsHeaders: (res: http.ServerResponse) => void,\n): void {\n  setCorsHeaders(res);\n  const reject = (message: string, requestId?: string): void => {\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"?\",\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 400, fixture: null, source: \"internal\" },\n    });\n    const rendered = google\n      ? renderError(\"gemini\", \"fixture\", { message, bodyCode: 400, status: \"INVALID_ARGUMENT\" })\n      : renderError(\"messages\", \"thinking\", { message, requestId });\n    res.writeHead(400, { \"Content-Type\": \"application/json\", ...rendered.headers });\n    res.end(rendered.body);\n  };\n  const policy = resolvePolicy(defaults, req.headers);\n  const missingHeader =\n    !google && policy.requiredHeaders === \"enforce\"\n      ? missingRequiredHeader(\"messages\", req.headers)\n      : undefined;\n  if (missingHeader) {\n    reject(`${missingHeader}: header is required`, resolveRequestId(req.headers).id);\n    return;\n  }\n  let body: unknown;\n  try {\n    body = JSON.parse(raw);\n  } catch (error) {\n    reject(`Malformed JSON body: ${error instanceof Error ? error.message : \"unknown\"}`);\n    return;\n  }\n  if (google && isJsonObject(body) && body.generateContentRequest !== undefined)\n    body = body.generateContentRequest;\n  if (!isJsonObject(body)) {\n    reject(\"Request body must be an object\");\n    return;\n  }\n  const invalidContent = contentError(google ? body.contents : body.messages, google);\n  const invalidTools = toolsError(body.tools, google);\n  const invalidBody = invalidContent ?? invalidTools;\n  if (invalidBody) {\n    reject(invalidBody);\n    return;\n  }\n  if (!google && body.system !== undefined) {\n    if (typeof body.system !== \"string\" && !Array.isArray(body.system)) {\n      reject(\"system must be a string or an array\");\n      return;\n    }\n    if (Array.isArray(body.system) && body.system.some((block: unknown) => !isJsonObject(block))) {\n      reject(\"system blocks must be objects\");\n      return;\n    }\n    if (Array.isArray(body.system) && hasObjectText(body.system, false)) {\n      reject(\"system text must not be an object or array\");\n      return;\n    }\n  }\n  if (google && body.systemInstruction !== undefined && body.systemInstruction !== null) {\n    if (\n      !isJsonObject(body.systemInstruction) ||\n      (body.systemInstruction.parts !== undefined &&\n        body.systemInstruction.parts !== null &&\n        !Array.isArray(body.systemInstruction.parts))\n    ) {\n      reject(\"systemInstruction must be an object with an array of parts\");\n      return;\n    }\n    if (\n      Array.isArray(body.systemInstruction.parts) &&\n      body.systemInstruction.parts.some((part: unknown) => !isJsonObject(part))\n    ) {\n      reject(\"systemInstruction parts must be objects\");\n      return;\n    }\n    if (\n      Array.isArray(body.systemInstruction.parts) &&\n      hasObjectText(body.systemInstruction.parts, true, true)\n    ) {\n      reject(\"systemInstruction text must not be an object or array\");\n      return;\n    }\n  }\n  // The guards above validate the fields consumed by the existing converters.\n  const converted: ChatCompletionRequest = google\n    ? geminiToCompletionRequest(body as GeminiRequest, model, false)\n    : claudeToCompletionRequest({\n        ...body,\n        model: typeof body.model === \"string\" ? body.model : \"\",\n        max_tokens: 0,\n      } as ClaudeRequest);\n  const count =\n    estimatePromptTokens(converted.messages) +\n    (converted.tools === undefined ? 0 : estimateTokens(JSON.stringify(converted.tools)));\n  journal.add({\n    method: req.method ?? \"POST\",\n    path: req.url ?? \"?\",\n    headers: flattenHeaders(req.headers),\n    body: converted,\n    response: { status: 200, fixture: null, source: \"internal\" },\n  });\n  res.writeHead(200, { \"Content-Type\": \"application/json\" });\n  res.end(JSON.stringify(google ? { totalTokens: count } : { input_tokens: count }));\n}\n"],"mappings":";;;;;;;;;AAcA,SAAS,WAAW,OAAgB,QAAqC;AACvE,KAAI,UAAU,OAAW;AACzB,KAAI,CAAC,MAAM,QAAQ,MAAM,CAAE,QAAO;AAClC,MAAK,MAAM,QAAQ,OAAO;AACxB,MAAI,CAACA,6BAAa,KAAK,CAAE,QAAO;AAChC,MAAI,UAAU,KAAK,yBAAyB,QAAW;AACrD,OAAI,CAAC,MAAM,QAAQ,KAAK,qBAAqB,CAAE,QAAO;AACtD,OAAI,KAAK,qBAAqB,MAAM,gBAAyB,CAACA,6BAAa,YAAY,CAAC,CACtF,QAAO;;;;AAKf,SAAS,cAAc,QAAmB,QAAiB,iBAAiB,OAAgB;AAC1F,QAAO,OAAO,MACX,UACCA,6BAAa,MAAM,KAClB,SAAS,kBAAkB,CAAC,MAAM,UAAU,MAAM,SAAS,WAC5D,MAAM,SAAS,QACf,OAAO,MAAM,SAAS,SACzB;;AAGH,SAAS,aAAa,UAAmB,QAAqC;AAC5E,KAAI,CAAC,MAAM,QAAQ,SAAS,CAC1B,QAAO,SAAS,8BAA8B;AAChD,MAAK,MAAM,WAAW,UAAU;AAC9B,MAAI,CAACA,6BAAa,QAAQ,CACxB,QAAO,SAAS,qCAAqC;EACvD,MAAM,QAAQ,SAAS,QAAQ,QAAQ,QAAQ;AAC/C,MAAI,QACF;OAAI,UAAU,UAAa,CAAC,MAAM,QAAQ,MAAM,CAAE,QAAO;aAChD,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,CAC3D,QAAO;AAET,MAAI,MAAM,QAAQ,MAAM,IAAI,MAAM,MAAM,SAAkB,CAACA,6BAAa,KAAK,CAAC,CAC5E,QAAO;EACT,MAAM,OAAO,SAAU,QAAQ,QAAQ,SAAU,QAAQ;EACzD,MAAM,WAAW,SACb,SAAS,UAAU,SAAS,UAC5B,SAAS,UAAU,SAAS;AAChC,MAAI,MAAM,QAAQ,MAAM,IAAI,UAAU;AACpC,OAAI,cAAc,OAAO,OAAO,CAAE,QAAO;AACzC,OACE,CAAC,UACD,SAAS,UACT,MAAM,MACH,UACCA,6BAAa,MAAM,IACnB,MAAM,SAAS,iBACf,MAAM,QAAQ,MAAM,QAAQ,IAC5B,cAAc,MAAM,SAAS,MAAM,CACtC,CAED,QAAO;;;;;AAMf,SAAgB,kBACd,KACA,KACA,KACA,QACA,OACA,SACA,UACA,gBACM;AACN,gBAAe,IAAI;CACnB,MAAM,UAAU,SAAiB,cAA6B;AAC5D,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAASC,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM,QAAQ;IAAY;GAC7D,CAAC;EACF,MAAM,WAAW,SACbC,2BAAY,UAAU,WAAW;GAAE;GAAS,UAAU;GAAK,QAAQ;GAAoB,CAAC,GACxFA,2BAAY,YAAY,YAAY;GAAE;GAAS;GAAW,CAAC;AAC/D,MAAI,UAAU,KAAK;GAAE,gBAAgB;GAAoB,GAAG,SAAS;GAAS,CAAC;AAC/E,MAAI,IAAI,SAAS,KAAK;;CAExB,MAAM,SAASC,6BAAc,UAAU,IAAI,QAAQ;CACnD,MAAM,gBACJ,CAAC,UAAU,OAAO,oBAAoB,YAClCC,uCAAsB,YAAY,IAAI,QAAQ,GAC9C;AACN,KAAI,eAAe;AACjB,SAAO,GAAG,cAAc,uBAAuBC,iCAAiB,IAAI,QAAQ,CAAC,GAAG;AAChF;;CAEF,IAAI;AACJ,KAAI;AACF,SAAO,KAAK,MAAM,IAAI;UACf,OAAO;AACd,SAAO,wBAAwB,iBAAiB,QAAQ,MAAM,UAAU,YAAY;AACpF;;AAEF,KAAI,UAAUL,6BAAa,KAAK,IAAI,KAAK,2BAA2B,OAClE,QAAO,KAAK;AACd,KAAI,CAACA,6BAAa,KAAK,EAAE;AACvB,SAAO,iCAAiC;AACxC;;CAEF,MAAM,iBAAiB,aAAa,SAAS,KAAK,WAAW,KAAK,UAAU,OAAO;CACnF,MAAM,eAAe,WAAW,KAAK,OAAO,OAAO;CACnD,MAAM,cAAc,kBAAkB;AACtC,KAAI,aAAa;AACf,SAAO,YAAY;AACnB;;AAEF,KAAI,CAAC,UAAU,KAAK,WAAW,QAAW;AACxC,MAAI,OAAO,KAAK,WAAW,YAAY,CAAC,MAAM,QAAQ,KAAK,OAAO,EAAE;AAClE,UAAO,sCAAsC;AAC7C;;AAEF,MAAI,MAAM,QAAQ,KAAK,OAAO,IAAI,KAAK,OAAO,MAAM,UAAmB,CAACA,6BAAa,MAAM,CAAC,EAAE;AAC5F,UAAO,gCAAgC;AACvC;;AAEF,MAAI,MAAM,QAAQ,KAAK,OAAO,IAAI,cAAc,KAAK,QAAQ,MAAM,EAAE;AACnE,UAAO,6CAA6C;AACpD;;;AAGJ,KAAI,UAAU,KAAK,sBAAsB,UAAa,KAAK,sBAAsB,MAAM;AACrF,MACE,CAACA,6BAAa,KAAK,kBAAkB,IACpC,KAAK,kBAAkB,UAAU,UAChC,KAAK,kBAAkB,UAAU,QACjC,CAAC,MAAM,QAAQ,KAAK,kBAAkB,MAAM,EAC9C;AACA,UAAO,6DAA6D;AACpE;;AAEF,MACE,MAAM,QAAQ,KAAK,kBAAkB,MAAM,IAC3C,KAAK,kBAAkB,MAAM,MAAM,SAAkB,CAACA,6BAAa,KAAK,CAAC,EACzE;AACA,UAAO,0CAA0C;AACjD;;AAEF,MACE,MAAM,QAAQ,KAAK,kBAAkB,MAAM,IAC3C,cAAc,KAAK,kBAAkB,OAAO,MAAM,KAAK,EACvD;AACA,UAAO,wDAAwD;AAC/D;;;CAIJ,MAAM,YAAmC,SACrCM,yCAA0B,MAAuB,OAAO,MAAM,GAC9DC,2CAA0B;EACxB,GAAG;EACH,OAAO,OAAO,KAAK,UAAU,WAAW,KAAK,QAAQ;EACrD,YAAY;EACb,CAAkB;CACvB,MAAM,QACJC,sCAAqB,UAAU,SAAS,IACvC,UAAU,UAAU,SAAY,IAAIC,gCAAe,KAAK,UAAU,UAAU,MAAM,CAAC;AACtF,SAAQ,IAAI;EACV,QAAQ,IAAI,UAAU;EACtB,MAAM,IAAI,OAAO;EACjB,SAASR,+BAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK,SAAS;GAAM,QAAQ;GAAY;EAC7D,CAAC;AACF,KAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,KAAI,IAAI,KAAK,UAAU,SAAS,EAAE,aAAa,OAAO,GAAG,EAAE,cAAc,OAAO,CAAC,CAAC"}

package/dist/count-tokens.js

@@ -0,0 +1,146 @@
+import { estimatePromptTokens, estimateTokens } from "./response/usage/estimate.js";
+import { missingRequiredHeader } from "./response/profiles.js";
+import { renderError } from "./response/errors.js";
+import { resolvePolicy } from "./response/policy.js";
+import { flattenHeaders, isJsonObject, resolveRequestId } from "./helpers.js";
+import { claudeToCompletionRequest } from "./messages.js";
+import { geminiToCompletionRequest } from "./gemini.js";
+
+//#region src/count-tokens.ts
+function toolsError(tools, google) {
+	if (tools === void 0) return;
+	if (!Array.isArray(tools)) return "tools must be an array";
+	for (const tool of tools) {
+		if (!isJsonObject(tool)) return "tools entries must be objects";
+		if (google && tool.functionDeclarations !== void 0) {
+			if (!Array.isArray(tool.functionDeclarations)) return "functionDeclarations must be an array";
+			if (tool.functionDeclarations.some((declaration) => !isJsonObject(declaration))) return "functionDeclarations entries must be objects";
+		}
+	}
+}
+function hasObjectText(blocks, google, includeThought = false) {
+	return blocks.some((block) => isJsonObject(block) && (google ? includeThought || !block.thought : block.type === "text") && block.text !== null && typeof block.text === "object");
+}
+function contentError(contents, google) {
+	if (!Array.isArray(contents)) return google ? "contents must be an array" : "messages must be an array";
+	for (const content of contents) {
+		if (!isJsonObject(content)) return google ? "contents entries must be objects" : "messages entries must be objects";
+		const parts = google ? content.parts : content.content;
+		if (google) {
+			if (parts !== void 0 && !Array.isArray(parts)) return "parts must be an array";
+		} else if (typeof parts !== "string" && !Array.isArray(parts)) return "message content must be a string or an array";
+		if (Array.isArray(parts) && parts.some((part) => !isJsonObject(part))) return "content blocks must be objects";
+		const role = google ? content.role ?? "user" : content.role;
+		const consumed = google ? role === "user" || role === "model" : role === "user" || role === "assistant";
+		if (Array.isArray(parts) && consumed) {
+			if (hasObjectText(parts, google)) return "content text must not be an object or array";
+			if (!google && role === "user" && parts.some((block) => isJsonObject(block) && block.type === "tool_result" && Array.isArray(block.content) && hasObjectText(block.content, false))) return "tool_result content text must not be an object or array";
+		}
+	}
+}
+/** Counts always use the existing conversion and estimator; no fixture or egress path. */
+function handleCountTokens(req, res, raw, google, model, journal, defaults, setCorsHeaders) {
+	setCorsHeaders(res);
+	const reject = (message, requestId) => {
+		journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? "?",
+			headers: flattenHeaders(req.headers),
+			body: null,
+			response: {
+				status: 400,
+				fixture: null,
+				source: "internal"
+			}
+		});
+		const rendered = google ? renderError("gemini", "fixture", {
+			message,
+			bodyCode: 400,
+			status: "INVALID_ARGUMENT"
+		}) : renderError("messages", "thinking", {
+			message,
+			requestId
+		});
+		res.writeHead(400, {
+			"Content-Type": "application/json",
+			...rendered.headers
+		});
+		res.end(rendered.body);
+	};
+	const policy = resolvePolicy(defaults, req.headers);
+	const missingHeader = !google && policy.requiredHeaders === "enforce" ? missingRequiredHeader("messages", req.headers) : void 0;
+	if (missingHeader) {
+		reject(`${missingHeader}: header is required`, resolveRequestId(req.headers).id);
+		return;
+	}
+	let body;
+	try {
+		body = JSON.parse(raw);
+	} catch (error) {
+		reject(`Malformed JSON body: ${error instanceof Error ? error.message : "unknown"}`);
+		return;
+	}
+	if (google && isJsonObject(body) && body.generateContentRequest !== void 0) body = body.generateContentRequest;
+	if (!isJsonObject(body)) {
+		reject("Request body must be an object");
+		return;
+	}
+	const invalidContent = contentError(google ? body.contents : body.messages, google);
+	const invalidTools = toolsError(body.tools, google);
+	const invalidBody = invalidContent ?? invalidTools;
+	if (invalidBody) {
+		reject(invalidBody);
+		return;
+	}
+	if (!google && body.system !== void 0) {
+		if (typeof body.system !== "string" && !Array.isArray(body.system)) {
+			reject("system must be a string or an array");
+			return;
+		}
+		if (Array.isArray(body.system) && body.system.some((block) => !isJsonObject(block))) {
+			reject("system blocks must be objects");
+			return;
+		}
+		if (Array.isArray(body.system) && hasObjectText(body.system, false)) {
+			reject("system text must not be an object or array");
+			return;
+		}
+	}
+	if (google && body.systemInstruction !== void 0 && body.systemInstruction !== null) {
+		if (!isJsonObject(body.systemInstruction) || body.systemInstruction.parts !== void 0 && body.systemInstruction.parts !== null && !Array.isArray(body.systemInstruction.parts)) {
+			reject("systemInstruction must be an object with an array of parts");
+			return;
+		}
+		if (Array.isArray(body.systemInstruction.parts) && body.systemInstruction.parts.some((part) => !isJsonObject(part))) {
+			reject("systemInstruction parts must be objects");
+			return;
+		}
+		if (Array.isArray(body.systemInstruction.parts) && hasObjectText(body.systemInstruction.parts, true, true)) {
+			reject("systemInstruction text must not be an object or array");
+			return;
+		}
+	}
+	const converted = google ? geminiToCompletionRequest(body, model, false) : claudeToCompletionRequest({
+		...body,
+		model: typeof body.model === "string" ? body.model : "",
+		max_tokens: 0
+	});
+	const count = estimatePromptTokens(converted.messages) + (converted.tools === void 0 ? 0 : estimateTokens(JSON.stringify(converted.tools)));
+	journal.add({
+		method: req.method ?? "POST",
+		path: req.url ?? "?",
+		headers: flattenHeaders(req.headers),
+		body: converted,
+		response: {
+			status: 200,
+			fixture: null,
+			source: "internal"
+		}
+	});
+	res.writeHead(200, { "Content-Type": "application/json" });
+	res.end(JSON.stringify(google ? { totalTokens: count } : { input_tokens: count }));
+}
+
+//#endregion
+export { handleCountTokens };
+//# sourceMappingURL=count-tokens.js.map

package/dist/count-tokens.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"count-tokens.js","names":[],"sources":["../src/count-tokens.ts"],"sourcesContent":["import type * as http from \"node:http\";\nimport { claudeToCompletionRequest } from \"./messages.js\";\nimport { geminiToCompletionRequest } from \"./gemini.js\";\nimport { flattenHeaders, isJsonObject, resolveRequestId } from \"./helpers.js\";\nimport type { Journal } from \"./journal.js\";\nimport type { ChatCompletionRequest, HandlerDefaults } from \"./types.js\";\nimport { renderError } from \"./response/errors.js\";\nimport { resolvePolicy } from \"./response/policy.js\";\nimport { missingRequiredHeader } from \"./response/profiles.js\";\nimport { estimatePromptTokens, estimateTokens } from \"./response/usage/estimate.js\";\n\ntype ClaudeRequest = Parameters<typeof claudeToCompletionRequest>[0];\ntype GeminiRequest = Parameters<typeof geminiToCompletionRequest>[0];\n\nfunction toolsError(tools: unknown, google: boolean): string | undefined {\n  if (tools === undefined) return;\n  if (!Array.isArray(tools)) return \"tools must be an array\";\n  for (const tool of tools) {\n    if (!isJsonObject(tool)) return \"tools entries must be objects\";\n    if (google && tool.functionDeclarations !== undefined) {\n      if (!Array.isArray(tool.functionDeclarations)) return \"functionDeclarations must be an array\";\n      if (tool.functionDeclarations.some((declaration: unknown) => !isJsonObject(declaration)))\n        return \"functionDeclarations entries must be objects\";\n    }\n  }\n}\n\nfunction hasObjectText(blocks: unknown[], google: boolean, includeThought = false): boolean {\n  return blocks.some(\n    (block) =>\n      isJsonObject(block) &&\n      (google ? includeThought || !block.thought : block.type === \"text\") &&\n      block.text !== null &&\n      typeof block.text === \"object\",\n  );\n}\n\nfunction contentError(contents: unknown, google: boolean): string | undefined {\n  if (!Array.isArray(contents))\n    return google ? \"contents must be an array\" : \"messages must be an array\";\n  for (const content of contents) {\n    if (!isJsonObject(content))\n      return google ? \"contents entries must be objects\" : \"messages entries must be objects\";\n    const parts = google ? content.parts : content.content;\n    if (google) {\n      if (parts !== undefined && !Array.isArray(parts)) return \"parts must be an array\";\n    } else if (typeof parts !== \"string\" && !Array.isArray(parts)) {\n      return \"message content must be a string or an array\";\n    }\n    if (Array.isArray(parts) && parts.some((part: unknown) => !isJsonObject(part)))\n      return \"content blocks must be objects\";\n    const role = google ? (content.role ?? \"user\") : content.role;\n    const consumed = google\n      ? role === \"user\" || role === \"model\"\n      : role === \"user\" || role === \"assistant\";\n    if (Array.isArray(parts) && consumed) {\n      if (hasObjectText(parts, google)) return \"content text must not be an object or array\";\n      if (\n        !google &&\n        role === \"user\" &&\n        parts.some(\n          (block) =>\n            isJsonObject(block) &&\n            block.type === \"tool_result\" &&\n            Array.isArray(block.content) &&\n            hasObjectText(block.content, false),\n        )\n      )\n        return \"tool_result content text must not be an object or array\";\n    }\n  }\n}\n\n/** Counts always use the existing conversion and estimator; no fixture or egress path. */\nexport function handleCountTokens(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  raw: string,\n  google: boolean,\n  model: string,\n  journal: Journal,\n  defaults: HandlerDefaults,\n  setCorsHeaders: (res: http.ServerResponse) => void,\n): void {\n  setCorsHeaders(res);\n  const reject = (message: string, requestId?: string): void => {\n    journal.add({\n      method: req.method ?? \"POST\",\n      path: req.url ?? \"?\",\n      headers: flattenHeaders(req.headers),\n      body: null,\n      response: { status: 400, fixture: null, source: \"internal\" },\n    });\n    const rendered = google\n      ? renderError(\"gemini\", \"fixture\", { message, bodyCode: 400, status: \"INVALID_ARGUMENT\" })\n      : renderError(\"messages\", \"thinking\", { message, requestId });\n    res.writeHead(400, { \"Content-Type\": \"application/json\", ...rendered.headers });\n    res.end(rendered.body);\n  };\n  const policy = resolvePolicy(defaults, req.headers);\n  const missingHeader =\n    !google && policy.requiredHeaders === \"enforce\"\n      ? missingRequiredHeader(\"messages\", req.headers)\n      : undefined;\n  if (missingHeader) {\n    reject(`${missingHeader}: header is required`, resolveRequestId(req.headers).id);\n    return;\n  }\n  let body: unknown;\n  try {\n    body = JSON.parse(raw);\n  } catch (error) {\n    reject(`Malformed JSON body: ${error instanceof Error ? error.message : \"unknown\"}`);\n    return;\n  }\n  if (google && isJsonObject(body) && body.generateContentRequest !== undefined)\n    body = body.generateContentRequest;\n  if (!isJsonObject(body)) {\n    reject(\"Request body must be an object\");\n    return;\n  }\n  const invalidContent = contentError(google ? body.contents : body.messages, google);\n  const invalidTools = toolsError(body.tools, google);\n  const invalidBody = invalidContent ?? invalidTools;\n  if (invalidBody) {\n    reject(invalidBody);\n    return;\n  }\n  if (!google && body.system !== undefined) {\n    if (typeof body.system !== \"string\" && !Array.isArray(body.system)) {\n      reject(\"system must be a string or an array\");\n      return;\n    }\n    if (Array.isArray(body.system) && body.system.some((block: unknown) => !isJsonObject(block))) {\n      reject(\"system blocks must be objects\");\n      return;\n    }\n    if (Array.isArray(body.system) && hasObjectText(body.system, false)) {\n      reject(\"system text must not be an object or array\");\n      return;\n    }\n  }\n  if (google && body.systemInstruction !== undefined && body.systemInstruction !== null) {\n    if (\n      !isJsonObject(body.systemInstruction) ||\n      (body.systemInstruction.parts !== undefined &&\n        body.systemInstruction.parts !== null &&\n        !Array.isArray(body.systemInstruction.parts))\n    ) {\n      reject(\"systemInstruction must be an object with an array of parts\");\n      return;\n    }\n    if (\n      Array.isArray(body.systemInstruction.parts) &&\n      body.systemInstruction.parts.some((part: unknown) => !isJsonObject(part))\n    ) {\n      reject(\"systemInstruction parts must be objects\");\n      return;\n    }\n    if (\n      Array.isArray(body.systemInstruction.parts) &&\n      hasObjectText(body.systemInstruction.parts, true, true)\n    ) {\n      reject(\"systemInstruction text must not be an object or array\");\n      return;\n    }\n  }\n  // The guards above validate the fields consumed by the existing converters.\n  const converted: ChatCompletionRequest = google\n    ? geminiToCompletionRequest(body as GeminiRequest, model, false)\n    : claudeToCompletionRequest({\n        ...body,\n        model: typeof body.model === \"string\" ? body.model : \"\",\n        max_tokens: 0,\n      } as ClaudeRequest);\n  const count =\n    estimatePromptTokens(converted.messages) +\n    (converted.tools === undefined ? 0 : estimateTokens(JSON.stringify(converted.tools)));\n  journal.add({\n    method: req.method ?? \"POST\",\n    path: req.url ?? \"?\",\n    headers: flattenHeaders(req.headers),\n    body: converted,\n    response: { status: 200, fixture: null, source: \"internal\" },\n  });\n  res.writeHead(200, { \"Content-Type\": \"application/json\" });\n  res.end(JSON.stringify(google ? { totalTokens: count } : { input_tokens: count }));\n}\n"],"mappings":";;;;;;;;;AAcA,SAAS,WAAW,OAAgB,QAAqC;AACvE,KAAI,UAAU,OAAW;AACzB,KAAI,CAAC,MAAM,QAAQ,MAAM,CAAE,QAAO;AAClC,MAAK,MAAM,QAAQ,OAAO;AACxB,MAAI,CAAC,aAAa,KAAK,CAAE,QAAO;AAChC,MAAI,UAAU,KAAK,yBAAyB,QAAW;AACrD,OAAI,CAAC,MAAM,QAAQ,KAAK,qBAAqB,CAAE,QAAO;AACtD,OAAI,KAAK,qBAAqB,MAAM,gBAAyB,CAAC,aAAa,YAAY,CAAC,CACtF,QAAO;;;;AAKf,SAAS,cAAc,QAAmB,QAAiB,iBAAiB,OAAgB;AAC1F,QAAO,OAAO,MACX,UACC,aAAa,MAAM,KAClB,SAAS,kBAAkB,CAAC,MAAM,UAAU,MAAM,SAAS,WAC5D,MAAM,SAAS,QACf,OAAO,MAAM,SAAS,SACzB;;AAGH,SAAS,aAAa,UAAmB,QAAqC;AAC5E,KAAI,CAAC,MAAM,QAAQ,SAAS,CAC1B,QAAO,SAAS,8BAA8B;AAChD,MAAK,MAAM,WAAW,UAAU;AAC9B,MAAI,CAAC,aAAa,QAAQ,CACxB,QAAO,SAAS,qCAAqC;EACvD,MAAM,QAAQ,SAAS,QAAQ,QAAQ,QAAQ;AAC/C,MAAI,QACF;OAAI,UAAU,UAAa,CAAC,MAAM,QAAQ,MAAM,CAAE,QAAO;aAChD,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,CAC3D,QAAO;AAET,MAAI,MAAM,QAAQ,MAAM,IAAI,MAAM,MAAM,SAAkB,CAAC,aAAa,KAAK,CAAC,CAC5E,QAAO;EACT,MAAM,OAAO,SAAU,QAAQ,QAAQ,SAAU,QAAQ;EACzD,MAAM,WAAW,SACb,SAAS,UAAU,SAAS,UAC5B,SAAS,UAAU,SAAS;AAChC,MAAI,MAAM,QAAQ,MAAM,IAAI,UAAU;AACpC,OAAI,cAAc,OAAO,OAAO,CAAE,QAAO;AACzC,OACE,CAAC,UACD,SAAS,UACT,MAAM,MACH,UACC,aAAa,MAAM,IACnB,MAAM,SAAS,iBACf,MAAM,QAAQ,MAAM,QAAQ,IAC5B,cAAc,MAAM,SAAS,MAAM,CACtC,CAED,QAAO;;;;;AAMf,SAAgB,kBACd,KACA,KACA,KACA,QACA,OACA,SACA,UACA,gBACM;AACN,gBAAe,IAAI;CACnB,MAAM,UAAU,SAAiB,cAA6B;AAC5D,UAAQ,IAAI;GACV,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK,SAAS;IAAM,QAAQ;IAAY;GAC7D,CAAC;EACF,MAAM,WAAW,SACb,YAAY,UAAU,WAAW;GAAE;GAAS,UAAU;GAAK,QAAQ;GAAoB,CAAC,GACxF,YAAY,YAAY,YAAY;GAAE;GAAS;GAAW,CAAC;AAC/D,MAAI,UAAU,KAAK;GAAE,gBAAgB;GAAoB,GAAG,SAAS;GAAS,CAAC;AAC/E,MAAI,IAAI,SAAS,KAAK;;CAExB,MAAM,SAAS,cAAc,UAAU,IAAI,QAAQ;CACnD,MAAM,gBACJ,CAAC,UAAU,OAAO,oBAAoB,YAClC,sBAAsB,YAAY,IAAI,QAAQ,GAC9C;AACN,KAAI,eAAe;AACjB,SAAO,GAAG,cAAc,uBAAuB,iBAAiB,IAAI,QAAQ,CAAC,GAAG;AAChF;;CAEF,IAAI;AACJ,KAAI;AACF,SAAO,KAAK,MAAM,IAAI;UACf,OAAO;AACd,SAAO,wBAAwB,iBAAiB,QAAQ,MAAM,UAAU,YAAY;AACpF;;AAEF,KAAI,UAAU,aAAa,KAAK,IAAI,KAAK,2BAA2B,OAClE,QAAO,KAAK;AACd,KAAI,CAAC,aAAa,KAAK,EAAE;AACvB,SAAO,iCAAiC;AACxC;;CAEF,MAAM,iBAAiB,aAAa,SAAS,KAAK,WAAW,KAAK,UAAU,OAAO;CACnF,MAAM,eAAe,WAAW,KAAK,OAAO,OAAO;CACnD,MAAM,cAAc,kBAAkB;AACtC,KAAI,aAAa;AACf,SAAO,YAAY;AACnB;;AAEF,KAAI,CAAC,UAAU,KAAK,WAAW,QAAW;AACxC,MAAI,OAAO,KAAK,WAAW,YAAY,CAAC,MAAM,QAAQ,KAAK,OAAO,EAAE;AAClE,UAAO,sCAAsC;AAC7C;;AAEF,MAAI,MAAM,QAAQ,KAAK,OAAO,IAAI,KAAK,OAAO,MAAM,UAAmB,CAAC,aAAa,MAAM,CAAC,EAAE;AAC5F,UAAO,gCAAgC;AACvC;;AAEF,MAAI,MAAM,QAAQ,KAAK,OAAO,IAAI,cAAc,KAAK,QAAQ,MAAM,EAAE;AACnE,UAAO,6CAA6C;AACpD;;;AAGJ,KAAI,UAAU,KAAK,sBAAsB,UAAa,KAAK,sBAAsB,MAAM;AACrF,MACE,CAAC,aAAa,KAAK,kBAAkB,IACpC,KAAK,kBAAkB,UAAU,UAChC,KAAK,kBAAkB,UAAU,QACjC,CAAC,MAAM,QAAQ,KAAK,kBAAkB,MAAM,EAC9C;AACA,UAAO,6DAA6D;AACpE;;AAEF,MACE,MAAM,QAAQ,KAAK,kBAAkB,MAAM,IAC3C,KAAK,kBAAkB,MAAM,MAAM,SAAkB,CAAC,aAAa,KAAK,CAAC,EACzE;AACA,UAAO,0CAA0C;AACjD;;AAEF,MACE,MAAM,QAAQ,KAAK,kBAAkB,MAAM,IAC3C,cAAc,KAAK,kBAAkB,OAAO,MAAM,KAAK,EACvD;AACA,UAAO,wDAAwD;AAC/D;;;CAIJ,MAAM,YAAmC,SACrC,0BAA0B,MAAuB,OAAO,MAAM,GAC9D,0BAA0B;EACxB,GAAG;EACH,OAAO,OAAO,KAAK,UAAU,WAAW,KAAK,QAAQ;EACrD,YAAY;EACb,CAAkB;CACvB,MAAM,QACJ,qBAAqB,UAAU,SAAS,IACvC,UAAU,UAAU,SAAY,IAAI,eAAe,KAAK,UAAU,UAAU,MAAM,CAAC;AACtF,SAAQ,IAAI;EACV,QAAQ,IAAI,UAAU;EACtB,MAAM,IAAI,OAAO;EACjB,SAAS,eAAe,IAAI,QAAQ;EACpC,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK,SAAS;GAAM,QAAQ;GAAY;EAC7D,CAAC;AACF,KAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,KAAI,IAAI,KAAK,UAAU,SAAS,EAAE,aAAa,OAAO,GAAG,EAAE,cAAc,OAAO,CAAC,CAAC"}