@blade-hq/agent-kit 0.5.16 → 0.5.19

package/dist/{chunk-CZYQQABG.js → chunk-SPHQXYME.js}

@@ -1491,13 +1491,14 @@ function extractAskAnswers(turns) {
   }
   return answers;
 }
-function registerCreatedSessionState(set, session, mode = DEFAULT_SESSION_MODE) {
+function registerCreatedSessionState(set, session, mode = DEFAULT_SESSION_MODE, options = {}) {
+  const fresh = options.fresh ?? true;
   useChatStore.getState().setTurns(session.id, []);
   useTaskStore.getState().setTasks(session.id, []);
   set((state) => ({
     sessions: [session, ...state.sessions.filter((item) => item.id !== session.id)],
     modes: { ...state.modes, [session.id]: state.modes[session.id] ?? mode },
-    _freshSessions: new Set(state._freshSessions).add(session.id)
+    _freshSessions: fresh ? new Set(state._freshSessions).add(session.id) : new Set([...state._freshSessions].filter((id) => id !== session.id))
   }));
 }
 async function revalidateViewerSessions(existingSessions) {
@@ -1599,8 +1600,8 @@ var useSessionStore = create5()((set, get) => ({
     });
     return session_id;
   },
-  registerCreatedSession: (session, mode = DEFAULT_SESSION_MODE) => {
-    registerCreatedSessionState(set, session, mode);
+  registerCreatedSession: (session, mode = DEFAULT_SESSION_MODE, options = {}) => {
+    registerCreatedSessionState(set, session, mode, options);
   },
   upsertSession: (session) => {
     set((state) => ({
@@ -2270,6 +2271,9 @@ var AgentSocket = class {
   pendingReplayModes = /* @__PURE__ */ new Map();
   oomNotifiedKeys = /* @__PURE__ */ new Set();
   pendingTurnPatches = /* @__PURE__ */ new Map();
+  joinedSessions = /* @__PURE__ */ new Set();
+  requestedJoins = /* @__PURE__ */ new Set();
+  pendingJoins = /* @__PURE__ */ new Map();
   patchFlushHandle = null;
   patchFlushCancel = null;
   _ensureConnected() {
@@ -2392,6 +2396,7 @@ var AgentSocket = class {
         }
         useChatStore.getState().setStreaming(sessionId, false);
         invalidateContextStats(sessionId);
+        void this._syncSessionTurnsFromHistory(sessionId);
         useRuntimeStore.getState().addEvent(sessionId, {
           type: "chat:end",
           title: status === "paused" ? "Waiting for input" : status === "interrupted" ? "\u8FD0\u884C\u5DF2\u4E2D\u65AD" : status === "failed" ? "Run failed" : "Run completed",
@@ -2578,10 +2583,13 @@ var AgentSocket = class {
     this.socket.on("connect", () => {
       useConnectionStore.getState().markConnected();
       console.log("[agent-socket] connected, id:", this.socket.id);
+      this.joinedSessions.clear();
+      this.requestedJoins.clear();
+      this.pendingJoins.clear();
       const sessionId = this.subscribedSession;
       if (sessionId) {
         useSessionStore.getState().setActiveSession(sessionId);
-        this.socket.emit("session:subscribe", { session_id: sessionId });
+        void this._ensureJoined(sessionId);
       }
     });
     this.socket.on("connect_error", (err) => {
@@ -2687,6 +2695,14 @@ var AgentSocket = class {
       detail: OOM_MESSAGE
     });
   }
+  async _syncSessionTurnsFromHistory(sessionId) {
+    try {
+      const turns = await this.client.sessions.getSessionTurns(sessionId);
+      if (turns.length === 0) return;
+      useChatStore.getState().setTurns(sessionId, turns);
+    } catch {
+    }
+  }
   _applySystemNotification(sessionId, loopId, notification) {
     const notificationType = notification.notification_type ?? "system_notification";
     const title = notification.title ?? notificationType;
@@ -2728,7 +2744,7 @@ var AgentSocket = class {
       useUiStore.getState().clearArtifacts();
       useCardStateStore.getState().clearAllStates();
     }
-    this.socket.emit("session:subscribe", { session_id: sessionId });
+    void this._ensureJoined(sessionId);
   }
   unsubscribe() {
     const previousSessionId = this.subscribedSession;
@@ -2736,6 +2752,9 @@ var AgentSocket = class {
       this.socket.emit("session:unsubscribe", { session_id: previousSessionId });
       useUiBridgeStore.getState().clearSession(previousSessionId);
       this.subscribedSession = null;
+      this.requestedJoins.delete(previousSessionId);
+      this.joinedSessions.delete(previousSessionId);
+      this.pendingJoins.delete(previousSessionId);
     }
     this.previewArtifactsByToolCall.clear();
   }
@@ -2749,7 +2768,44 @@ var AgentSocket = class {
    * 幂等：后端 session:subscribe handler 只做 enter_room，重复 emit 无副作用。
    */
   attachSession(sessionId) {
-    this.socket.emit("session:subscribe", { session_id: sessionId });
+    void this._ensureJoined(sessionId);
+  }
+  _ensureJoined(sessionId) {
+    if (this.joinedSessions.has(sessionId)) {
+      return Promise.resolve();
+    }
+    this.requestedJoins.add(sessionId);
+    const pending = this.pendingJoins.get(sessionId);
+    if (pending) {
+      return pending;
+    }
+    const join = new Promise((resolve, reject) => {
+      this.socket.timeout(5e3).emit(
+        "session:subscribe",
+        { session_id: sessionId },
+        (err, response) => {
+          if (err) {
+            reject(err);
+            return;
+          }
+          if (response && response.ok === false) {
+            reject(new Error(response.message || "Session subscription failed"));
+            return;
+          }
+          if (this.requestedJoins.has(sessionId)) {
+            this.joinedSessions.add(sessionId);
+          }
+          resolve();
+        }
+      );
+    }).finally(() => {
+      this.pendingJoins.delete(sessionId);
+    });
+    this.pendingJoins.set(sessionId, join);
+    return join;
+  }
+  _shouldWaitForJoinBeforeSend(sessionId) {
+    return this.subscribedSession === sessionId || this.joinedSessions.has(sessionId) || this.pendingJoins.has(sessionId) || this.requestedJoins.has(sessionId);
   }
   send(sessionId, message, mode, askuserAnswer, extras) {
     this._ensureConnected();
@@ -2781,7 +2837,7 @@ var AgentSocket = class {
       this.pendingReplayMessages.set(sessionId, message);
       this.pendingReplayModes.set(sessionId, mode);
     }
-    this.socket.emit("chat:send", {
+    const payload = {
       session_id: sessionId,
       message,
       mode,
@@ -2789,7 +2845,19 @@ var AgentSocket = class {
       model: extras?.model,
       whatif: extras?.whatif,
       replay_decision: extras?.replay_decision
-    });
+    };
+    const sendPayload = () => {
+      this.socket.emit("chat:send", payload);
+    };
+    if (this._shouldWaitForJoinBeforeSend(sessionId)) {
+      void this._ensureJoined(sessionId).then(sendPayload).catch((error) => {
+        useChatStore.getState().setStreaming(sessionId, false);
+        useSessionStore.getState().updateSessionStatus(sessionId, "failed");
+        useChatStore.getState().addErrorMessage(sessionId, error.message || "\u4F1A\u8BDD\u8BA2\u9605\u5931\u8D25\uFF0C\u6D88\u606F\u672A\u53D1\u9001");
+      });
+    } else {
+      sendPayload();
+    }
     const skillId = extractSkillIdForAnalytics(message);
     trackEvent("message_sent", {
       session_id: sessionId,
@@ -3073,4 +3141,4 @@ export {
   bootstrapBladeClient,
   getBootstrappedClient
 };
-//# sourceMappingURL=chunk-CZYQQABG.js.map
+//# sourceMappingURL=chunk-SPHQXYME.js.map