@theokit/sdk 2.2.0 → 2.4.0

package/dist/index.js

@@ -2796,6 +2796,71 @@ var init_agent_factory_registry = __esm({
   }
 });
 
+// src/internal/runtime/lifecycle/run-to-completion.ts
+var run_to_completion_exports = {};
+__export(run_to_completion_exports, {
+  classifyRound: () => classifyRound,
+  runToCompletionImpl: () => runToCompletionImpl
+});
+function isEmptyRound(result) {
+  return (result.result ?? "").trim() === "";
+}
+function classifyRound(result, round, maxRounds, emptyStreak) {
+  if (result.stoppedAtIterationLimit !== true) return "done";
+  if (isEmptyRound(result) && emptyStreak >= 1) return "no_progress";
+  if (round >= maxRounds) return "step_limit";
+  return "continue";
+}
+function addUsage(acc, u) {
+  if (u === void 0) return acc;
+  const inputTokens = (acc?.inputTokens ?? 0) + u.inputTokens;
+  const outputTokens = (acc?.outputTokens ?? 0) + u.outputTokens;
+  const sumOpt = (a, b) => a === void 0 && b === void 0 ? void 0 : (a ?? 0) + (b ?? 0);
+  return {
+    inputTokens,
+    outputTokens,
+    totalTokens: inputTokens + outputTokens,
+    cacheReadTokens: sumOpt(acc?.cacheReadTokens, u.cacheReadTokens),
+    cacheWriteTokens: sumOpt(acc?.cacheWriteTokens, u.cacheWriteTokens),
+    reasoningTokens: sumOpt(acc?.reasoningTokens, u.reasoningTokens)
+  };
+}
+function buildResult(terminal, rounds, lastResult, usage) {
+  return { terminal, rounds, lastResult, ...usage !== void 0 ? { usage } : {} };
+}
+async function stepRound(agent, prompt, sendOptions, round, maxRounds, state4) {
+  const run = await agent.send(prompt, sendOptions);
+  const result = await run.wait();
+  const usage = addUsage(state4.usage, result.usage);
+  const decision = classifyRound(result, round, maxRounds, state4.emptyStreak);
+  if (decision !== "continue") return { terminal: buildResult(decision, round, result, usage) };
+  const emptyStreak = isEmptyRound(result) ? state4.emptyStreak + 1 : 0;
+  return { next: { usage, emptyStreak }, lastResult: result };
+}
+async function runToCompletionImpl(agent, message, options) {
+  const maxRounds = options?.maxRounds ?? DEFAULT_MAX_ROUNDS;
+  const continuationPrompt = options?.continuationPrompt ?? DEFAULT_CONTINUATION_PROMPT;
+  const { onTruncated, signal, sendOptions } = options ?? {};
+  let state4 = { usage: void 0, emptyStreak: 0 };
+  for (let round = 0; ; round += 1) {
+    const prompt = round === 0 ? message : continuationPrompt;
+    const outcome = await stepRound(agent, prompt, sendOptions, round, maxRounds, state4);
+    if ("terminal" in outcome) return outcome.terminal;
+    state4 = outcome.next;
+    await onTruncated?.({ round });
+    if (signal?.aborted === true) {
+      return buildResult("step_limit", round, outcome.lastResult, state4.usage);
+    }
+  }
+}
+var DEFAULT_MAX_ROUNDS, DEFAULT_CONTINUATION_PROMPT;
+var init_run_to_completion = __esm({
+  "src/internal/runtime/lifecycle/run-to-completion.ts"() {
+    DEFAULT_MAX_ROUNDS = 5;
+    DEFAULT_CONTINUATION_PROMPT = "Continue from where you left off and finish the task. If it is already complete, give the final answer.";
+  }
+});
+
 // src/internal/runtime/lifecycle/fork-agent.ts
 var fork_agent_exports = {};
 __export(fork_agent_exports, {
@@ -2884,10 +2949,10 @@ var RingBuffer;
 var init_ring_buffer = __esm({
   "src/internal/task/ring-buffer.ts"() {
     RingBuffer = class {
-      constructor(cap) {
-        this.cap = cap;
-        if (!Number.isInteger(cap) || cap < 1) {
-          throw new Error(`RingBuffer capacity must be a positive integer, got ${cap}`);
+      constructor(cap2) {
+        this.cap = cap2;
+        if (!Number.isInteger(cap2) || cap2 < 1) {
+          throw new Error(`RingBuffer capacity must be a positive integer, got ${cap2}`);
         }
       }
       cap;
@@ -7351,6 +7416,18 @@ var CloudAgent = class {
       "fork"
     );
   }
+  /**
+   * The continuation driver re-sends against a stateful local session; the
+   * cloud runtime manages its own continuation policy server-side (M1 Phase 3).
+   *
+   * @public
+   */
+  runToCompletion() {
+    throw new UnsupportedRunOperationError(
+      "Agent.runToCompletion() is not supported on cloud agents. Cloud runtime manages continuation server-side. Use a local agent.",
+      "runToCompletion"
+    );
+  }
   /**
    * Personality presets require consistent server-side enforcement that
    * the cloud runtime (pre-release) does not yet provide. Reject explicitly
@@ -8879,6 +8956,8 @@ function parseSubagentMarkdown(raw, filename) {
   if (fields.model !== void 0) {
     definition.model = fields.model === "inherit" ? "inherit" : { id: fields.model };
   }
+  const tools = fields.tools?.split(/[\s,]+/).map((t) => t.trim()).filter((t) => t.length > 0);
+  if (tools !== void 0 && tools.length > 0) definition.tools = tools;
   return { name, definition };
 }
 function splitFrontmatter2(raw, filename) {
@@ -9042,21 +9121,24 @@ ${lines.join("\n")}
   }
 };
 
+// src/internal/runtime/skills/skills-block.ts
+function buildSkillsBlock(skills) {
+  if (skills.length === 0) return void 0;
+  const lines = skills.map(
+    (skill) => `  - ${escapeBlockBody(skill.name)}: ${escapeBlockBody(skill.description)}`
+  );
+  return `<skills>
+${lines.join("\n")}
+</skills>`;
+}
+
 // src/internal/runtime/system-prompt/sources/skills-provider.ts
 var SkillsPromptProvider = class {
   id = "skills";
   priority = 20;
   contribute(ctx) {
     if (ctx.skillsAutoInject === false) return Promise.resolve(void 0);
-    if (ctx.skills.length === 0) return Promise.resolve(void 0);
-    const lines = ctx.skills.map((skill) => {
-      const name = escapeBlockBody(skill.name);
-      const description = escapeBlockBody(skill.description);
-      return `  - ${name}: ${description}`;
-    });
-    return Promise.resolve(`<skills>
-${lines.join("\n")}
-</skills>`);
+    return Promise.resolve(buildSkillsBlock(ctx.skills));
   }
 };
 
@@ -10122,7 +10204,7 @@ async function loadPluginManifestFromJson(manifestPath, folderName) {
   const record = parsed;
   const name = typeof record.name === "string" ? record.name : folderName;
   const version = typeof record.version === "string" ? record.version : "0.0.0";
-  const capabilities = Array.isArray(record.capabilities) ? record.capabilities.filter((cap) => typeof cap === "string") : [];
+  const capabilities = Array.isArray(record.capabilities) ? record.capabilities.filter((cap2) => typeof cap2 === "string") : [];
   const source = manifestPath.slice(manifestPath.indexOf(".theokit/"));
   const metadata = { name, version, capabilities, source };
   if (typeof record.entry === "string") metadata.entry = record.entry;
@@ -10165,7 +10247,7 @@ async function loadPluginManifestFromMarkdown(pluginsRoot, folderName) {
   return metadata;
 }
 
-// src/internal/runtime/skills/skills-manager.ts
+// src/internal/runtime/skills/discover-skills.ts
 init_errors();
 init_path_guard();
 
@@ -10238,6 +10320,61 @@ function hasContent(value) {
   return value !== void 0 && value.trim().length > 0;
 }
 
+// src/internal/runtime/skills/discover-skills.ts
+async function discoverSkills(dir, options) {
+  let entries;
+  try {
+    entries = await readWorkspaceDir(dir, "skills_read_error", "skills directory");
+  } catch {
+    return [];
+  }
+  const skills = [];
+  for (const entry of entries) {
+    if (!entry.isDirectory()) continue;
+    let skillDir;
+    try {
+      skillDir = safePathJoin(dir, entry.name);
+      assertNoSymlinkEscape(skillDir, dir);
+    } catch {
+      continue;
+    }
+    const skillPath = join(skillDir, "SKILL.md");
+    let raw;
+    try {
+      raw = await readFile(skillPath, "utf8");
+    } catch {
+      continue;
+    }
+    const skill = tryParseSkill(raw, entry.name, skillPath, options);
+    if (skill !== void 0) skills.push(skill);
+  }
+  return skills;
+}
+function tryParseSkill(raw, fallbackName, source, options) {
+  try {
+    const frontmatter = parseSkillFrontmatter(raw, fallbackName);
+    const skill = {
+      name: frontmatter.name,
+      description: frontmatter.description,
+      source
+    };
+    if (frontmatter.category !== void 0) skill.category = frontmatter.category;
+    if (frontmatter.dependencies !== void 0) skill.dependencies = frontmatter.dependencies;
+    return skill;
+  } catch (cause) {
+    if (cause instanceof ConfigurationError) {
+      options?.onInvalidSkill?.({
+        name: fallbackName,
+        source,
+        code: cause.code ?? "unknown",
+        message: cause.message
+      });
+      return void 0;
+    }
+    throw cause;
+  }
+}
+
 // src/internal/runtime/skills/skills-manager.ts
 var SkillsManager = class {
   constructor(cwd, _enabled, settingSourcesIncludeProject) {
@@ -10255,56 +10392,20 @@ var SkillsManager = class {
     await this.refresh();
   }
   async refresh() {
-    this.skills = [];
     const skillsRoot = join(this.cwd, ".theokit", "skills");
-    const entries = await readWorkspaceDir(skillsRoot, "skills_read_error", "skills directory");
-    for (const entry of entries) {
-      if (!entry.isDirectory()) continue;
-      let skillDir;
-      try {
-        skillDir = safePathJoin(skillsRoot, entry.name);
-        assertNoSymlinkEscape(skillDir, skillsRoot);
-      } catch {
-        continue;
-      }
-      const skillPath = join(skillDir, "SKILL.md");
-      let raw;
-      try {
-        raw = await readFile(skillPath, "utf8");
-      } catch {
-        continue;
+    this.skills = await discoverSkills(skillsRoot, {
+      onInvalidSkill: (info) => {
+        process.stderr.write(
+          `[theokit-sdk] skill ${info.name} skipped (${info.code}): ${info.message}
+`
+        );
       }
-      const metadata = tryParseSkill(raw, entry.name, skillPath);
-      if (metadata !== void 0) this.skills.push(metadata);
-    }
+    });
   }
   list() {
     return Promise.resolve(this.skills);
   }
 };
-function tryParseSkill(raw, fallbackName, source) {
-  try {
-    const frontmatter = parseSkillFrontmatter(raw, fallbackName);
-    const metadata = {
-      name: frontmatter.name,
-      description: frontmatter.description,
-      source
-    };
-    if (frontmatter.category !== void 0) metadata.category = frontmatter.category;
-    if (frontmatter.dependencies !== void 0) metadata.dependencies = frontmatter.dependencies;
-    return metadata;
-  } catch (cause) {
-    if (cause instanceof ConfigurationError) {
-      const code = cause.code ?? "unknown";
-      process.stderr.write(
-        `[theokit-sdk] skill ${fallbackName} skipped (${code}): ${cause.message}
-`
-      );
-      return void 0;
-    }
-    throw cause;
-  }
-}
 
 // src/internal/runtime/local-agent/local-agent-bootstrap.ts
 function registerLocalAgent(args) {
@@ -10717,6 +10818,7 @@ async function initLoopContext(inputs) {
     finalStatus: "finished",
     usage: new UsageAccumulator(),
     nudgeAttempts: 0,
+    stopFeedbackAttempts: 0,
     ...memoryProviderHandle !== void 0 ? { memoryProviderHandle } : {},
     ...memorySystemPromptAdditions !== void 0 ? { memorySystemPromptAdditions } : {}
   };
@@ -10861,8 +10963,9 @@ 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 metaCode = cause?.metadata?.code;
   const rawCode = cause?.code;
-  const code = typeof rawCode === "string" ? rawCode : void 0;
+  const code = typeof metaCode === "string" ? metaCode : typeof rawCode === "string" ? rawCode : void 0;
   ctx.error = code !== void 0 ? { message, code, cause } : { message, cause };
 }
 async function runCollectorLoop(generator, inputs, ctx) {
@@ -11658,6 +11761,7 @@ function computeUsageCost(inputs, usage) {
 
 // src/internal/agent-loop/loop.ts
 var MAX_NUDGE_ATTEMPTS = 2;
+var MAX_STOP_FEEDBACK_ATTEMPTS = 2;
 async function runAgentLoop(inputs) {
   const sendSpan = inputs.telemetry?.startSpan("agent.send", {
     agentId: inputs.agentId,
@@ -11789,8 +11893,8 @@ function handleToolErrorContinuation(inputs, ctx, toolResults) {
   const hasError = toolResults.some((part) => part.type === "tool_result" && part.isError === true);
   if (hasError) {
     ctx._consecutiveToolErrors = (ctx._consecutiveToolErrors ?? 0) + 1;
-    const cap = inputs.maxConsecutiveToolErrors ?? 3;
-    if (ctx._consecutiveToolErrors >= cap) return "error";
+    const cap2 = inputs.maxConsecutiveToolErrors ?? 3;
+    if (ctx._consecutiveToolErrors >= cap2) return "error";
     return "continue";
   }
   ctx._consecutiveToolErrors = 0;
@@ -11815,6 +11919,28 @@ function shouldNudgeAndContinue(ctx, llmOutput) {
   });
   return true;
 }
+async function reflectAfterStop(inputs, ctx) {
+  const result = await inputs.hooks.run({
+    event: "stop",
+    agentId: inputs.agentId,
+    runId: inputs.runId
+  });
+  if (result.blocked) return false;
+  if (ctx.stopFeedbackAttempts >= MAX_STOP_FEEDBACK_ATTEMPTS) return false;
+  const feedback = result.decisions.find(
+    (d) => d.decision === "feedback" && (d.feedback ?? "").length > 0
+  )?.feedback;
+  if (feedback === void 0) return false;
+  ctx.stopFeedbackAttempts += 1;
+  ctx.messages.push({ role: "user", content: [{ type: "text", text: feedback }] });
+  return true;
+}
+async function finishOrReflect(inputs, ctx, llmOutput) {
+  if (shouldNudgeAndContinue(ctx, llmOutput)) return "continue";
+  if (await reflectAfterStop(inputs, ctx)) return "continue";
+  ctx.finalStatus = "finished";
+  return "done";
+}
 async function runIteration(inputs, ctx) {
   const llmOutput = await streamLlmTurn(inputs, ctx);
   accumulateUsage(ctx.usage, llmOutput);
@@ -11848,9 +11974,7 @@ async function continueOrTerminate(inputs, ctx, llmOutput) {
     await emitAssistantTextStep(inputs, ctx, llmOutput.text);
   }
   if (llmOutput.stopReason !== "tool_use" || llmOutput.toolCalls.length === 0) {
-    if (shouldNudgeAndContinue(ctx, llmOutput)) return "continue";
-    ctx.finalStatus = "finished";
-    return "done";
+    return finishOrReflect(inputs, ctx, llmOutput);
   }
   ctx.messages.push(buildAssistantTurn(llmOutput.text, llmOutput.toolCalls));
   const toolResults = await dispatchTools(inputs, ctx.tools, llmOutput.toolCalls, ctx.events);
@@ -15284,7 +15408,7 @@ var MEMORY_GET_SCHEMA = {
 };
 var DEFAULT_MAX_TOTAL_CHARS = 16384;
 function createMemorySearchTool(opts) {
-  const cap = opts.maxTotalChars ?? DEFAULT_MAX_TOTAL_CHARS;
+  const cap2 = opts.maxTotalChars ?? DEFAULT_MAX_TOTAL_CHARS;
   return {
     name: "memory_search",
     description: SEARCH_DESCRIPTION,
@@ -15301,7 +15425,7 @@ function createMemorySearchTool(opts) {
         ...sources !== void 0 ? { sources } : {}
       };
       const hits = await opts.index.search(query, searchOptions);
-      return JSON.stringify(capByTotalChars(hits, cap));
+      return JSON.stringify(capByTotalChars(hits, cap2));
     }
   };
 }
@@ -15869,6 +15993,13 @@ function localAgentRunUntil(agent, goal, options) {
   }
   return wrap();
 }
+function localAgentRunToCompletion(agent, message, options) {
+  async function run() {
+    const { runToCompletionImpl: runToCompletionImpl2 } = await Promise.resolve().then(() => (init_run_to_completion(), run_to_completion_exports));
+    return runToCompletionImpl2({ send: (m, o) => agent.send(m, o) }, message, options);
+  }
+  return run();
+}
 async function localAgentFork(parent, options) {
   const { forkAgentImpl: forkAgentImpl2 } = await Promise.resolve().then(() => (init_fork_agent(), fork_agent_exports));
   const { getAgentFacade: getAgentFacade2 } = await Promise.resolve().then(() => (init_agent_factory_registry(), agent_factory_registry_exports));
@@ -15932,8 +16063,8 @@ async function applyPreUserSendHook(args) {
     ...args.options.memoryContext !== void 0 ? { memoryContext: args.options.memoryContext } : {},
     ...args.sendOptions.signal !== void 0 ? { signal: args.sendOptions.signal } : {}
   };
-  const cap = args.options.maxRecallContextBytes ?? DEFAULT_MAX_RECALL_BYTES;
-  const recalled = await args.pluginManager.runPreUserSendHooks(ctx, cap);
+  const cap2 = args.options.maxRecallContextBytes ?? DEFAULT_MAX_RECALL_BYTES;
+  const recalled = await args.pluginManager.runPreUserSendHooks(ctx, cap2);
   if (recalled === void 0 || recalled.length === 0) return args.original;
   const wrapped = `<memory-context>
 ${recalled}
@@ -16409,6 +16540,10 @@ var LocalAgent = class {
   fork(options) {
     return localAgentFork({ agentId: this.agentId, options: this.options, personalitySlugSnapshot: this.personalityStore.active(this.agentId) }, options);
   }
+  // biome-ignore format: G8 budget — see runUntil comment above.
+  runToCompletion(message, options) {
+    return localAgentRunToCompletion(this, message, options);
+  }
 };
 function resolveCwd(cwd) {
   return (Array.isArray(cwd) ? cwd[0] : cwd) ?? process.cwd();
@@ -17924,6 +18059,77 @@ function createCounterBudgetTracker(options = {}) {
   };
 }
 
+// src/internal/runtime/context/replay-history.ts
+var CHARS_PER_TOKEN = 4;
+var DEFAULT_RESERVE_TOKENS = 8e3;
+function finiteOr(value, fallback) {
+  return Number.isFinite(value) ? value : fallback;
+}
+function charBudget(options) {
+  const window = finiteOr(options.contextWindowTokens, 0);
+  const reserve = finiteOr(options.reserveTokens ?? DEFAULT_RESERVE_TOKENS, DEFAULT_RESERVE_TOKENS);
+  return Math.max(0, window - reserve) * CHARS_PER_TOKEN;
+}
+function assistantText2(event) {
+  return event.message.content.filter((block) => block.type === "text").map((block) => block.text).join("");
+}
+function stringifyPayload(value) {
+  if (value === void 0) return "";
+  return typeof value === "string" ? value : JSON.stringify(value) ?? "";
+}
+function cap(content, perItemCap) {
+  return truncateWithMarker(content, Math.max(0, perItemCap)).finalContent;
+}
+function mapEvent(event, perItemCap) {
+  if (event.type === "assistant") {
+    const text = assistantText2(event);
+    return text.length > 0 ? { message: { role: "assistant", content: cap(text, perItemCap) } } : null;
+  }
+  if (event.type === "tool_call") {
+    if (event.status === "running") {
+      const content2 = cap(`[tool_call ${event.name}] ${stringifyPayload(event.args)}`, perItemCap);
+      return { message: { role: "tool_call", content: content2 }, pairId: event.call_id };
+    }
+    const content = cap(
+      `[tool_result ${event.name}] ${stringifyPayload(event.result)}`,
+      perItemCap
+    );
+    return { message: { role: "tool_result", content }, pairId: event.call_id };
+  }
+  return null;
+}
+function evictionIndices(turns) {
+  const head = turns[0];
+  if (head?.pairId === void 0) return [0];
+  const indices = [];
+  for (let i = 0; i < turns.length; i += 1) {
+    if (turns[i]?.pairId === head.pairId) indices.push(i);
+  }
+  return indices;
+}
+function totalChars(turns) {
+  return turns.reduce((n, t) => n + t.message.content.length, 0);
+}
+function trimToBudget(turns, budgetChars) {
+  const kept = [...turns];
+  while (kept.length > 1 && totalChars(kept) > budgetChars) {
+    const evict = evictionIndices(kept);
+    if (kept.length - evict.length < 1) break;
+    for (const idx of evict.sort((a, b) => b - a)) kept.splice(idx, 1);
+  }
+  return kept;
+}
+function buildReplayHistory(base, events, options) {
+  const budgetChars = charBudget(options);
+  const perItemCap = Math.max(0, options.perItemCap ?? Math.floor(budgetChars / 2));
+  const turns = base.map((message) => ({ message }));
+  for (const event of events) {
+    const turn = mapEvent(event, perItemCap);
+    if (turn !== null) turns.push(turn);
+  }
+  return trimToBudget(turns, budgetChars).map((t) => t.message);
+}
+
 // src/internal/runtime/memory/memory-provider-noop.ts
 var NOOP_ADAPTER_ID = "noop";
 var NOOP_CAPABILITIES = {
@@ -19294,6 +19500,6 @@ function safeStringify2(v) {
   }
 }
 
-export { Agent, AgentBuilder, AgentDisposedError, AgentRunError, AuthenticationError, Budget, BudgetExceededError, ConfigurationError, Cron, EventBus, FileSystemConversationStorage, GenerateObjectError, InMemoryConversationStorage, IntegrationNotConnectedError, InvalidTaskIdError, JobQueue, Memory, MemoryAdapterError, NetworkError, PermissionEngine, RateLimitError, Security, StreamObjectError, Task, TaskNotFoundError, Theokit, TheokitAgentError, UnknownAgentError, UnsupportedBudgetOperationError, UnsupportedRunOperationError, UnsupportedTaskOperationError, UsageAccumulator, chargeAndCheckThresholds, computeCost, createAgentFactory, createCounterBudgetTracker, createNoopMemoryProvider, createSquad, definePlugin, defineProvider, defineTool, extractRawId, getPricingEntry, inferApiMode, isTransientError, migrateSqliteToLance2 as migrateSqliteToLance, mkMemoryId, normalizeUsage, preflightCheck, toShareGptTrajectory, withCwdMutex };
+export { Agent, AgentBuilder, AgentDisposedError, AgentRunError, AuthenticationError, Budget, BudgetExceededError, ConfigurationError, Cron, EventBus, FileSystemConversationStorage, GenerateObjectError, InMemoryConversationStorage, IntegrationNotConnectedError, InvalidTaskIdError, JobQueue, Memory, MemoryAdapterError, NetworkError, PermissionEngine, RateLimitError, Security, StreamObjectError, Task, TaskNotFoundError, Theokit, TheokitAgentError, UnknownAgentError, UnsupportedBudgetOperationError, UnsupportedRunOperationError, UnsupportedTaskOperationError, UsageAccumulator, buildReplayHistory, chargeAndCheckThresholds, computeCost, createAgentFactory, createCounterBudgetTracker, createNoopMemoryProvider, createSquad, definePlugin, defineProvider, defineTool, extractRawId, getPricingEntry, inferApiMode, isTransientError, migrateSqliteToLance2 as migrateSqliteToLance, mkMemoryId, normalizeUsage, preflightCheck, toShareGptTrajectory, withCwdMutex };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map