@theokit/sdk 4.18.0 → 4.18.1

package/dist/index.js

@@ -7529,16 +7529,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;
@@ -7573,7 +7595,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 {
@@ -7581,7 +7608,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 {
@@ -7589,7 +7621,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 };
   }
@@ -7598,7 +7635,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 [