theokit 0.1.0-alpha.11 → 0.1.0-alpha.12

package/dist/cli/index.js

@@ -23,7 +23,7 @@ cli.command("build", "Build for production").option("--target <target>", "Deploy
 });
 cli.command("start", "Start production server").option("--port <port>", "Port number").action(async (options) => {
   try {
-    const { startCommand } = await import("../start-CSLJQXV3.js");
+    const { startCommand } = await import("../start-7O423RHL.js");
     await startCommand({ port: options.port ? Number(options.port) : void 0 });
   } catch (err) {
     const msg = err instanceof Error ? err.message : String(err);

package/dist/{dist-UZV7J3YI.js → dist-IZY2ZRTK.js}

@@ -2319,6 +2319,57 @@ var init_registry = __esm({
     });
   }
 });
+function toJsonSchema(schema, options = { unrepresentable: "any" }) {
+  const native = tryNativeZod4(schema, options);
+  if (native !== void 0) return native;
+  const fallback = tryFallbackLib(schema);
+  if (fallback !== void 0) return fallback;
+  throw new Error(
+    'Zod \u2192 JSON Schema conversion failed: neither Zod 4\'s native `z.toJSONSchema` nor the `zod-to-json-schema` peer dep are available. On Zod 4: ensure `zod >= 4.0.0` is installed. On Zod 3: add `"zod-to-json-schema": "^3.24.0"` to your package.json.'
+  );
+}
+function tryNativeZod4(schema, options) {
+  if (cachedZodHasNativeApi === false) return void 0;
+  let z12;
+  try {
+    z12 = requireSdk("zod");
+  } catch {
+    cachedZodHasNativeApi = false;
+    return void 0;
+  }
+  if (typeof z12.toJSONSchema !== "function") {
+    cachedZodHasNativeApi = false;
+    return void 0;
+  }
+  cachedZodHasNativeApi = true;
+  return z12.toJSONSchema(schema, options);
+}
+function tryFallbackLib(schema) {
+  if (cachedFallbackLib !== void 0) {
+    return cachedFallbackLib(schema);
+  }
+  if (triedFallback) return void 0;
+  triedFallback = true;
+  try {
+    const mod = requireSdk("zod-to-json-schema");
+    const fn2 = typeof mod.zodToJsonSchema === "function" ? mod.zodToJsonSchema : typeof mod.default === "function" ? mod.default : void 0;
+    if (fn2 === void 0) return void 0;
+    cachedFallbackLib = fn2;
+    return fn2(schema);
+  } catch {
+    return void 0;
+  }
+}
+var requireSdk;
+var cachedZodHasNativeApi;
+var cachedFallbackLib;
+var triedFallback;
+var init_to_json_schema = __esm({
+  "src/internal/zod/to-json-schema.ts"() {
+    requireSdk = createRequire(import.meta.url);
+    triedFallback = false;
+  }
+});
 function requireZod() {
   if (cachedZ !== void 0) return cachedZ;
   const r = createRequire(import.meta.url);
@@ -2369,9 +2420,7 @@ Respond by calling the \`output\` tool with the structured answer that matches t
 }
 function setupStructuredOutput(schema, maxRetries) {
   const z12 = requireZod();
-  const jsonSchema = z12.toJSONSchema(schema, {
-    unrepresentable: "any"
-  });
+  const jsonSchema = toJsonSchema(schema, { unrepresentable: "any" });
   return {
     z: z12,
     jsonSchema,
@@ -2390,6 +2439,7 @@ var cachedZ;
 var init_structured_output_helpers = __esm({
   "src/internal/structured-output-helpers.ts"() {
     init_errors();
+    init_to_json_schema();
   }
 });
 var generate_object_exports = {};
@@ -3169,7 +3219,7 @@ function buildHandoffTool(parentAgentId, descriptor, maxHandoffDepth) {
   const inputZod = descriptor.options.inputType ?? z.object({
     reason: z.string().optional().describe("Brief reason for the transfer (one short sentence).")
   });
-  const inputSchema = z.toJSONSchema(inputZod);
+  const inputSchema = toJsonSchema(inputZod);
   return {
     name: descriptor.resolvedToolName,
     description,
@@ -3204,6 +3254,7 @@ function buildHandoffTool(parentAgentId, descriptor, maxHandoffDepth) {
 var init_tool_injector = __esm({
   "src/internal/handoff/tool-injector.ts"() {
     init_handoff();
+    init_to_json_schema();
     init_dispatcher();
     init_registry2();
   }
@@ -10316,6 +10367,167 @@ function buildBedrockBody(request) {
 init_credential_pool();
 init_credential_pool_context();
 init_errors();
+function mapOpenAICompatibleError(args) {
+  const { providerId, status, body, headers, endpoint } = args;
+  const code = mapOpenAiStatusToCode(status, body);
+  const message = formatMessage2(providerId, status, code);
+  const metadata = buildErrorMetadata({
+    provider: providerId,
+    endpoint,
+    code,
+    status,
+    headers,
+    body
+  });
+  if (status === 401 || status === 403) {
+    return new AuthenticationError(message, { code: `${providerId}_auth_failed`, metadata });
+  }
+  if (status === 429) {
+    return new RateLimitError(message, { code: `${providerId}_rate_limit`, metadata });
+  }
+  if (status === 400) {
+    return new ConfigurationError(message, { code: `${providerId}_${code}`, metadata });
+  }
+  if (status === 408) {
+    return new NetworkError(message, { code: `${providerId}_timeout`, metadata });
+  }
+  if (status >= 500 && status < 600) {
+    return new NetworkError(message, { code: `${providerId}_server_error`, metadata });
+  }
+  return new UnknownAgentError(message, { code: `${providerId}_unknown`, metadata });
+}
+function extractOpenAiErrorCode(body) {
+  if (body === null || typeof body !== "object") return void 0;
+  const err = body.error;
+  if (err === void 0 || err === null) return void 0;
+  if (typeof err.code === "string") return err.code;
+  if (typeof err.type === "string") return err.type;
+  return void 0;
+}
+function mapOpenAiStatusToCode(status, body) {
+  const rawCode = extractOpenAiErrorCode(body)?.toLowerCase() ?? "";
+  if (rawCode.includes("context_length") || rawCode.includes("context_window") || rawCode.includes("too_many_tokens") || rawCode.includes("max_tokens")) {
+    return "context_too_long";
+  }
+  if (rawCode.includes("content_filter") || rawCode.includes("content_policy") || rawCode.includes("safety")) {
+    return "content_filtered";
+  }
+  if (rawCode.includes("model_not_found") || rawCode.includes("model_unavailable") || rawCode.includes("invalid_model")) {
+    return "model_unavailable";
+  }
+  if (status === 402 || rawCode.includes("insufficient_quota") || rawCode.includes("quota_exceeded")) {
+    return "invalid_request";
+  }
+  if (status === 401 || status === 403) return "auth_failed";
+  if (status === 429) return "rate_limit";
+  if (status === 408) return "timeout";
+  if (status === 400) return "invalid_request";
+  if (status >= 500 && status < 600) return "server_error";
+  return "unknown";
+}
+function formatMessage2(providerId, status, code) {
+  return `${providerId} API error: ${code} (HTTP ${status})`;
+}
+var ENV_GATE_KEY = "NODE_ENV";
+var ENV_GATE_VALUE = "test";
+var ENV_OVERRIDE_KEY = "THEOKIT_TEST_RESPONSE_OVERRIDE";
+function maybeWrapWithFaultInjection(client) {
+  if (!isGateOn()) return client;
+  return new FaultInjectingLlmClient(client);
+}
+var FaultInjectingLlmClient = class {
+  name;
+  /**
+   * Inner (real) client. Exposed so layered transport assertions (router
+   * pool-wiring tests, telemetry inspectors) can walk one level deep —
+   * same pattern PoolAwareLlmClient uses for its inner transport.
+   */
+  inner;
+  constructor(inner) {
+    this.inner = inner;
+    this.name = inner.name;
+  }
+  // biome-ignore lint/complexity/noExcessiveCognitiveComplexity: explicit gate → parse → branch is clearer than splitting
+  async *stream(request, signal) {
+    if (!isGateOn()) {
+      return yield* this.inner.stream(request, signal);
+    }
+    const parsed = parseOverrideOrWarnOnce();
+    if (parsed === void 0) {
+      return yield* this.inner.stream(request, signal);
+    }
+    if (parsed.status >= 400) {
+      throw mapOpenAICompatibleError({
+        providerId: this.name,
+        status: parsed.status,
+        body: parsed.body,
+        headers: void 0,
+        endpoint: "/v1/chat/completions"
+      });
+    }
+    const text = extractText(parsed.body);
+    yield { type: "text_delta", text };
+    yield { type: "stop", reason: "end_turn" };
+    return {
+      stopReason: "end_turn",
+      text,
+      toolCalls: [],
+      inputTokens: 0,
+      outputTokens: 0
+    };
+  }
+};
+function isGateOn() {
+  return process.env[ENV_GATE_KEY] === ENV_GATE_VALUE;
+}
+var warnedInvalidJson = false;
+function parseOverrideOrWarnOnce() {
+  const raw = process.env[ENV_OVERRIDE_KEY];
+  if (raw === void 0 || raw.length === 0) return void 0;
+  let candidate;
+  try {
+    candidate = JSON.parse(raw);
+  } catch (_err) {
+    warnInvalidJsonOnce();
+    return void 0;
+  }
+  if (candidate === null || typeof candidate !== "object") {
+    warnInvalidJsonOnce();
+    return void 0;
+  }
+  const obj = candidate;
+  if (typeof obj.status !== "number" || !Number.isFinite(obj.status)) {
+    warnInvalidJsonOnce();
+    return void 0;
+  }
+  return { status: obj.status, body: obj.body };
+}
+function warnInvalidJsonOnce(_raw) {
+  if (warnedInvalidJson) return;
+  warnedInvalidJson = true;
+  process.stderr.write(
+    `[theokit-sdk] ${ENV_OVERRIDE_KEY} is set but value is not valid JSON of shape \`{"status": number, "body": object | string}\`. Falling back to real LLM call. This warning is shown once per process.
+`
+  );
+}
+function extractText(body) {
+  if (typeof body === "string") return body;
+  if (body === null || typeof body !== "object") return "";
+  const fromOpenAi = readOpenAiContent(body);
+  if (fromOpenAi !== void 0) return fromOpenAi;
+  const obj = body;
+  if (typeof obj.text === "string") return obj.text;
+  return "";
+}
+function readOpenAiContent(body) {
+  const obj = body;
+  if (!Array.isArray(obj.choices) || obj.choices.length === 0) return void 0;
+  const first = obj.choices[0];
+  const content = first?.message?.content;
+  if (typeof content === "string") return content;
+  return void 0;
+}
+init_errors();
 function mapOllamaTransportError(args) {
   if (args.providerId !== "ollama") return void 0;
   const causeCode = extractCauseCode(args.cause);
@@ -10614,68 +10826,6 @@ function toOllamaTools(tools) {
     }
   }));
 }
