theokit 0.1.0-alpha.8 → 0.2.0

package/dist/dispatcher-IK3FVMX5.js

@@ -0,0 +1,8 @@
+import {
+  dispatcher
+} from "./chunk-OK2PJOLK.js";
+import "./chunk-DGUM43GV.js";
+export {
+  dispatcher
+};
+//# sourceMappingURL=dispatcher-IK3FVMX5.js.map

package/dist/dispatcher-IK3FVMX5.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/{dist-UZV7J3YI.js → dist-Q5VBFWVL.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();
   }
@@ -9361,7 +9412,10 @@ async function runAgentLoop(inputs) {
       result: ctx.finalText,
       conversation: ctx.conversation,
       ...usage !== void 0 ? { usage } : {},
-      ...cost !== void 0 ? { cost } : {}
+      ...cost !== void 0 ? { cost } : {},
+      // Finding-B fix: surface structured error so runtime can package
+      // it as RunResult.error instead of leaking it as assistant content.
+      ...ctx.error !== void 0 ? { error: ctx.error } : {}
     };
   } finally {
     sendSpan?.end();
@@ -9517,13 +9571,25 @@ async function collectLlmEvents(generator, inputs, ctx) {
     errored = result.errored;
     finishValue = result.finishValue;
   } catch (cause) {
-    const text = inputs.signal?.aborted === true ? "[aborted]" : cause instanceof Error ? cause.message : String(cause);
-    ctx.finalText = text;
-    ctx.events.push(buildAssistantEvent(inputs, text));
+    if (inputs.signal?.aborted === true) {
+      ctx.finalText = "[aborted]";
+      ctx.events.push(buildAssistantEvent(inputs, "[aborted]"));
+    } else {
+      registerLoopError(ctx, cause);
+      ctx.finalText = "";
+    }
     errored = true;
   }
   return { accumulatedText, errored, finishValue };
 }
+function registerLoopError(ctx, cause) {
+  if (ctx.error !== void 0) return;
+  const rawMessage = cause?.message;
+  const message = typeof rawMessage === "string" ? rawMessage : cause instanceof Error ? cause.message : String(cause);
+  const rawCode = cause?.code;
+  const code = typeof rawCode === "string" ? rawCode : void 0;
+  ctx.error = code !== void 0 ? { message, code, cause } : { message, cause };
+}
 async function runCollectorLoop(generator, inputs, ctx) {
   let accumulatedText = "";
   let errored = false;
@@ -9539,8 +9605,8 @@ async function runCollectorLoop(generator, inputs, ctx) {
       await emitTextDeltaCallback(inputs, next.value.text);
     }
     if (next.value.type === "error") {
-      ctx.finalText = next.value.message;
-      ctx.events.push(buildAssistantEvent(inputs, next.value.message));
+      registerLoopError(ctx, next.value);
+      ctx.finalText = "";
       errored = true;
       break;
     }
@@ -10316,6 +10382,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 +10841,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 +11382,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);
@@ -11988,6 +12153,13 @@ var RealLocalRun = class extends FixtureRunBase {
       if (output.result.length > 0) this.script.result = output.result;
       if (output.usage !== void 0) this.script.usage = output.usage;
       if (output.cost !== void 0) this.script.cost = output.cost;
+      if (output.error !== void 0 && this.script.errorDetail === void 0) {
+        this.script.errorDetail = {
+          message: output.error.message,
+          ...output.error.code !== void 0 ? { code: output.error.code } : {},
+          cause: output.error.cause
+        };
+      }
       this.transitionTo(output.finalStatus);
     } catch (cause) {
       this.emitErrorEvent(cause, "Agent loop failed", "", "agent_loop_failed");
@@ -11999,24 +12171,15 @@ var RealLocalRun = class extends FixtureRunBase {
     }
   }
   emitErrorEvent(cause, fallback, prefix = "", code) {
-    const message = cause instanceof Error ? cause.message : String(cause);
-    const display = prefix.length > 0 ? `${prefix}${message}` : message || fallback;
     if (this.script.errorDetail === void 0) {
+      const message = cause instanceof Error ? cause.message : String(cause);
+      const display = prefix.length > 0 ? `${prefix}${message}` : message || fallback;
       this.script.errorDetail = {
         message: display,
         ...code !== void 0 ? { code } : {},
         cause
       };
     }
-    this.script.events.push({
-      type: "assistant",
-      agent_id: this.agentId,
-      run_id: this.id,
-      message: {
-        role: "assistant",
-        content: [{ type: "text", text: display }]
-      }
-    });
     this.notifyNewEvents();
   }
 };
@@ -16679,6 +16842,7 @@ async function updateJobStatus(jobId, enabled) {
   }
   return updated;
 }
+init_to_json_schema();
 var cachedZ2;
 function requireZod2() {
   if (cachedZ2 !== void 0) return cachedZ2;
@@ -16695,10 +16859,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 +18796,4 @@ export {
   preflightCheck,
   toShareGptTrajectory
 };
-//# sourceMappingURL=dist-UZV7J3YI.js.map
+//# sourceMappingURL=dist-Q5VBFWVL.js.map