@blade-hq/agent-client 1.2.0-beta.1 → 1.2.0-beta.2

package/dist/index.js

@@ -892,6 +892,11 @@ function toolStatusFromResult(result) {
 }
 
 // src/shared/projection/builder.ts
+function isConversationTurn(turn) {
+  return (turn.role === "user" || turn.role === "assistant") && (turn.kind ?? "message") === "message" && !turn.blocks.some(
+    (block) => block.type === "mode_change" || block.type === "plan_status" || block.type === "ask_user_answer"
+  );
+}
 var ClientProjectionBuilder = class {
   activeTurns = /* @__PURE__ */ new Map();
   lastTurnIds = /* @__PURE__ */ new Map();
@@ -948,6 +953,8 @@ var ClientProjectionBuilder = class {
         return this.onAskUserAnswer(loopId, payload);
       case "user:message":
         return this.onUserMessage(loopId, payload);
+      case "session:post_chat_followup":
+        return this.onPostChatFollowup(payload);
     }
     if (event.type.startsWith("compaction:"))
       return this.onCompaction(event.type, loopId, payload);
@@ -1005,6 +1012,18 @@ var ClientProjectionBuilder = class {
       this.seq = maxSeq;
     }
   }
+  _seedTurns(turns) {
+    const latestTurns = /* @__PURE__ */ new Map();
+    for (const turn of turns) {
+      this.seedSequence(turn.sequence);
+      if (!isConversationTurn(turn)) continue;
+      const current = latestTurns.get(turn.loop_id);
+      if (!current || turn.sequence >= current.sequence) {
+        latestTurns.set(turn.loop_id, turn);
+      }
+    }
+    this.latestTurns = latestTurns;
+  }
   // --- Event handlers ---
   onMemoryInject(loopId, payload) {
     const refs = payload.memory_refs;
@@ -1193,6 +1212,27 @@ var ClientProjectionBuilder = class {
       usage: null,
       duration_ms: 0
     };
+    this.latestTurns.set("root", turn);
+    return [{ kind: "upsert", turn }];
+  }
+  onPostChatFollowup(payload) {
+    const assistantEntryId = String(payload.assistant_entry_id ?? "");
+    const latest = this.latestTurns.get("root");
+    if (!latest || latest.role !== "assistant" || latest.turn_id !== assistantEntryId) {
+      return null;
+    }
+    const suggestions = payload.suggestions;
+    const postChatFollowup = Array.isArray(suggestions) && suggestions.length === 3 && suggestions.every((value) => typeof value === "string") ? {
+      assistant_entry_id: assistantEntryId,
+      recaption: String(payload.recaption ?? ""),
+      suggestions
+    } : null;
+    if (!postChatFollowup) return null;
+    const turn = {
+      ...latest,
+      post_chat_followup: postChatFollowup
+    };
+    this.latestTurns.set("root", turn);
     return [{ kind: "upsert", turn }];
   }
   // --- Delta / streaming events ---
@@ -1545,11 +1585,24 @@ function projectHistory(entries) {
   const toolOwners = /* @__PURE__ */ new Map();
   const latestAssistantByLoop = /* @__PURE__ */ new Map();
   const pendingChildPauses = /* @__PURE__ */ new Map();
+  let latestRootMessage = null;
   let sequence = 0;
   for (const entry of entries) {
     const kind = String(entry.kind ?? "");
     const loopId = String(entry.loop_name ?? "root");
     const id = String(entry.id ?? `history-${sequence + 1}`);
+    if (kind === "post_chat_followup" && isRecord3(entry.data)) {
+      const assistantEntryId = String(entry.data.assistant_entry_id ?? "");
+      const suggestions = entry.data.suggestions;
+      if (latestRootMessage?.role === "assistant" && latestRootMessage.turn_id === assistantEntryId && Array.isArray(suggestions) && suggestions.length === 3 && suggestions.every((value) => typeof value === "string")) {
+        latestRootMessage.post_chat_followup = {
+          assistant_entry_id: assistantEntryId,
+          recaption: String(entry.data.recaption ?? ""),
+          suggestions
+        };
+      }
+      continue;
+    }
     if (kind === "message" && isRecord3(entry.message)) {
       const message = entry.message;
       const role = String(message.role ?? "");
@@ -1586,6 +1639,9 @@ function projectHistory(entries) {
         duration_ms: numberOrZero(message._duration_ms),
         started_at: typeof entry.timestamp === "string" ? entry.timestamp : void 0
       };
+      if (loopId === "root" && (role === "user" || role === "assistant")) {
+        latestRootMessage = turn;
+      }
       turns.push(turn);
       if (role === "assistant") latestAssistantByLoop.set(loopId, turn);
       for (const toolCall of toolCalls) toolOwners.set(toolCall.id, turn);
@@ -3237,11 +3293,14 @@ var AgentSession = class _AgentSession {
   _handleTurnEvents(events) {
     if (!events.length) return;
     const terminalChatEndStatus = extractTerminalChatEndStatus(events);
-    if (!terminalChatEndStatus) {
+    const followupOnly = events.every(
+      (event) => event.type === "session:post_chat_followup"
+    );
+    if (!terminalChatEndStatus && !followupOnly) {
       this.markRunningIfIdle();
     }
     if (this.state.turns.length) {
-      this.projectionBuilder.seedSequence(Math.max(...this.state.turns.map((t) => t.sequence)));
+      this.projectionBuilder._seedTurns(this.state.turns);
     }
     const updates = this.projectionBuilder.processBatch(events);
     for (const update of updates) {
@@ -3894,7 +3953,7 @@ function resolveAuthToken(options) {
 
 // src/version.ts
 var SDK_NAME = "agent-client";
-var SDK_VERSION = true ? "1.2.0-beta.1" : "1.1.1";
+var SDK_VERSION = true ? "1.2.0-beta.2" : "1.1.1";
 
 // src/socket.ts
 function withSdkIdentity(auth) {