-init_errors();
-function mapOpenAICompatibleError(args) {
-  const { providerId, status, body, headers, endpoint } = args;
-  const code = mapOpenAiStatusToCode(status, body);
-  const message = formatMessage2(providerId, status, code);
-  const metadata = buildErrorMetadata({
-    provider: providerId,
-    endpoint,
-    code,
-    status,
-    headers,
-    body
-  });
-  if (status === 401 || status === 403) {
-    return new AuthenticationError(message, { code: `${providerId}_auth_failed`, metadata });
-  }
-  if (status === 429) {
-    return new RateLimitError(message, { code: `${providerId}_rate_limit`, metadata });
-  }
-  if (status === 400) {
-    return new ConfigurationError(message, { code: `${providerId}_${code}`, metadata });
-  }
-  if (status === 408) {
-    return new NetworkError(message, { code: `${providerId}_timeout`, metadata });
-  }
-  if (status >= 500 && status < 600) {
-    return new NetworkError(message, { code: `${providerId}_server_error`, metadata });
-  }
-  return new UnknownAgentError(message, { code: `${providerId}_unknown`, metadata });
-}
-function extractOpenAiErrorCode(body) {
-  if (body === null || typeof body !== "object") return void 0;
-  const err = body.error;
-  if (err === void 0 || err === null) return void 0;
-  if (typeof err.code === "string") return err.code;
-  if (typeof err.type === "string") return err.type;
-  return void 0;
-}
-function mapOpenAiStatusToCode(status, body) {
-  const rawCode = extractOpenAiErrorCode(body)?.toLowerCase() ?? "";
-  if (rawCode.includes("context_length") || rawCode.includes("context_window") || rawCode.includes("too_many_tokens") || rawCode.includes("max_tokens")) {
-    return "context_too_long";
-  }
-  if (rawCode.includes("content_filter") || rawCode.includes("content_policy") || rawCode.includes("safety")) {
-    return "content_filtered";
-  }
-  if (rawCode.includes("model_not_found") || rawCode.includes("model_unavailable") || rawCode.includes("invalid_model")) {
-    return "model_unavailable";
-  }
-  if (status === 402 || rawCode.includes("insufficient_quota") || rawCode.includes("quota_exceeded")) {
-    return "invalid_request";
-  }
-  if (status === 401 || status === 403) return "auth_failed";
-  if (status === 429) return "rate_limit";
-  if (status === 408) return "timeout";
-  if (status === 400) return "invalid_request";
-  if (status >= 500 && status < 600) return "server_error";
-  return "unknown";
-}
-function formatMessage2(providerId, status, code) {
-  return `${providerId} API error: ${code} (HTTP ${status})`;
-}
 var OpenAIClient = class {
   constructor(options) {
     this.options = options;
@@ -11217,7 +11367,7 @@ function buildChain(options) {
     if (seen.has(lowered)) return;
     seen.add(lowered);
     const client = buildClient(lowered, options);
-    if (client !== void 0) clients.push(client);
+    if (client !== void 0) clients.push(maybeWrapWithFaultInjection(client));
   };
   addClient(options.primary);
   for (const fallback of options.fallback ?? []) addClient(fallback);
@@ -16679,6 +16829,7 @@ async function updateJobStatus(jobId, enabled) {
   }
   return updated;
 }
+init_to_json_schema();
 var cachedZ2;
 function requireZod2() {
   if (cachedZ2 !== void 0) return cachedZ2;
@@ -16695,10 +16846,8 @@ function requireZod2() {
   return cachedZ2;
 }
 function defineTool(spec) {
-  const z12 = requireZod2();
-  const inputSchema = z12.toJSONSchema(spec.inputSchema, {
-    unrepresentable: "any"
-  });
+  requireZod2();
+  const inputSchema = toJsonSchema(spec.inputSchema, { unrepresentable: "any" });
   return {
     name: spec.name,
     description: spec.description,
@@ -18634,4 +18783,4 @@ export {
   preflightCheck,
   toShareGptTrajectory
 };
-//# sourceMappingURL=dist-UZV7J3YI.js.map
+//# sourceMappingURL=dist-IZY2ZRTK.js.map