@letta-ai/letta-code 0.1.17 → 0.1.18

package/letta.js

@@ -6677,7 +6677,7 @@ var package_default;
 var init_package = __esm(() => {
   package_default = {
     name: "@letta-ai/letta-code",
-    version: "0.1.17",
+    version: "0.1.18",
     description: "Letta Code is a CLI tool for interacting with stateful Letta agents from the terminal.",
     type: "module",
     bin: {
@@ -35659,6 +35659,11 @@ async function drainStream(stream, buffers, refresh, abortSignal) {
   let stopReason = null;
   let lastRunId = null;
   let lastSeqId = null;
+  const resetToolState = () => {
+    toolCallId = null;
+    toolName = null;
+    toolArgs = null;
+  };
   for await (const chunk of stream) {
     if (abortSignal?.aborted) {
       stopReason = "cancelled";
@@ -35672,6 +35677,9 @@ async function drainStream(stream, buffers, refresh, abortSignal) {
     }
     if (chunk.message_type === "ping")
       continue;
+    if (chunk.message_type === "tool_return_message") {
+      resetToolState();
+    }
     if (chunk.message_type === "approval_request_message") {
       approvalRequestId = chunk.id;
     }
@@ -35679,8 +35687,7 @@ async function drainStream(stream, buffers, refresh, abortSignal) {
       const toolCall = chunk.tool_call || (Array.isArray(chunk.tool_calls) && chunk.tool_calls.length > 0 ? chunk.tool_calls[0] : null);
       if (toolCall?.tool_call_id) {
         if (toolCallId && toolCall.tool_call_id !== toolCallId) {
-          toolName = null;
-          toolArgs = null;
+          resetToolState();
         }
         toolCallId = toolCall.tool_call_id;
       }
@@ -35709,11 +35716,25 @@ async function drainStream(stream, buffers, refresh, abortSignal) {
   }
   markCurrentLineAsFinished(buffers);
   queueMicrotask(refresh);
-  const approval = toolCallId && toolName && toolArgs && approvalRequestId ? {
-    toolCallId,
-    toolName,
-    toolArgs
-  } : null;
+  let approval = null;
+  if (stopReason === "requires_approval") {
+    if (!toolCallId || !toolName || !toolArgs || !approvalRequestId) {
+      console.error("[drainStream] Incomplete approval state at end of turn:", {
+        hasToolCallId: !!toolCallId,
+        hasToolName: !!toolName,
+        hasToolArgs: !!toolArgs,
+        hasApprovalRequestId: !!approvalRequestId
+      });
+    } else {
+      approval = {
+        toolCallId,
+        toolName,
+        toolArgs
+      };
+    }
+    resetToolState();
+    approvalRequestId = null;
+  }
   const apiDurationMs = performance.now() - startTime;
   return { stopReason, approval, lastRunId, lastSeqId, apiDurationMs };
 }
@@ -45999,4 +46020,4 @@ Note: Flags should use double dashes for full names (e.g., --yolo, not -yolo)`);
 }
 main();
 
-//# debugId=A0E0449DEC4CAD6464756E2164756E21
+//# debugId=0F309FBC4BD78D6B64756E2164756E21

package/package.json

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