@blade-hq/agent-kit 1.0.16 → 1.0.17

package/dist/{chunk-J3PQU7BQ.js → chunk-2LKBNP7B.js}

@@ -5,12 +5,12 @@ import {
 import {
   getSessionFilePath,
   resolveSessionFilePreviewTarget
-} from "./chunk-U4KVB5OB.js";
+} from "./chunk-CDOBBZKQ.js";
 import {
   formatToolName,
   getAuthedUrl,
   useUiStore
-} from "./chunk-QEYZZDLM.js";
+} from "./chunk-5TKSNSAW.js";
 import {
   cn,
   copyToClipboard
@@ -1933,4 +1933,4 @@ export {
   PlanSummaryCard,
   extractLatestPlanMessages
 };
-//# sourceMappingURL=chunk-J3PQU7BQ.js.map
+//# sourceMappingURL=chunk-2LKBNP7B.js.map

package/dist/{chunk-QEYZZDLM.js → chunk-5TKSNSAW.js}

@@ -3389,6 +3389,21 @@ function isGisGoalListPayload(value) {
 function isGisResourceListPayload(value) {
   return isRecord4(value) && Array.isArray(value.resources);
 }
+function sessionStatusFromChatEnd(status) {
+  if (status === "paused") return "waiting_for_input";
+  if (status === "interrupted") return "interrupted";
+  if (status === "failed") return "failed";
+  return "completed";
+}
+function extractTerminalChatEndStatus(events) {
+  for (let i = events.length - 1; i >= 0; i--) {
+    const event = events[i];
+    if (event?.type !== "chat:end") continue;
+    const status = event.payload?.status;
+    return typeof status === "string" ? status : "completed";
+  }
+  return null;
+}
 function isGisTargetListPayload(value) {
   return isRecord4(value) && Array.isArray(value.targets);
 }
@@ -3552,7 +3567,8 @@ var AgentSocket = class {
     s.on("turn:events", (data) => {
       const sessionId = this._resolveSessionId(data);
       if (!sessionId || !data.events?.length) return;
-      if (!useChatStore.getState().isStreaming[sessionId]) {
+      const terminalChatEndStatus = extractTerminalChatEndStatus(data.events);
+      if (!terminalChatEndStatus && !useChatStore.getState().isStreaming[sessionId]) {
         useSessionStore.getState().updateSessionStatus(sessionId, "running");
         useChatStore.getState().setStreaming(sessionId, true);
       }
@@ -3575,6 +3591,16 @@ var AgentSocket = class {
           notifyWorkspaceFilesChanged(sessionId);
         }
       }
+      if (terminalChatEndStatus) {
+        useSessionStore.getState().updateSessionStatus(sessionId, sessionStatusFromChatEnd(terminalChatEndStatus));
+        if (terminalChatEndStatus === "interrupted") {
+          useChatStore.getState().markInterrupted(sessionId);
+        } else if (terminalChatEndStatus === "failed") {
+          useChatStore.getState().markFailed(sessionId);
+        }
+        useChatStore.getState().setStreaming(sessionId, false);
+        invalidateContextStats(sessionId);
+      }
     });
     s.on("chat:start", (data) => {
       const sessionId = this._resolveSessionId(data);
@@ -3588,10 +3614,7 @@ var AgentSocket = class {
       if (sessionId) {
         this._flushPendingTurnPatches(sessionId);
         const status = data?.status ?? "completed";
-        useSessionStore.getState().updateSessionStatus(
-          sessionId,
-          status === "paused" ? "waiting_for_input" : status === "interrupted" ? "interrupted" : status === "failed" ? "failed" : "completed"
-        );
+        useSessionStore.getState().updateSessionStatus(sessionId, sessionStatusFromChatEnd(status));
         if (status === "interrupted") {
           useChatStore.getState().markInterrupted(sessionId);
         } else if (status === "failed") {
@@ -4482,4 +4505,4 @@ export {
   bootstrapBladeClient,
   getBootstrappedClient
 };
-//# sourceMappingURL=chunk-QEYZZDLM.js.map
+//# sourceMappingURL=chunk-5TKSNSAW.js.map