@blade-hq/agent-kit 0.5.5 → 0.5.7

package/dist/{chunk-NEI66DW6.js → chunk-STCTXRMJ.js}

@@ -1,6 +1,6 @@
 import {
   useSessionStore
-} from "./chunk-HWOVPFWG.js";
+} from "./chunk-UM7G65GH.js";
 
 // src/react/hooks/use-session.ts
 function useSession() {
@@ -27,4 +27,4 @@ function useSession() {
 export {
   useSession
 };
-//# sourceMappingURL=chunk-NEI66DW6.js.map
+//# sourceMappingURL=chunk-STCTXRMJ.js.map

package/dist/{chunk-HWOVPFWG.js → chunk-UM7G65GH.js}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-J3XVFPOV.js";
 import {
   BladeClient
-} from "./chunk-Q6CSM5DE.js";
+} from "./chunk-ETHPRRT2.js";
 import {
   createClientActions,
   useCardStateStore
@@ -103,7 +103,8 @@ var useUiBridgeStore = create()((set, get) => ({
         {
           id: buildSignalId(),
           label: context.label,
-          content: context.content
+          content: context.content,
+          imageUrl: context.imageUrl
         }
       ]
     }
@@ -612,6 +613,7 @@ __export(sessions_exports, {
   listSessionsWithSkillData: () => listSessionsWithSkillData,
   pinSession: () => pinSession,
   previewImport: () => previewImport,
+  probeSessionUrl: () => probeSessionUrl,
   renameFile: () => renameFile,
   revokeShare: () => revokeShare,
   rewindSession: () => rewindSession,
@@ -658,6 +660,7 @@ var deleteSession = (...args) => r3().deleteSession(...args);
 var listBackgroundTasks = (...args) => r3().listBackgroundTasks(...args);
 var getBackgroundTask = (...args) => r3().getBackgroundTask(...args);
 var stopBackgroundTask = (...args) => r3().stopBackgroundTask(...args);
+var probeSessionUrl = (...args) => r3().probeSessionUrl(...args);
 var listDir = (...args) => r3().listDir(...args);
 var uploadFiles = (...args) => r3().uploadFiles(...args);
 var deleteFile = (...args) => r3().deleteFile(...args);
@@ -1235,8 +1238,16 @@ function projectionToMessage(turn) {
     ...turn.memory_refs?.length ? { memory_refs: turn.memory_refs } : {}
   };
 }
-function rebuildAgentLoops(turns) {
-  const messages = turns.map(projectionToMessage).filter(Boolean);
+var messageProjectionCache = /* @__PURE__ */ new WeakMap();
+function cachedProjectionToMessage(turn) {
+  if (messageProjectionCache.has(turn)) {
+    return messageProjectionCache.get(turn) ?? null;
+  }
+  const message = projectionToMessage(turn);
+  messageProjectionCache.set(turn, message);
+  return message;
+}
+function rebuildAgentLoops(turns, messages) {
   const childLoopNames = [...new Set(turns.map((turn) => turn.loop_id).filter((name) => name !== "root"))];
   if (childLoopNames.length === 0) return {};
   const agentToolCalls = messages.filter((message) => message.role === "assistant" && (message.loop_name ?? "root") === "root").flatMap((message) => message.tool_calls ?? []).filter((toolCall) => formatToolName(toolCall.name) === "Agent");
@@ -1276,7 +1287,8 @@ function applyPlanningSideEffects(sessionId, turns) {
   }
 }
 function materialize(turns) {
-  const messages = turns.map(projectionToMessage).filter((message) => message !== null).filter((message, index, allMessages) => {
+  const projectedMessages = turns.map(cachedProjectionToMessage).filter((message) => message !== null);
+  const messages = projectedMessages.filter((message, index, allMessages) => {
     if (!message.blocks?.some((block) => block.type === "ask_user_answer")) return true;
     const next = allMessages[index + 1];
     return !(next?.role === "user" && next.content === message.content);
@@ -1286,7 +1298,7 @@ function materialize(turns) {
   );
   return {
     messages,
-    agentLoops: rebuildAgentLoops(turns),
+    agentLoops: rebuildAgentLoops(turns, projectedMessages),
     activeCompaction: activeCompaction ? {
       turn_id: activeCompaction.turn_id,
       status: activeCompaction.status,
@@ -1306,9 +1318,28 @@ function materialize(turns) {
   };
 }
 var ERROR_ANCHOR_PREFIX = "error-anchor:";
+function orderTurns(turns) {
+  for (let index = 1; index < turns.length; index += 1) {
+    if (turns[index - 1].sequence > turns[index].sequence) {
+      return [...turns].sort((left, right) => left.sequence - right.sequence);
+    }
+  }
+  return turns;
+}
+function areAgentLoopsEqual(left, right) {
+  if (!left) return Object.keys(right).length === 0;
+  const leftEntries = Object.entries(left);
+  const rightKeys = Object.keys(right);
+  if (leftEntries.length !== rightKeys.length) return false;
+  return leftEntries.every(([key, value]) => {
+    const next = right[key];
+    return next !== void 0 && value.toolCallId === next.toolCallId && value.description === next.description && value.status === next.status;
+  });
+}
 function updateSessionState(state, sessionId, turns) {
-  const orderedTurns = [...turns].sort((left, right) => left.sequence - right.sequence);
+  const orderedTurns = orderTurns(turns);
   const { messages, agentLoops, activeCompaction } = materialize(orderedTurns);
+  const stableAgentLoops = areAgentLoopsEqual(state.agentLoops[sessionId], agentLoops) ? state.agentLoops[sessionId] : agentLoops;
   applyPlanningSideEffects(sessionId, orderedTurns);
   const lastTurnId = orderedTurns[orderedTurns.length - 1]?.turn_id ?? null;
   const preservedErrors = lastTurnId ? (state.messages[sessionId] ?? []).filter(
@@ -1318,7 +1349,7 @@ function updateSessionState(state, sessionId, turns) {
   return {
     turns: { ...state.turns, [sessionId]: orderedTurns },
     messages: { ...state.messages, [sessionId]: mergedMessages },
-    agentLoops: { ...state.agentLoops, [sessionId]: agentLoops },
+    agentLoops: { ...state.agentLoops, [sessionId]: stableAgentLoops ?? agentLoops },
     activeCompactions: { ...state.activeCompactions, [sessionId]: activeCompaction }
   };
 }
@@ -1917,11 +1948,18 @@ var useAuthStore = create6()(
       loading: false,
       error: null,
       logout: () => {
-        void logout().catch(() => {
-        });
-        set({ token: null, socketAuthToken: null, user: null, error: null });
         agentSocket?.disconnect();
         useSessionStore.getState().reset();
+        void logout().then(({ logout_url }) => {
+          set({ token: null, socketAuthToken: null, user: null, error: null });
+          if (logout_url) {
+            const redirect = encodeURIComponent(window.location.origin);
+            const sep = logout_url.includes("?") ? "&" : "?";
+            window.location.href = `${logout_url}${sep}redirect=${redirect}`;
+          }
+        }).catch(() => {
+          set({ token: null, socketAuthToken: null, user: null, error: null });
+        });
       },
       checkAuth: async () => {
         const token = get().token;
@@ -2347,6 +2385,9 @@ var AgentSocket = class {
   pendingReplayMessages = /* @__PURE__ */ new Map();
   pendingReplayModes = /* @__PURE__ */ new Map();
   oomNotifiedKeys = /* @__PURE__ */ new Set();
+  pendingTurnPatches = /* @__PURE__ */ new Map();
+  patchFlushHandle = null;
+  patchFlushCancel = null;
   _ensureConnected() {
     if (this.socket.connected || this.socket.active) {
       return;
@@ -2360,9 +2401,65 @@ var AgentSocket = class {
   _resolveSessionId(data) {
     return data?.session_id ?? this.subscribedSession;
   }
+  _queueTurnPatch(sessionId, patch) {
+    let sessionPatches = this.pendingTurnPatches.get(sessionId);
+    if (!sessionPatches) {
+      sessionPatches = /* @__PURE__ */ new Map();
+      this.pendingTurnPatches.set(sessionId, sessionPatches);
+    }
+    const current = sessionPatches.get(patch.turn_id);
+    if (!current || patch.sequence >= current.sequence) {
+      sessionPatches.set(patch.turn_id, patch);
+    }
+    this._scheduleTurnPatchFlush();
+  }
+  _scheduleTurnPatchFlush() {
+    if (this.patchFlushHandle != null) return;
+    const flush = () => {
+      this.patchFlushHandle = null;
+      this.patchFlushCancel = null;
+      this._flushPendingTurnPatches();
+    };
+    if (typeof window !== "undefined" && typeof window.requestAnimationFrame === "function") {
+      const handle2 = window.requestAnimationFrame(flush);
+      this.patchFlushHandle = handle2;
+      this.patchFlushCancel = () => window.cancelAnimationFrame(handle2);
+      return;
+    }
+    const handle = globalThis.setTimeout(flush, 33);
+    this.patchFlushHandle = handle;
+    this.patchFlushCancel = () => globalThis.clearTimeout(handle);
+  }
+  _flushPendingTurnPatches(sessionId) {
+    const sessions = sessionId === void 0 ? [...this.pendingTurnPatches.entries()] : this.pendingTurnPatches.has(sessionId) ? [[sessionId, this.pendingTurnPatches.get(sessionId)]] : [];
+    if (sessionId === void 0) {
+      this.pendingTurnPatches.clear();
+    } else {
+      this.pendingTurnPatches.delete(sessionId);
+    }
+    for (const [targetSessionId, patchesByTurn] of sessions) {
+      const patches = [...patchesByTurn.values()].sort((left, right) => left.sequence - right.sequence);
+      for (const patch of patches) {
+        useChatStore.getState().applyTurnPatch(targetSessionId, patch);
+        const turn = patch.data.turn;
+        if (!turn) continue;
+        const mode = extractModeFromBlocks3(turn.blocks);
+        if (mode === "planning" || mode === "executing") {
+          useSessionStore.getState().setMode(targetSessionId, mode);
+        }
+        this._applyProjectionSideEffects(targetSessionId, turn);
+      }
+    }
+    if (this.pendingTurnPatches.size === 0 && this.patchFlushCancel) {
+      this.patchFlushCancel();
+      this.patchFlushHandle = null;
+      this.patchFlushCancel = null;
+    }
+  }
   _bindStoreHandlers() {
     const s = this.socket;
     const applyTurn = (sessionId, turn) => {
+      this._flushPendingTurnPatches(sessionId);
       useChatStore.getState().upsertTurn(sessionId, turn);
       invalidateContextStats(sessionId);
       const mode = extractModeFromBlocks3(turn.blocks);
@@ -2379,17 +2476,10 @@ var AgentSocket = class {
     s.on("turn:patch", (data) => {
       const sessionId = this._resolveSessionId(data);
       if (!sessionId) return;
-      useChatStore.getState().applyTurnPatch(sessionId, data);
       if (data.data.workspace_changed) {
         notifyWorkspaceFilesChanged(sessionId);
       }
-      const turn = data.data.turn;
-      if (!turn) return;
-      const mode = extractModeFromBlocks3(turn.blocks);
-      if (mode === "planning" || mode === "executing") {
-        useSessionStore.getState().setMode(sessionId, mode);
-      }
-      this._applyProjectionSideEffects(sessionId, turn);
+      this._queueTurnPatch(sessionId, data);
     });
     s.on("turn:end", (data) => {
       const sessionId = this._resolveSessionId(data);
@@ -2405,6 +2495,7 @@ var AgentSocket = class {
     s.on("chat:end", (data) => {
       const sessionId = this._resolveSessionId(data);
       if (sessionId) {
+        this._flushPendingTurnPatches(sessionId);
         const status = data?.status ?? "completed";
         useSessionStore.getState().updateSessionStatus(
           sessionId,
@@ -2434,6 +2525,7 @@ var AgentSocket = class {
     s.on("system:error", (data) => {
       const sessionId = this._resolveSessionId(data);
       if (sessionId) {
+        this._flushPendingTurnPatches(sessionId);
         const message = isOomText(data.message) ? OOM_MESSAGE : data.message;
         useSessionStore.getState().updateSessionStatus(sessionId, "failed");
         useChatStore.getState().markFailed(sessionId);
@@ -2520,6 +2612,7 @@ var AgentSocket = class {
       invalidateSessionCheckpoints(sessionId);
     });
     const handleArtifact = (sessionId, data) => {
+      if (_isHiddenPath(data.file_path)) return;
       const type2 = resolveArtifactType(data.content_type);
       const filePath = `/api/sessions/${sessionId}/files/${encodeURIComponent(data.file_path)}`;
       const fileUrl = this.client.buildAuthedUrl(filePath);
@@ -2894,6 +2987,9 @@ function bridgeSocketEvents(client, _stores) {
   setAgentSocket(nextSocket);
   return nextSocket;
 }
+function _isHiddenPath(filePath) {
+  return filePath.split("/").some((seg) => seg.startsWith("."));
+}
 
 // src/react/stores/answer-callback-store.ts
 import { create as create11 } from "zustand";
@@ -3093,4 +3189,4 @@ export {
   bootstrapBladeClient,
   getBootstrappedClient
 };
-//# sourceMappingURL=chunk-HWOVPFWG.js.map
+//# sourceMappingURL=chunk-UM7G65GH.js.map