@theokit/sdk 2.18.1 → 2.20.0

package/dist/define-tool.d.ts

@@ -22,6 +22,7 @@ export interface DefineToolSpec<T extends ZodType> {
      */
     handler: (input: ZodNamespace.infer<T>, ctx?: {
         signal?: AbortSignal;
+        context?: unknown;
     }) => string | Promise<string>;
     /**
      * Sanitize the raw model-emitted args BEFORE schema validation (`@theokit/sdk/sanitize`).

package/dist/{errors-qyVYfk9H.d.ts → errors-5lj1EWgs.d.ts}

@@ -1,4 +1,4 @@
-import { p as RunOperation } from './run-pE-34AAo.js';
+import { p as RunOperation } from './run-BgfBWX-z.js';
 
 /**
  * Public type contract for the Budget enforcement primitive

package/dist/{errors-1tVcX3Fq.d.cts → errors-CE-lMBi2.d.cts}

@@ -1,4 +1,4 @@
-import { p as RunOperation } from './run-pE-34AAo.cjs';
+import { p as RunOperation } from './run-BgfBWX-z.cjs';
 
 /**
  * Public type contract for the Budget enforcement primitive

package/dist/errors.d.cts

@@ -1,2 +1,2 @@
-export { A as AgentDisposedError, c as AgentRunError, d as AgentRunErrorCode, e as AuthenticationError, g as BudgetExceededError, C as ConfigurationError, u as CredentialPoolExhaustedError, E as ErrorCode, m as ErrorMetadata, I as IntegrationNotConnectedError, n as InvalidTaskIdError, K as KnownAgentRunErrorCode, M as MemoryAdapterError, o as MemoryAdapterErrorCode, N as NetworkError, R as RateLimitError, p as TaskNotFoundError, T as TheokitAgentError, U as UnknownAgentError, q as UnsupportedBudgetOperationError, r as UnsupportedRunOperationError, s as UnsupportedTaskOperationError, t as isTransientError } from './errors-1tVcX3Fq.cjs';
-import './run-pE-34AAo.cjs';
+export { A as AgentDisposedError, c as AgentRunError, d as AgentRunErrorCode, e as AuthenticationError, g as BudgetExceededError, C as ConfigurationError, u as CredentialPoolExhaustedError, E as ErrorCode, m as ErrorMetadata, I as IntegrationNotConnectedError, n as InvalidTaskIdError, K as KnownAgentRunErrorCode, M as MemoryAdapterError, o as MemoryAdapterErrorCode, N as NetworkError, R as RateLimitError, p as TaskNotFoundError, T as TheokitAgentError, U as UnknownAgentError, q as UnsupportedBudgetOperationError, r as UnsupportedRunOperationError, s as UnsupportedTaskOperationError, t as isTransientError } from './errors-CE-lMBi2.cjs';
+import './run-BgfBWX-z.cjs';

package/dist/eval.cjs

@@ -2255,6 +2255,37 @@ __export(generate_object_exports, {
   GenerateObjectError: () => 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 [key, fieldSchema] of Object.entries(shape)) {
+    if (typeof fieldSchema?.safeParse !== "function") continue;
+    const parsed = fieldSchema.safeParse(rawObj[key]);
+    if (parsed.success) out[key] = parsed.data;
+  }
+  return out;
+}
+async function runReasoningPhase(options, deps) {
+  const reasoningOptions = {
+    model: options.model,
+    local: options.local,
+    ...options.systemPrompt !== void 0 ? { systemPrompt: options.systemPrompt } : {},
+    ...options.apiKey !== void 0 ? { apiKey: options.apiKey } : {},
+    ...options.providers !== void 0 ? { providers: options.providers } : {}
+  };
+  const reasoningAgent = await deps.create(reasoningOptions);
+  try {
+    const run = await reasoningAgent.send(options.prompt);
+    const result = await run.wait();
+    const text = result.result;
+    return typeof text === "string" ? text : options.prompt;
+  } finally {
+    await disposeAndDeleteTransient(reasoningAgent, deps.delete);
+  }
+}
 async function generateObjectImpl(options, deps) {
   const { jsonSchema, maxRetries, initialUsage } = setupStructuredOutput(
     options.schema,
@@ -2277,8 +2308,11 @@ async function generateObjectImpl(options, deps) {
     }
     throw new CaptureSentinel(input);
   });
+  const reasoningText = options.structuringModel !== void 0 ? await runReasoningPhase(options, deps) : void 0;
+  const structuringModel = options.structuringModel ?? options.model;
+  const structuringPrompt = reasoningText ?? options.prompt;
   const agentOptions = buildTransientAgentOptions({
-    model: options.model,
+    model: structuringModel,
     local: options.local,
     outputTool,
     ...options.systemPrompt !== void 0 ? { systemPrompt: options.systemPrompt } : {},
@@ -2287,7 +2321,7 @@ async function generateObjectImpl(options, deps) {
   });
   const agent = await deps.create(agentOptions);
   try {
-    const userMessage = buildToolPrompt(options.prompt);
+    const userMessage = buildToolPrompt(structuringPrompt);
     let lastParseError;
     for (let attempt = 0; attempt <= maxRetries; attempt += 1) {
       capturedRaw = void 0;
@@ -2321,6 +2355,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 GenerateObjectError(
       "parse_failed",
       "Schema parse failed after all retries.",
@@ -8307,23 +8357,27 @@ function tryParseSkill(raw, fallbackName, source, options) {
 
 // src/internal/runtime/skills/skills-manager.ts
 var SkillsManager = class {
-  constructor(cwd, _enabled, settingSourcesIncludeProject) {
+  constructor(cwd, _enabled, settingSourcesIncludeProject, skillsDir, inline) {
     this.cwd = cwd;
     this.settingSourcesIncludeProject = settingSourcesIncludeProject;
+    this.skillsDir = skillsDir;
+    this.inline = inline;
   }
   cwd;
   settingSourcesIncludeProject;
+  skillsDir;
+  inline;
   skills = [];
   async initialize() {
     if (!this.settingSourcesIncludeProject) {
-      this.skills = [];
+      this.skills = this.mergeInline([]);
       return;
     }
     await this.refresh();
   }
   async refresh() {
-    const skillsRoot = path.join(this.cwd, ".theokit", "skills");
-    this.skills = await discoverSkills(skillsRoot, {
+    const skillsRoot = this.skillsDir ?? path.join(this.cwd, ".theokit", "skills");
+    const discovered = await discoverSkills(skillsRoot, {
       onInvalidSkill: (info) => {
         process.stderr.write(
           `[theokit-sdk] skill ${info.name} skipped (${info.code}): ${info.message}
@@ -8331,6 +8385,13 @@ var SkillsManager = class {
         );
       }
     });
+    this.skills = this.mergeInline(discovered);
+  }
+  /** M22 — merge inline skills over discovered ones; inline wins on a name conflict. */
+  mergeInline(discovered) {
+    if (this.inline === void 0 || this.inline.length === 0) return discovered;
+    const inlineNames = new Set(this.inline.map((s) => s.name));
+    return [...discovered.filter((s) => !inlineNames.has(s.name)), ...this.inline];
   }
   list() {
     return Promise.resolve(this.skills);
@@ -8375,7 +8436,10 @@ function bootstrapSubmanagers(args) {
     out.skillsManager = new SkillsManager(
       args.workspaceCwd,
       args.options.skills?.enabled,
-      args.settingSourcesIncludeProject
+      args.settingSourcesIncludeProject,
+      // M22 — custom skills directory + inline (code-defined) skills.
+      args.options.skills?.skillsDir,
+      args.options.skills?.inline
     );
     const localSkills = out.skillsManager;
     out.skills = { list: () => localSkills.list() };
@@ -9215,22 +9279,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);
@@ -13113,6 +13178,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 } : {},