@theokit/sdk 2.18.0 → 2.19.0

package/dist/generate-object.d.ts

@@ -34,6 +34,17 @@ export interface GenerateObjectOptions<T extends ZodType> {
      * when the right env key is set under a different provider name.
      */
     providers?: ProviderRoutingSettings;
+    /**
+     * What to do when the model's output fails schema validation after all
+     * retries are exhausted (M14):
+     * - `'throw'` (default) — throw {@link GenerateObjectError} `parse_failed`.
+     * - `'return-raw'` — resolve with the raw, UNVALIDATED input the model sent
+     *   (`object` may not match the schema; inspect `raw` too).
+     * - `'return-partial'` — for object schemas, resolve with only the fields that
+     *   individually validate (best-effort salvage); non-object schemas fall back
+     *   to raw.
+     */
+    errorStrategy?: "throw" | "return-partial" | "return-raw";
 }
 /**
  * Successful return from {@link Agent.generateObject}.

package/dist/index.cjs

@@ -3800,6 +3800,19 @@ __export(generate_object_exports, {
   GenerateObjectError: () => exports.GenerateObjectError,
   generateObjectImpl: () => generateObjectImpl
 });
+function salvagePartial(schema, raw) {
+  if (typeof raw !== "object" || raw === null) return raw;
+  const shape = schema.shape;
+  if (shape === void 0 || typeof shape !== "object") return raw;
+  const rawObj = raw;
+  const out = {};
+  for (const [key2, fieldSchema] of Object.entries(shape)) {
+    if (typeof fieldSchema?.safeParse !== "function") continue;
+    const parsed = fieldSchema.safeParse(rawObj[key2]);
+    if (parsed.success) out[key2] = parsed.data;
+  }
+  return out;
+}
 async function generateObjectImpl(options, deps) {
   const { jsonSchema, maxRetries, initialUsage } = setupStructuredOutput(
     options.schema,
@@ -3866,6 +3879,22 @@ async function generateObjectImpl(options, deps) {
       }
       lastParseError = parsed.error;
     }
+    if (options.errorStrategy === "return-raw") {
+      return {
+        object: capturedRaw,
+        raw: capturedRaw,
+        usage: lastUsage,
+        finishReason: "tool_use"
+      };
+    }
+    if (options.errorStrategy === "return-partial") {
+      return {
+        object: salvagePartial(options.schema, capturedRaw),
+        raw: capturedRaw,
+        usage: lastUsage,
+        finishReason: "tool_use"
+      };
+    }
     throw new exports.GenerateObjectError(
       "parse_failed",
       "Schema parse failed after all retries.",
@@ -11682,22 +11711,23 @@ async function executeTool(inputs, resolved, call) {
     return { stdout: "", stderr: `Unknown tool ${call.name}`, exitCode: 127 };
   }
   if (resolved.origin === "shell") return runShellTool(inputs, call);
-  if (resolved.origin === "memory") return runMemoryTool(resolved, call);
-  if (resolved.origin === "custom") return runCustomTool(resolved, call, inputs.signal);
+  if (resolved.origin === "memory") return runMemoryTool(resolved, call, inputs.context);
+  if (resolved.origin === "custom")
+    return runCustomTool(resolved, call, inputs.signal, inputs.context);
   return runMcpTool(inputs, resolved, call);
 }
-async function runMemoryTool(resolved, call) {
-  return runHandlerTool("memory", resolved.memoryHandler, call);
+async function runMemoryTool(resolved, call, context) {
+  return runHandlerTool("memory", resolved.memoryHandler, call, void 0, context);
 }
-async function runCustomTool(resolved, call, signal) {
-  return runHandlerTool("custom", resolved.customHandler, call, signal);
+async function runCustomTool(resolved, call, signal, context) {
+  return runHandlerTool("custom", resolved.customHandler, call, signal, context);
 }
-async function runHandlerTool(kind, handler, call, signal) {
+async function runHandlerTool(kind, handler, call, signal, context) {
   if (handler === void 0) {
     return { stdout: "", stderr: `${kind} tool ${call.name} has no handler`, exitCode: 127 };
   }
   try {
-    const stdout = await handler(call.input, { signal });
+    const stdout = await handler(call.input, { signal, context });
     return { stdout, stderr: "", exitCode: 0 };
   } catch (cause) {
     const message = cause instanceof Error ? cause.message : String(cause);
@@ -15596,6 +15626,9 @@ function buildLoopInputs(options, runId, userText) {
     // D318 — forward SendOptions.signal to the agent loop so streamLlmTurn
     // can attach it to the LLM `fetch({ signal })` call.
     ...options.sendOptions.signal !== void 0 ? { signal: options.sendOptions.signal } : {},
+    // M7 — forward SendOptions.context to the loop so every tool handler receives
+    // it on `ctx.context` (shared run config set once, e.g. projectRoot).
+    ...options.sendOptions.context !== void 0 ? { context: options.sendOptions.context } : {},
     // #58 / #57 — forward the per-tool timeout + tool-result guard so a consumer
     // can enable them via SendOptions (not only internal AgentLoopInputs).
     ...options.sendOptions.perToolTimeoutMs !== void 0 ? { perToolTimeoutMs: options.sendOptions.perToolTimeoutMs } : {},