@prestyj/agent 5.3.0 → 5.4.0

package/dist/index.cjs

@@ -258,7 +258,7 @@ function closeIterator(iterator) {
 async function* agentLoop(messages, options) {
   const maxTurns = options.maxTurns ?? DEFAULT_MAX_TURNS;
   const maxContinuations = options.maxContinuations ?? 5;
-  const toolMap = new Map((options.tools ?? []).map((t) => [t.name, t]));
+  let toolMap = new Map((options.tools ?? []).map((t) => [t.name, t]));
   const totalUsage = { inputTokens: 0, outputTokens: 0 };
   let turn = 0;
   let hitMaxTurns = false;
@@ -280,6 +280,8 @@ async function* agentLoop(messages, options) {
   const MAX_OVERFLOW_COMPACTIONS = 2;
   const STALL_RETRIES_BEFORE_NON_STREAMING = 2;
   const STALL_DELAY_MS = 1e3;
+  const MIN_PARTIAL_PRESERVE_CHARS = 200;
+  const PARTIAL_CONTINUATION_PROMPT = "[Your previous response was cut off by a connection failure. The text above is what was already delivered to the user. Continue exactly from where it stopped \u2014 do not repeat or restart it.]";
   const OVERLOAD_BASE_DELAY_MS = 2e3;
   const OVERLOAD_MAX_DELAY_MS = 3e4;
   const STREAM_FIRST_EVENT_TIMEOUT_MS = 45e3;
@@ -298,6 +300,7 @@ async function* agentLoop(messages, options) {
     while (turn < maxTurns) {
       options.signal?.throwIfAborted();
       turn++;
+      toolMap = new Map((options.tools ?? []).map((t) => [t.name, t]));
       if (_diagFn) {
         let msgChars = 0;
         for (const m of messages) {
@@ -354,6 +357,7 @@ async function* agentLoop(messages, options) {
       let toolcallDeltaChars = 0;
       let toolcallDeltaCount = 0;
       let runawayDetected = null;
+      let attemptText = "";
       let lastYieldEndTime = Date.now();
       let maxConsumerLagMs = 0;
       const forwardAbort = () => streamController.abort();
@@ -479,6 +483,7 @@ async function* agentLoop(messages, options) {
             idleTimer = null;
           }
           if (event.type === "text_delta") {
+            attemptText += event.text;
             yield { type: "text_delta", text: event.text };
           } else if (event.type === "thinking_delta") {
             yield { type: "thinking_delta", text: event.text };
@@ -681,13 +686,23 @@ async function* agentLoop(messages, options) {
             });
           }
           const delayMs = Math.min(STALL_DELAY_MS * 2 ** (stallRetries - 1), 8e3);
+          let preservedChars = 0;
+          if (attemptText.length >= MIN_PARTIAL_PRESERVE_CHARS && toolcallDeltaCount === 0) {
+            messages.push({
+              role: "assistant",
+              content: [{ type: "text", text: attemptText }]
+            });
+            messages.push({ role: "user", content: PARTIAL_CONTINUATION_PROMPT });
+            preservedChars = attemptText.length;
+          }
           diag("retry", {
             reason: cause,
             attempt: stallRetries,
             maxAttempts: MAX_STALL_RETRIES,
             delayMs,
             events: streamEventCount,
-            nonStreaming: useNonStreamingFallback
+            nonStreaming: useNonStreamingFallback,
+            preservedChars
           });
           yield {
             type: "retry",
@@ -695,7 +710,8 @@ async function* agentLoop(messages, options) {
             attempt: stallRetries,
             maxAttempts: MAX_STALL_RETRIES,
             delayMs,
-            silent: stallRetries <= 2
+            silent: stallRetries <= 2,
+            ...preservedChars > 0 ? { preservedChars } : {}
           };
           await abortableSleep(delayMs, options.signal);
           turn--;