@theokit/sdk 4.18.0 → 4.19.0

package/dist/index.cjs

@@ -7532,16 +7532,38 @@ async function* runUntilImpl(agent, goal, options, deps) {
   while (turn < maxTurns) {
     if (isAborted()) {
       yield { type: "status_change", status: "paused", reason: "aborted via AbortSignal" };
-      return { status: "paused", turnsUsed: turn, tokensUsed, finalResponse: lastResponse || void 0 };
+      return {
+        status: "paused",
+        turnsUsed: turn,
+        tokensUsed,
+        finalResponse: lastResponse || void 0
+      };
     }
     turn += 1;
     yield { type: "turn_start", turn, goal };
     const continuationPrompt = turn === 1 ? goal : composeContinuation(goal, lastResponse);
     const run = await agent.send(continuationPrompt);
+    const cancelOnAbort = () => {
+      void run.cancel?.();
+    };
+    if (signal !== void 0) {
+      if (signal.aborted) cancelOnAbort();
+      else signal.addEventListener("abort", cancelOnAbort, { once: true });
+    }
     const result = await run.wait();
+    if (signal !== void 0) signal.removeEventListener("abort", cancelOnAbort);
     lastResponse = result.result ?? "";
     const turnTokens = result.usage?.totalTokens;
     if (typeof turnTokens === "number") tokensUsed += turnTokens;
+    if (isAborted()) {
+      yield { type: "status_change", status: "paused", reason: "aborted via AbortSignal" };
+      return {
+        status: "paused",
+        turnsUsed: turn,
+        tokensUsed,
+        finalResponse: lastResponse || void 0
+      };
+    }
     yield { type: "agent_response", turn, content: lastResponse };
     const judgeOpts = {};
     if (options?.judgeModel !== void 0) judgeOpts.judgeModel = options.judgeModel;
@@ -7576,7 +7598,12 @@ async function* runUntilImpl(agent, goal, options, deps) {
     }
     if (judgment.verdict === "done") {
       yield { type: "status_change", status: "completed", reason: judgment.reason };
-      return { status: "completed", turnsUsed: turn, tokensUsed, finalResponse: lastResponse || void 0 };
+      return {
+        status: "completed",
+        turnsUsed: turn,
+        tokensUsed,
+        finalResponse: lastResponse || void 0
+      };
     }
     if (judgment.verdict === "skipped") {
       yield {
@@ -7584,7 +7611,12 @@ async function* runUntilImpl(agent, goal, options, deps) {
         status: "completed",
         reason: `skipped: ${judgment.reason}`
       };
-      return { status: "completed", turnsUsed: turn, tokensUsed, finalResponse: lastResponse || void 0 };
+      return {
+        status: "completed",
+        turnsUsed: turn,
+        tokensUsed,
+        finalResponse: lastResponse || void 0
+      };
     }
     if (tokenBudget !== void 0 && tokensUsed >= tokenBudget) {
       yield {
@@ -7592,7 +7624,12 @@ async function* runUntilImpl(agent, goal, options, deps) {
         status: "budget_limited",
         reason: `token budget (${tokenBudget}) reached: ${tokensUsed} used`
       };
-      return { status: "budget_limited", turnsUsed: turn, tokensUsed, finalResponse: lastResponse || void 0 };
+      return {
+        status: "budget_limited",
+        turnsUsed: turn,
+        tokensUsed,
+        finalResponse: lastResponse || void 0
+      };
     }
     yield { type: "continuation", turn, prompt: continuationPrompt };
   }
@@ -7601,7 +7638,12 @@ async function* runUntilImpl(agent, goal, options, deps) {
     status: "failed",
     reason: `max turns (${maxTurns}) exhausted`
   };
-  return { status: "failed", turnsUsed: turn, tokensUsed, finalResponse: lastResponse || void 0 };
+  return {
+    status: "failed",
+    turnsUsed: turn,
+    tokensUsed,
+    finalResponse: lastResponse || void 0
+  };
 }
 function composeContinuation(goal, lastResponse) {
   return [
@@ -22628,6 +22670,23 @@ async function updateJobStatus(jobId, enabled) {
   return updated;
 }
 
+// src/goal-loop.ts
+function runGoalLoop(agent, goal, options, depsOverride) {
+  async function* wrap() {
+    const { runUntilImpl: runUntilImpl2 } = await Promise.resolve().then(() => (init_run_until(), run_until_exports));
+    const deps = depsOverride ?? await (async () => {
+      const { judgeCallImpl: judgeCallImpl2 } = await Promise.resolve().then(() => (init_judge_call(), judge_call_exports));
+      const { getAgentFacade: getAgentFacade2 } = await Promise.resolve().then(() => (init_agent_factory_registry(), agent_factory_registry_exports));
+      const create = getAgentFacade2().create;
+      return {
+        judge: (ctx, opts) => judgeCallImpl2(ctx, opts, { create })
+      };
+    })();
+    return yield* runUntilImpl2(agent, goal, options, deps);
+  }
+  return wrap();
+}
+
 // src/define-provider.ts
 init_builtin();
 init_registry();
@@ -24502,6 +24561,7 @@ exports.mkMemoryId = mkMemoryId;
 exports.normalizeSchema = normalizeSchema;
 exports.normalizeUsage = normalizeUsage;
 exports.preflightCheck = preflightCheck;
+exports.runGoalLoop = runGoalLoop;
 exports.scopedConversationId = scopedConversationId;
 exports.sessionScopePrefix = sessionScopePrefix;
 exports.toShareGptTrajectory = toShareGptTrajectory;