@letta-ai/letta-code 0.21.13 → 0.21.14

package/letta.js

@@ -3269,7 +3269,7 @@ var package_default;
 var init_package = __esm(() => {
   package_default = {
     name: "@letta-ai/letta-code",
-    version: "0.21.13",
+    version: "0.21.14",
     description: "Letta Code is a CLI tool for interacting with stateful Letta agents from the terminal.",
     type: "module",
     bin: {
@@ -74338,6 +74338,84 @@ var init_check_approval = __esm(() => {
   ];
 });
 
+// src/utils/createRelayedAbortController.ts
+function createRelayedAbortController(parentSignal) {
+  const controller = new AbortController;
+  if (!parentSignal) {
+    return {
+      controller,
+      signal: controller.signal,
+      cleanup: () => {}
+    };
+  }
+  if (parentSignal.aborted) {
+    controller.abort(parentSignal.reason);
+    return {
+      controller,
+      signal: controller.signal,
+      cleanup: () => {}
+    };
+  }
+  const relayAbort = () => {
+    controller.abort(parentSignal.reason);
+  };
+  parentSignal.addEventListener("abort", relayAbort, { once: true });
+  return {
+    controller,
+    signal: controller.signal,
+    cleanup: () => {
+      parentSignal.removeEventListener("abort", relayAbort);
+    }
+  };
+}
+
+// src/utils/streamAbortRelay.ts
+function composeCleanups(first, second) {
+  let cleaned = false;
+  return () => {
+    if (cleaned) {
+      return;
+    }
+    cleaned = true;
+    first();
+    second();
+  };
+}
+function createStreamAbortRelay(parentSignal) {
+  if (!parentSignal) {
+    return null;
+  }
+  const requestAbort = createRelayedAbortController(parentSignal);
+  let cleaned = false;
+  const cleanup = () => {
+    if (cleaned) {
+      return;
+    }
+    cleaned = true;
+    requestAbort.cleanup();
+  };
+  return {
+    signal: requestAbort.signal,
+    attach(stream2) {
+      const existingCleanup = streamAbortRelayCleanupByStream.get(stream2);
+      streamAbortRelayCleanupByStream.set(stream2, existingCleanup ? composeCleanups(existingCleanup, cleanup) : cleanup);
+    },
+    cleanup
+  };
+}
+function cleanupStreamAbortRelay(stream2) {
+  const cleanup = streamAbortRelayCleanupByStream.get(stream2);
+  if (!cleanup) {
+    return;
+  }
+  streamAbortRelayCleanupByStream.delete(stream2);
+  cleanup();
+}
+var streamAbortRelayCleanupByStream;
+var init_streamAbortRelay = __esm(() => {
+  streamAbortRelayCleanupByStream = new WeakMap;
+});
+
 // src/agent/approval-result-normalization.ts
 function normalizeToolReturnText(value) {
   if (typeof value === "string")
@@ -74582,19 +74660,23 @@ async function sendMessageStream(conversationId, messages, opts = { streamTokens
   const firstOtid = messages[0]?.otid;
   debugLog("send-message-stream", "request_start conversation_id=%s agent_id=%s messages=%s otid=%s stream_tokens=%s background=%s max_retries=%s", resolvedConversationId, opts.agentId ?? "none", messageSummary || "(empty)", firstOtid ?? "none", opts.streamTokens ?? true, opts.background ?? true, requestOptions.maxRetries ?? "default");
   let stream2;
+  const abortRelay = createStreamAbortRelay(requestOptions.signal);
   try {
     stream2 = await client.conversations.messages.create(resolvedConversationId, requestBody, {
       ...requestOptions,
+      ...abortRelay ? { signal: abortRelay.signal } : {},
       headers: {
         ...requestOptions.headers ?? {},
         ...extraHeaders
       }
     });
   } catch (error) {
+    abortRelay?.cleanup();
     debugWarn("send-message-stream", "request_error conversation_id=%s otid=%s status=%s error=%s", resolvedConversationId, firstOtid ?? "none", error?.status ?? "none", error instanceof Error ? error.message : String(error));
     throw error;
   }
   debugLog("send-message-stream", "request_ok conversation_id=%s otid=%s", resolvedConversationId, firstOtid ?? "none");
+  abortRelay?.attach(stream2);
   if (requestStartTime !== undefined) {
     streamRequestStartTimes.set(stream2, requestStartTime);
   }
@@ -74611,6 +74693,7 @@ async function sendMessageStream(conversationId, messages, opts = { streamTokens
 var streamRequestStartTimes, streamToolContextIds, streamRequestContexts;
 var init_message = __esm(async () => {
   init_debug();
+  init_streamAbortRelay();
   init_timing();
   init_approval_result_normalization();
   init_clientSkills();
@@ -76701,6 +76784,7 @@ async function drainStream(stream2, buffers, refresh, abortSignal, onFirstMessag
     if (abortSignal) {
       abortSignal.removeEventListener("abort", abortHandler);
     }
+    cleanupStreamAbortRelay(stream2);
     clearLastSDKDiagnostic();
   }
   if (!stopReason && streamProcessor.stopReason) {
@@ -76792,15 +76876,23 @@ async function drainStreamWithResume(stream2, buffers, refresh, abortSignal, onF
       const client = await lazyClient();
       buffers.commitGeneration = (buffers.commitGeneration || 0) + 1;
       buffers.interrupted = false;
-      const resumeStream = runIdSource === "otid" && streamOtid && streamRequestContext ? await client.conversations.messages.stream(streamRequestContext.resolvedConversationId, {
-        agent_id: streamRequestContext.conversationId === "default" ? streamRequestContext.agentId ?? undefined : undefined,
-        otid: streamOtid,
-        starting_after: result.lastSeqId ?? 0,
-        batch_size: 1000
-      }) : await client.runs.messages.stream(runIdToResume, {
-        starting_after: result.lastSeqId ?? 0,
-        batch_size: 1000
-      });
+      const resumeAbortRelay = createStreamAbortRelay(abortSignal);
+      let resumeStream;
+      try {
+        resumeStream = runIdSource === "otid" && streamOtid && streamRequestContext ? await client.conversations.messages.stream(streamRequestContext.resolvedConversationId, {
+          agent_id: streamRequestContext.conversationId === "default" ? streamRequestContext.agentId ?? undefined : undefined,
+          otid: streamOtid,
+          starting_after: result.lastSeqId ?? 0,
+          batch_size: 1000
+        }, resumeAbortRelay ? { signal: resumeAbortRelay.signal } : undefined) : await client.runs.messages.stream(runIdToResume, {
+          starting_after: result.lastSeqId ?? 0,
+          batch_size: 1000
+        }, resumeAbortRelay ? { signal: resumeAbortRelay.signal } : undefined);
+      } catch (resumeError) {
+        resumeAbortRelay?.cleanup();
+        throw resumeError;
+      }
+      resumeAbortRelay?.attach(resumeStream);
       const resumeResult = await drainStream(resumeStream, buffers, refresh, abortSignal, undefined, onChunkProcessed, contextTracker, seenSeqIdThreshold, true);
       if (resumeResult.stopReason !== "error") {
         debugWarn("stream", "[MID-STREAM RESUME] ✅ Success (runId=%s, stopReason=%s)", runIdToResume, resumeResult.stopReason);
@@ -76865,6 +76957,7 @@ var FALLBACK_RUN_DISCOVERY_TIMEOUT_MS = 5000;
 var init_stream = __esm(async () => {
   init_error();
   init_debug();
+  init_streamAbortRelay();
   init_timing();
   init_chunkLog();
   await __promiseAll([
@@ -80804,15 +80897,23 @@ async function sendMessageStreamWithRetry(conversationId, messages, opts, socket
         try {
           const client = await getClient();
           const messageOtid = messages.map((item) => item.otid).find((value) => typeof value === "string");
+          const resumeAbortRelay = createStreamAbortRelay(abortSignal);
           if (abortSignal?.aborted) {
             throw new Error("Cancelled by user");
           }
-          return await client.conversations.messages.stream(conversationId, {
-            agent_id: conversationId === "default" ? runtime.agentId ?? undefined : undefined,
-            otid: messageOtid ?? undefined,
-            starting_after: 0,
-            batch_size: 1000
-          });
+          try {
+            const resumeStream = await client.conversations.messages.stream(conversationId, {
+              agent_id: conversationId === "default" ? runtime.agentId ?? undefined : undefined,
+              otid: messageOtid ?? undefined,
+              starting_after: 0,
+              batch_size: 1000
+            }, resumeAbortRelay ? { signal: resumeAbortRelay.signal } : undefined);
+            resumeAbortRelay?.attach(resumeStream);
+            return resumeStream;
+          } catch (resumeError) {
+            resumeAbortRelay?.cleanup();
+            throw resumeError;
+          }
         } catch (resumeError) {
           if (abortSignal?.aborted) {
             throw new Error("Cancelled by user");
@@ -80927,15 +81028,23 @@ async function sendApprovalContinuationWithRetry(conversationId, messages, opts,
         try {
           const client = await getClient();
           const messageOtid = messages.map((item) => item.otid).find((value) => typeof value === "string");
+          const resumeAbortRelay = createStreamAbortRelay(abortSignal);
           if (abortSignal?.aborted) {
             throw new Error("Cancelled by user");
           }
-          return await client.conversations.messages.stream(conversationId, {
-            agent_id: conversationId === "default" ? runtime.agentId ?? undefined : undefined,
-            otid: messageOtid ?? undefined,
-            starting_after: 0,
-            batch_size: 1000
-          });
+          try {
+            const resumeStream = await client.conversations.messages.stream(conversationId, {
+              agent_id: conversationId === "default" ? runtime.agentId ?? undefined : undefined,
+              otid: messageOtid ?? undefined,
+              starting_after: 0,
+              batch_size: 1000
+            }, resumeAbortRelay ? { signal: resumeAbortRelay.signal } : undefined);
+            resumeAbortRelay?.attach(resumeStream);
+            return resumeStream;
+          } catch (resumeError) {
+            resumeAbortRelay?.cleanup();
+            throw resumeError;
+          }
         } catch (resumeError) {
           if (abortSignal?.aborted) {
             throw new Error("Cancelled by user");
@@ -80965,6 +81074,7 @@ var init_send = __esm(async () => {
   init_errorFormatter();
   init_diffPreview();
   init_interactivePolicy();
+  init_streamAbortRelay();
   init_constants2();
   init_cwd();
   init_permissionMode();
@@ -83809,6 +83919,19 @@ function runDetachedListenerTask(commandName, task2) {
     }
   });
 }
+async function replaySyncStateForRuntime(listenerRuntime, socket, scope, opts) {
+  const syncScopedRuntime = getOrCreateScopedRuntime(listenerRuntime, scope.agent_id, scope.conversation_id);
+  const recoverFn = opts?.recoverApprovalStateForSync ?? recoverApprovalStateForSync;
+  try {
+    await recoverFn(syncScopedRuntime, scope);
+  } catch (error) {
+    trackListenerError("listener_sync_recovery_failed", error, "listener_sync_recovery");
+    if (isDebugEnabled()) {
+      console.warn("[Listen] Sync approval recovery failed:", error);
+    }
+  }
+  emitStateSync(socket, listenerRuntime, scope);
+}
 function getParsedRuntimeScope(parsed) {
   if (!parsed || typeof parsed !== "object" || !("runtime" in parsed)) {
     return null;
@@ -84944,9 +85067,7 @@ async function connectWithRetry(runtime, opts, attempt = 0, startTime = Date.now
           console.log(`[Listen V2] Dropping sync: runtime mismatch or closed`);
           return;
         }
-        const syncScopedRuntime = getOrCreateScopedRuntime(runtime, parsed.runtime.agent_id, parsed.runtime.conversation_id);
-        await recoverApprovalStateForSync(syncScopedRuntime, parsed.runtime);
-        emitStateSync(socket, runtime, parsed.runtime);
+        await replaySyncStateForRuntime(runtime, socket, parsed.runtime);
         return;
       }
       if (parsed.type === "input") {
@@ -85911,6 +86032,7 @@ var init_client4 = __esm(async () => {
     handleCreateAgentCommand,
     handleReflectionSettingsCommand,
     scheduleQueuePump,
+    replaySyncStateForRuntime,
     recoverApprovalStateForSync,
     clearRecoveredApprovalStateForScope: (runtime, scope) => clearRecoveredApprovalStateForScope(asListenerRuntimeForTests(runtime), scope),
     emitStateSync
@@ -153424,4 +153546,4 @@ Error during initialization: ${message}`);
 }
 main();
 
-//# debugId=9A1A4B9B002D45C364756E2164756E21
+//# debugId=DEDBAA251791817164756E2164756E21

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@letta-ai/letta-code",
-  "version": "0.21.13",
+  "version": "0.21.14",
   "description": "Letta Code is a CLI tool for interacting with stateful Letta agents from the terminal.",
   "type": "module",
   "bin": {