@blade-hq/agent-client 2608.0.6 → 2608.0.7-beta.0

package/dist/index.js

@@ -2,6 +2,9 @@
 function isAuthCallbackMessage(value) {
   return typeof value === "object" && value !== null && value.type === "blade-agent:sdk-auth";
 }
+var AUTH_ERROR_HINTS = {
+  pat_unavailable: "\u6CA1\u80FD\u4E3A\u4F60\u51C6\u5907\u8BBF\u95EE\u51ED\u636E\u3002\u8BF7\u9000\u51FA\u540E\u91CD\u65B0\u767B\u5F55 Blade \u518D\u8BD5\u4E00\u6B21\uFF1B\u82E5\u4ECD\u7136\u5931\u8D25\uFF0C\u8BF7\u8054\u7CFB\u7BA1\u7406\u5458\u3002"
+};
 function randomState() {
   const bytes = new Uint8Array(24);
   crypto.getRandomValues(bytes);
@@ -79,18 +82,22 @@ async function loginWithPopup(client, options = {}) {
       if (event.data.state !== state) return;
       const message = event.data;
       stopCloseWatch();
-      try {
-        popup.close();
-      } catch {
-      }
+      const closePopup = () => {
+        try {
+          popup.close();
+        } catch {
+        }
+      };
       if (message.status === "denied") {
+        closePopup();
         settle(() => reject(new Error("\u4F60\u53D6\u6D88\u4E86\u6388\u6743\uFF0C\u672A\u5B8C\u6210\u767B\u5F55\u3002")));
         return;
       }
       if (message.status !== "success" || !message.code) {
-        settle(() => reject(new Error(message.error || "\u767B\u5F55\u5931\u8D25\uFF0C\u672A\u53D6\u5F97\u6388\u6743\u7801\u3002")));
+        settle(() => reject(new Error(AUTH_ERROR_HINTS[message.error ?? ""] ?? message.error ?? "\u767B\u5F55\u5931\u8D25\uFF0C\u672A\u53D6\u5F97\u6388\u6743\u7801\u3002")));
         return;
       }
+      closePopup();
       client.json("POST", "/api/auth/sdk/token", {
         code: message.code,
         state,
@@ -174,6 +181,24 @@ var AuthResource = class {
   login(options) {
     return this.client.login(options);
   }
+  /**
+   * 用一次性 code 换 PAT。给**服务端**用：浏览器把 code 交给你自己的后端，
+   * 后端换到的 PAT 不进浏览器。浏览器里直接用 `login()` 就够了。
+   *
+   * 服务端调用时会自己补上 `Origin` 头（浏览器会自动带，服务端不会，缺了
+   * 后端报 `missing_request_origin`）。
+   */
+  exchangeCode(params) {
+    return this.client.jsonFromInit("/api/auth/sdk/token", {
+      method: "POST",
+      headers: { Origin: params.clientOrigin },
+      body: JSON.stringify({
+        code: params.code,
+        state: params.state,
+        client_origin: params.clientOrigin
+      })
+    });
+  }
   /** 清除 login() 存储的令牌。 */
   logoutToken() {
     this.client.logoutToken();
@@ -325,7 +350,11 @@ var HeadlessResource = class {
       socket.on("chat:start", onChatStart);
       socket.on("chat:end", onChatEnd);
       socket.on("system:error", onSystemError);
-      socket.emit("session:subscribe", { session_id, event_format: "raw" });
+      socket.emit("session:subscribe", {
+        session_id,
+        event_format: "raw",
+        stream_tokens: this.client.options.streamTokens !== false
+      });
       socket.emit(
         "chat:send",
         {
@@ -407,6 +436,47 @@ var ModelsResource = class {
   }
 };
 
+// src/schemas/session.ts
+import { type } from "arktype";
+var SessionStatus = type(
+  "'created' | 'running' | 'completed' | 'failed' | 'interrupted' | 'waiting_for_input'"
+);
+var DEFAULT_REPLAY_SPEED = 5;
+var SessionInfo = type({
+  id: "string",
+  intent: "string",
+  status: SessionStatus,
+  created_at: "string",
+  updated_at: "string",
+  "shared?": "boolean",
+  "memory_enabled?": "boolean",
+  "is_persistent?": "boolean",
+  "is_headless?": "boolean",
+  "viewer_role?": "'owner' | 'viewer'",
+  "template_id?": "string | null",
+  "model?": "string | null",
+  "enable_thinking?": "boolean | null",
+  "solution_id?": "string | null",
+  "solution_ref?": "unknown",
+  "biz_role_id?": "string | null",
+  "solution?": "unknown",
+  "plan_summary?": "string | null",
+  "primary_skill_id?": "string | null",
+  // ship-attack v2：后端 engine.set_bound_skill 写入的"绑定 skill id"。
+  // 未绑定 / 非 ship-attack session 为 null。
+  "bound_skill_id?": "string | null",
+  "replay_state?": "unknown",
+  "is_pinned?": "boolean",
+  "pinned_at?": "string | null",
+  "is_example?": "boolean",
+  "ports?": "unknown",
+  "disable_tools?": "string[]",
+  "runtime_type?": "string | null",
+  "daemon_id?": "string | null",
+  "workspace_path?": "string | null",
+  "match?": "unknown"
+});
+
 // src/session/definition.ts
 var SessionSetupError = class extends Error {
   constructor(message, stage, sessionId, options) {
@@ -827,6 +897,28 @@ function sourceLoopFor(loopId, loopDescriptions) {
   if (!description) return null;
   return { loop_name: loopId, description };
 }
+var MAX_FINAL_ARTIFACTS = 20;
+function normalizeFinalArtifacts(raw) {
+  if (!Array.isArray(raw)) return [];
+  const artifacts = [];
+  const seen = /* @__PURE__ */ new Set();
+  for (const item of raw) {
+    if (typeof item !== "object" || item === null) continue;
+    const record = item;
+    const target = String(record.target ?? "").trim();
+    if (!target || seen.has(target)) continue;
+    const isLink = /^https?:\/\//i.test(target);
+    if (!isLink && target.split("/").includes("..")) continue;
+    seen.add(target);
+    artifacts.push({
+      kind: isLink ? "link" : "file",
+      target,
+      label: String(record.label ?? "").trim()
+    });
+    if (artifacts.length === MAX_FINAL_ARTIFACTS) break;
+  }
+  return artifacts;
+}
 function normalizePostChatFollowup(data, assistantEntryId) {
   const raw = data.suggestions;
   const suggestions = [];
@@ -838,11 +930,17 @@ function normalizePostChatFollowup(data, assistantEntryId) {
     }
   }
   const recaption = String(data.recaption ?? "");
-  if (suggestions.length === 0 && !recaption) return null;
+  const finalArtifacts = normalizeFinalArtifacts(data.final_artifacts);
+  const feedbackEligible = data.feedback_eligible === true;
+  if (suggestions.length === 0 && !recaption && finalArtifacts.length === 0 && !feedbackEligible)
+    return null;
   return {
     assistant_entry_id: assistantEntryId,
     recaption,
-    suggestions: suggestions.slice(0, 3)
+    suggestions: suggestions.slice(0, 3),
+    final_artifacts: finalArtifacts,
+    feedback_eligible: feedbackEligible,
+    feedback_scope: "latest_result"
   };
 }
 function normalizeChildPause(pausePayload) {
@@ -926,11 +1024,60 @@ function toolStatusFromResult(result) {
 }
 
 // src/shared/projection/builder.ts
+function hasChatRunEvent(events) {
+  return events.some(
+    (event) => event.type !== "session:post_chat_followup" && !event.type.startsWith("compaction:")
+  );
+}
+function acceptedPostChatFollowupCompletesLatestRun(events, updates) {
+  const acceptedAssistantEntryIds = new Set(
+    updates.flatMap((update) => {
+      if (update.kind !== "upsert") return [];
+      const followup = update.turn.post_chat_followup;
+      return followup?.assistant_entry_id ? [followup.assistant_entry_id] : [];
+    })
+  );
+  for (let index = events.length - 1; index >= 0; index -= 1) {
+    const event = events[index];
+    if (event?.type !== "session:post_chat_followup") continue;
+    const assistantEntryId = String(event.payload.assistant_entry_id ?? "");
+    return acceptedAssistantEntryIds.has(assistantEntryId) && !hasChatRunEvent(events.slice(index + 1));
+  }
+  return false;
+}
 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"
   );
 }
+function completedResponseText(content) {
+  if (typeof content === "string") return content;
+  if (!Array.isArray(content)) return content == null ? "" : String(content);
+  return content.filter(
+    (part) => typeof part === "object" && part !== null && !Array.isArray(part)
+  ).filter((part) => part.type === "text").map((part) => String(part.text ?? "")).join("");
+}
+function replaceTextContent(state, content) {
+  if (state.blocks.some((block) => block.type === "text")) return;
+  const replacement = { type: "text", content };
+  const blocks = [];
+  let inserted = false;
+  for (const block of state.blocks) {
+    if (block.type === "text") {
+      if (content && !inserted) {
+        blocks.push(replacement);
+        inserted = true;
+      }
+      continue;
+    }
+    blocks.push(block);
+  }
+  if (content && !inserted) {
+    const toolIndex = blocks.findIndex((block) => block.type === "tool_use");
+    blocks.splice(toolIndex < 0 ? blocks.length : toolIndex, 0, replacement);
+  }
+  state.blocks = blocks;
+}
 var ClientProjectionBuilder = class {
   activeTurns = /* @__PURE__ */ new Map();
   lastTurnIds = /* @__PURE__ */ new Map();
@@ -1013,6 +1160,9 @@ var ClientProjectionBuilder = class {
         return this.onDelta(state, "thinking", "", String(payload.content ?? ""));
       case "llm:content:delta":
         return this.onDelta(state, "text", "", String(payload.content ?? ""));
+      case "llm:content:complete":
+        replaceTextContent(state, String(payload.content ?? ""));
+        return this.syncPatch(state);
       case "llm:tool_call:created":
         return this.onToolCallCreated(state, payload);
       case "llm:response:done":
@@ -1302,6 +1452,9 @@ var ClientProjectionBuilder = class {
     if (typeof payload.usage === "object" && payload.usage !== null) {
       state.usage = { ...payload.usage };
     }
+    if ("content" in payload) {
+      replaceTextContent(state, completedResponseText(payload.content));
+    }
     const toolCalls = payload.tool_calls;
     if (Array.isArray(toolCalls)) {
       for (const raw of toolCalls) {
@@ -2105,6 +2258,11 @@ var SessionsResource = class {
   listSessionsWithSkillData() {
     return this.client.json("GET", "/api/sessions/with-skill-data");
   }
+  searchOwnedContent(params) {
+    const searchParams = new URLSearchParams({ q: params.q.trim() });
+    if (params.limit !== void 0) searchParams.set("limit", String(params.limit));
+    return this.client.json("GET", `/api/search?${searchParams.toString()}`);
+  }
   createSession(intent, template_id, primary_skill_id) {
     return this.createSessionWithRequest({ intent, template_id, primary_skill_id });
   }
@@ -2130,12 +2288,16 @@ var SessionsResource = class {
   pinSession(sessionId, pinned) {
     return this.client.json("PATCH", `/api/sessions/${sessionId}/pin`, { pinned });
   }
-  startReplaySession(sourceSessionId, speed = 5) {
+  startReplaySession(sourceSessionId, speed = DEFAULT_REPLAY_SPEED) {
     return this.client.json("POST", `/api/sessions/${sourceSessionId}/replay`, { speed });
   }
   updateReplaySession(sessionId, payload) {
     return this.client.json("PATCH", `/api/sessions/${sessionId}/replay`, payload);
   }
+  /** 查询会话能否作为回放来源；不支持时 reason 可直接展示，用来提前禁用入口。 */
+  getReplayPreview(sessionId) {
+    return this.client.json("GET", `/api/sessions/${sessionId}/replay/preview`);
+  }
   updateSharing(sessionId, shared) {
     return this.client.json("PATCH", `/api/sessions/${sessionId}/sharing`, { shared });
   }
@@ -2167,6 +2329,16 @@ var SessionsResource = class {
     );
     return projectHistory(entries);
   }
+  listResultFeedback(sessionId) {
+    return this.client.json("GET", `/api/sessions/${encodeURIComponent(sessionId)}/feedback`);
+  }
+  putResultFeedback(sessionId, assistantEntryId, feedback) {
+    return this.client.json(
+      "PUT",
+      `/api/sessions/${encodeURIComponent(sessionId)}/turns/${encodeURIComponent(assistantEntryId)}/feedback`,
+      feedback
+    );
+  }
   getSessionContextStats(sessionId, init) {
     return this.client.jsonFromInit(`/api/sessions/${sessionId}/context-stats`, init);
   }
@@ -2589,10 +2761,15 @@ function contentPreview(content, maxLen = 80) {
   const text = getTextContent(content).trim();
   return text.length > maxLen ? `${text.slice(0, maxLen)}\u2026` : text;
 }
-function transformSlashCommand(skillName, rawInput) {
+function transformSlashCommand(skillName, rawInput, { local = true, installed = true } = {}) {
   const prompt = rawInput.replace(`<skill>${skillName}</skill>`, "").trim();
-  return prompt ? `\u8BF7\u4F7F\u7528 ${skillName} skill \u5B8C\u6210\u4EFB\u52A1
+  if (local) {
+    return prompt ? `\u8BF7\u4F7F\u7528 ${skillName} skill \u5B8C\u6210\u4EFB\u52A1
 ${prompt}` : `\u8BF7\u4F7F\u7528 ${skillName} skill \u5B8C\u6210\u4EFB\u52A1`;
+  }
+  const steps = installed ? `\u8BF7\u5148\u8FD0\u884C blade hub skill use ${skillName} \u52A0\u8F7D\u6280\u80FD` : `\u8BE5\u6280\u80FD\u5C1A\u672A\u5B89\u88C5\u3002\u8BF7\u5148\u8FD0\u884C blade hub skill install ${skillName} \u5B89\u88C5\uFF0C\u518D\u8FD0\u884C blade hub skill use ${skillName} \u52A0\u8F7D`;
+  return prompt ? `${steps}\uFF0C\u7136\u540E\u5B8C\u6210\u4EFB\u52A1\uFF1A
+${prompt}` : `${steps}\uFF0C\u7136\u540E\u4F7F\u7528\u5B83\u5B8C\u6210\u4EFB\u52A1`;
 }
 var ATTACHMENT_TAG_RE = /\[附件:\s*([^\]]+)\](?:[ \t]*已上传到工作区路径:[ \t]*([^\r\n]+))?/g;
 var CONTEXT_PART_RE = /^\[上下文:\s*([^\]]+)\]\n([\s\S]*)$/;
@@ -2678,6 +2855,14 @@ function groupMessagesByLoop(messages) {
 }
 
 // src/session/state.ts
+function toReplaySnapshot(raw) {
+  if (!raw || !raw.status) return null;
+  return {
+    isReplay: raw.status === "replay",
+    speed: raw.speed ?? 1,
+    sourceSessionId: raw.source_session_id ?? null
+  };
+}
 function createInitialSessionState(sessionId) {
   return {
     sessionId,
@@ -2690,7 +2875,9 @@ function createInitialSessionState(sessionId) {
     errorMessage: null,
     askAnswers: {},
     agentLoops: {},
-    activeCompaction: null
+    activeCompaction: null,
+    replay: null,
+    viewerRole: null
   };
 }
 function parseAgentDescription(argumentsJson) {
@@ -2953,6 +3140,25 @@ function orderTurns(turns) {
   }
   return turns;
 }
+function reconcileOptimisticUserTurns(currentTurns, incomingTurns) {
+  const currentById = new Map(currentTurns.map((turn) => [turn.turn_id, turn]));
+  const optimisticUserTurn = currentTurns.find(
+    (turn) => turn.role === "user" && turn.turn_id.startsWith("local-user-")
+  );
+  let optimisticUserMatched = false;
+  return incomingTurns.map((turn) => {
+    const current = currentById.get(turn.turn_id);
+    if (current) {
+      const currentRenderId = current.client_render_id;
+      return turn.client_render_id || !currentRenderId ? turn : { ...turn, client_render_id: currentRenderId };
+    }
+    if (!optimisticUserMatched && optimisticUserTurn && turn.role === "user" && !turn.turn_id.startsWith("local-user-")) {
+      optimisticUserMatched = true;
+      return { ...turn, client_render_id: optimisticUserTurn.turn_id };
+    }
+    return turn;
+  });
+}
 function areAgentLoopsEqual(left, right) {
   if (!left) return Object.keys(right).length === 0;
   const leftEntries = Object.entries(left);
@@ -2964,7 +3170,7 @@ function areAgentLoopsEqual(left, right) {
   });
 }
 function withTurns(state, turns) {
-  const orderedTurns = orderTurns(turns);
+  const orderedTurns = orderTurns(reconcileOptimisticUserTurns(state.turns, turns));
   const { messages, agentLoops, activeCompaction } = materialize(orderedTurns);
   const isWaitingForInput = orderedTurns.some(
     (turn) => turn.tool_calls.some((toolCall) => toolCall.status === "awaiting_answer")
@@ -2974,7 +3180,13 @@ function withTurns(state, turns) {
   const preservedErrors = lastTurnId ? state.messages.filter(
     (m) => m.role === "error" && typeof m.entry_id === "string" && m.entry_id.startsWith(`${ERROR_ANCHOR_PREFIX}${lastTurnId}:`)
   ) : [];
-  const mergedMessages = preservedErrors.length > 0 ? [...messages, ...preservedErrors] : messages;
+  const turnById = new Map(orderedTurns.map((turn) => [turn.turn_id, turn]));
+  const messagesWithRenderIds = messages.map((message) => {
+    if (!message.entry_id) return message;
+    const renderId = turnById.get(message.entry_id)?.client_render_id;
+    return renderId ? { ...message, render_id: renderId } : message;
+  });
+  const mergedMessages = preservedErrors.length > 0 ? [...messagesWithRenderIds, ...preservedErrors] : messagesWithRenderIds;
   const latestMode = [...orderedTurns].reverse().map((turn) => extractModeFromBlocks(turn.blocks)).find((mode) => mode !== null);
   return {
     ...state,
@@ -3009,14 +3221,17 @@ function addUserMessage(state, content) {
 }
 function upsertTurn(state, turn) {
   let existing = [...state.turns];
-  if (turn.role === "user" && !turn.turn_id.startsWith("local-user-")) {
-    existing = existing.filter((t) => !t.turn_id.startsWith("local-user-"));
+  const [nextTurn] = reconcileOptimisticUserTurns(existing, [turn]);
+  if (nextTurn.client_render_id && turn.role === "user") {
+    existing = existing.filter(
+      (item) => !(item.role === "user" && item.turn_id.startsWith("local-user-"))
+    );
   }
-  const index = existing.findIndex((item) => item.turn_id === turn.turn_id);
+  const index = existing.findIndex((item) => item.turn_id === nextTurn.turn_id);
   if (index >= 0) {
-    existing[index] = turn;
+    existing[index] = nextTurn;
   } else {
-    existing.push(turn);
+    existing.push(nextTurn);
   }
   return withTurns(state, existing);
 }
@@ -3120,7 +3335,7 @@ function extractTerminalChatEndStatus(events) {
   }
   return null;
 }
-function hasChatRunEvent(events) {
+function hasChatRunEvent2(events) {
   return events.some(
     (event) => event.type !== "session:post_chat_followup" && !event.type.startsWith("compaction:")
   );
@@ -3141,6 +3356,11 @@ var AgentSession = class _AgentSession {
   patchFlushCancel = null;
   joined = false;
   joinPending = null;
+  // 回放控制请求排成一条队列。连着点"5x"再点"退出回放"时，并发发出去的两个
+  // 请求响应可能乱序回来，晚到的旧值会把已经退出的会话又显示成回放中；用
+  // 序号丢弃过期响应又会在失败时留下中间态。串行执行就没有这些情况：每个
+  // 响应都是当时最新的，失败也只影响自己。
+  replayQueue = Promise.resolve();
   pendingReplayMessage = null;
   pendingReplayMode;
   recentChatEndAt = 0;
@@ -3167,10 +3387,15 @@ var AgentSession = class _AgentSession {
     const session = new _AgentSession(
       sessionId,
       {
+        streamTokens: false,
         emitEvent: readonlyError,
         emitWithAck: async () => readonlyError(),
         ensureConnected: readonlyError,
         fetchTurns: async () => turns,
+        // 静态历史没有回放身份可言，给个空详情即可，不要抛错——_bootstrap
+        // 不会跑到这里，但保持只读语义一致。
+        fetchSessionInfo: async () => ({ replay_state: null }),
+        updateReplay: async () => readonlyError(),
         listDir: async () => readonlyError(),
         countFiles: async () => readonlyError(),
         downloadFile: async () => readonlyError(),
@@ -3462,10 +3687,13 @@ var AgentSession = class _AgentSession {
     this.joinPending = null;
     try {
       await this.ensureJoined();
-      await this.syncTurnsFromHistory({
-        clearOnEmpty: true,
-        preserveChangesSince: turnsBeforeRejoin
-      });
+      await Promise.all([
+        this.syncTurnsFromHistory({
+          clearOnEmpty: true,
+          preserveChangesSince: turnsBeforeRejoin
+        }),
+        this.syncReplayFromSession()
+      ]);
     } catch {
     }
   }
@@ -3473,6 +3701,40 @@ var AgentSession = class _AgentSession {
   async _bootstrap() {
     await this.syncTurnsFromHistory({ suppressErrors: false });
     await this.ensureJoined();
+    await this.syncReplayFromSession();
+  }
+  /** 改回放是 owner 专属（服务端 PATCH 走 _get_owned_session）。 */
+  get canControlReplay() {
+    return this.state.viewerRole !== "viewer";
+  }
+  /** 拉一次会话详情，填回放状态和只读身份；失败按非回放会话处理，不挡启动。 */
+  async syncReplayFromSession() {
+    try {
+      const info = await this.runtime.fetchSessionInfo(this.sessionId);
+      const replay = toReplaySnapshot(info.replay_state);
+      const viewerRole = info.viewer_role ?? null;
+      this.update((s) => ({ ...s, replay, viewerRole }));
+    } catch {
+    }
+  }
+  /** 调整回放倍速；不是回放会话、或只读身份时什么都不做。 */
+  async setReplaySpeed(speed) {
+    if (!this.state.replay?.isReplay || !this.canControlReplay) return;
+    await this.applyReplayControl({ speed });
+  }
+  /** 退出回放，之后的对话交给真实模型运行；不是回放会话、或只读身份时什么都不做。 */
+  async exitReplay() {
+    if (!this.state.replay?.isReplay || !this.canControlReplay) return;
+    await this.applyReplayControl({ status: "autonomous" });
+  }
+  /** 把一次回放控制请求排进队列；前一个完成后才发出，响应直接安装。 */
+  applyReplayControl(payload) {
+    const run = this.replayQueue.then(async () => {
+      const result = await this.runtime.updateReplay(this.sessionId, payload);
+      this.update((s) => ({ ...s, replay: toReplaySnapshot(result) }));
+    });
+    this.replayQueue = run.catch(() => void 0);
+    return run;
   }
   /** @internal */
   _handleTurnStart(turn) {
@@ -3499,7 +3761,7 @@ var AgentSession = class _AgentSession {
   _handleTurnEvents(events) {
     if (!events.length) return;
     const terminalChatEndStatus = extractTerminalChatEndStatus(events);
-    if (!terminalChatEndStatus && hasChatRunEvent(events)) {
+    if (!terminalChatEndStatus && hasChatRunEvent2(events)) {
       this.markRunningIfIdle();
     }
     if (this.state.turns.length) {
@@ -3513,8 +3775,8 @@ var AgentSession = class _AgentSession {
         this.emitter.emit("workspaceChanged", {});
       }
     }
-    if (terminalChatEndStatus) {
-      this.finishStreaming(terminalChatEndStatus);
+    if (terminalChatEndStatus || acceptedPostChatFollowupCompletesLatestRun(events, updates)) {
+      this.completeChat(terminalChatEndStatus ?? "completed");
     }
   }
   /** @internal */
@@ -3524,10 +3786,7 @@ var AgentSession = class _AgentSession {
   /** @internal */
   _handleChatEnd(status) {
     this.flushPatches();
-    this.finishStreaming(status);
-    this.recentChatEndAt = Date.now();
-    void this.syncTurnsFromHistory();
-    this.emitter.emit("chatEnd", { status });
+    this.completeChat(status);
   }
   /** @internal */
   _handleSystemError(message) {
@@ -3552,6 +3811,10 @@ var AgentSession = class _AgentSession {
     if (data.status) {
       this.update((s) => ({ ...s, status: data.status ?? s.status }));
     }
+    if (data.replay_state !== void 0) {
+      const replay = toReplaySnapshot(data.replay_state);
+      this.update((s) => ({ ...s, replay }));
+    }
     this.emitter.emit("sessionUpdated", {
       intent: data.intent,
       status: data.status,
@@ -3667,6 +3930,12 @@ var AgentSession = class _AgentSession {
       return next;
     });
   }
+  completeChat(status) {
+    this.finishStreaming(status);
+    this.recentChatEndAt = Date.now();
+    void this.syncTurnsFromHistory();
+    this.emitter.emit("chatEnd", { status });
+  }
   applyTurn(turn) {
     this.flushPatches();
     const previous = this.state;
@@ -3849,7 +4118,12 @@ var AgentSession = class _AgentSession {
     if (this.joinPending) return this.joinPending;
     const join = this.runtime.emitWithAck(
       "session:subscribe",
-      { session_id: this.sessionId, event_format: "raw", resume: true },
+      {
+        session_id: this.sessionId,
+        event_format: "raw",
+        resume: true,
+        stream_tokens: this.runtime.streamTokens
+      },
       5e3
     ).then((response) => {
       if (response && response.ok === false) {
@@ -3985,6 +4259,7 @@ var SessionHub = class {
   }
   runtimeFor() {
     return {
+      streamTokens: this.client.options?.streamTokens !== false,
       emitEvent: (event, data) => {
         ;
         this.client.socket().emit(event, data);
@@ -4001,6 +4276,8 @@ var SessionHub = class {
       }),
       ensureConnected: () => this.ensureConnected(),
       fetchTurns: (sessionId) => this.client.sessions.getSessionTurns(sessionId),
+      fetchSessionInfo: (sessionId) => this.client.sessions.getSession(sessionId),
+      updateReplay: (sessionId, payload) => this.client.sessions.updateReplaySession(sessionId, payload).then((result) => result.replay_state ?? null),
       listDir: (sessionId, dirPath) => this.client.sessions.listDir(sessionId, dirPath),
       countFiles: (sessionId, dirPath) => this.client.sessions.countFiles(sessionId, dirPath),
       downloadFile: (sessionId, filePath, downloadName) => this.client.sessions.downloadFile(sessionId, filePath, downloadName),
@@ -4222,7 +4499,7 @@ function resolveAuthToken(options) {
 
 // src/version.ts
 var SDK_NAME = "agent-client";
-var SDK_VERSION = true ? "2608.0.6" : "1.1.1";
+var SDK_VERSION = true ? "2608.0.7-beta.0" : "1.1.1";
 
 // src/socket.ts
 function withSdkIdentity(auth) {
@@ -4674,45 +4951,18 @@ function connectEmbedded(options) {
   };
 }
 
-// src/schemas/session.ts
-import { type } from "arktype";
-var SessionStatus = type(
-  "'created' | 'running' | 'completed' | 'failed' | 'interrupted' | 'waiting_for_input'"
-);
-var SessionInfo = type({
-  id: "string",
-  intent: "string",
-  status: SessionStatus,
-  created_at: "string",
-  updated_at: "string",
-  "shared?": "boolean",
-  "memory_enabled?": "boolean",
-  "is_persistent?": "boolean",
-  "is_headless?": "boolean",
-  "viewer_role?": "'owner' | 'viewer'",
-  "template_id?": "string | null",
-  "model?": "string | null",
-  "enable_thinking?": "boolean | null",
-  "solution_id?": "string | null",
-  "solution_ref?": "unknown",
-  "biz_role_id?": "string | null",
-  "solution?": "unknown",
-  "plan_summary?": "string | null",
-  "primary_skill_id?": "string | null",
-  // ship-attack v2：后端 engine.set_bound_skill 写入的"绑定 skill id"。
-  // 未绑定 / 非 ship-attack session 为 null。
-  "bound_skill_id?": "string | null",
-  "replay_state?": "unknown",
-  "is_pinned?": "boolean",
-  "pinned_at?": "string | null",
-  "is_example?": "boolean",
-  "ports?": "unknown",
-  "disable_tools?": "string[]",
-  "runtime_type?": "string | null",
-  "daemon_id?": "string | null",
-  "workspace_path?": "string | null",
-  "match?": "unknown"
-});
+// src/schemas/projection-utils.ts
+function latestPostChatFollowup(turns) {
+  for (let index = turns.length - 1; index >= 0; index -= 1) {
+    const turn = turns[index];
+    if (turn.loop_id === "root" && (turn.role === "user" || turn.role === "assistant") && (turn.kind ?? "message") === "message" && !turn.blocks.some(
+      (block) => ["mode_change", "plan_status", "ask_user_answer"].includes(block.type)
+    )) {
+      return turn.role === "assistant" ? turn.post_chat_followup ?? null : null;
+    }
+  }
+  return null;
+}
 
 // src/schemas/solution.ts
 var LayoutType = /* @__PURE__ */ ((LayoutType2) => {
@@ -4745,6 +4995,7 @@ export {
   BladeApiError,
   BladeClient,
   ClientProjectionBuilder,
+  DEFAULT_REPLAY_SPEED,
   LayoutType,
   ModelsResource,
   SDK_NAME,
@@ -4755,6 +5006,7 @@ export {
   SessionStatus,
   Task,
   TaskStatus,
+  acceptedPostChatFollowupCompletesLatestRun,
   buildMessageContent,
   connectEmbedded,
   contentPreview,
@@ -4768,7 +5020,10 @@ export {
   isCommandEnvelope,
   isHiddenInternalMessage,
   isInboundEnvelope,
+  latestPostChatFollowup,
   normalizeMessageContent,
+  reconcileOptimisticUserTurns,
+  toReplaySnapshot,
   transformSlashCommand
 };
 //# sourceMappingURL=index.js.map