@blade-hq/agent-client 2610.0.0-beta.27 → 2610.0.0-beta.28

package/dist/index.js

@@ -472,6 +472,7 @@ var SessionInfo = type({
   "disable_tools?": "string[]",
   "runtime_type?": "string | null",
   "daemon_id?": "string | null",
+  "agent_runtime_id?": "string | null",
   "workspace_path?": "string | null",
   "match?": "unknown"
 });
@@ -1566,7 +1567,10 @@ var ClientProjectionBuilder = class {
         );
       }
     }
-    if (state.toolCalls.length === 0) {
+    const isNativeProviderResponse = typeof payload.provider === "string" && payload.provider !== "";
+    if (state.toolCalls.length === 0 || isNativeProviderResponse && state.toolCalls.every(
+      (toolCall) => toolCall.status !== "pending" && toolCall.status !== "awaiting_answer"
+    )) {
       return this.finalize(loopId, "completed");
     }
     return this.syncPatch(state);
@@ -1580,7 +1584,8 @@ var ClientProjectionBuilder = class {
     const durationMs = typeof rawDuration === "number" && rawDuration >= 0 ? Math.round(rawDuration) : null;
     const sl = sourceLoopFor(state.loopId, this.loopDescriptions);
     applyToolResult(state, toolCallId, content, status, durationMs, sl);
-    if (this.shouldFinalizeAfterToolResults(state)) {
+    const isNativeProviderResult = typeof payload.provider === "string" && payload.provider !== "";
+    if (!isNativeProviderResult && this.shouldFinalizeAfterToolResults(state)) {
       return this.finalize(loopId, "completed");
     }
     return this.syncPatch(state);
@@ -1621,6 +1626,11 @@ var ClientProjectionBuilder = class {
     const toolName = String(payload.tool_name ?? "");
     const displayName = this.displayNameResolver(toolName);
     upsertToolCall(state, toolCallId, toolName, displayName, "");
+    const relatedToolCallId = String(payload.related_tool_call_id ?? "");
+    if (relatedToolCallId && relatedToolCallId !== toolCallId) {
+      const related = state.toolCalls.find((toolCall) => toolCall.id === relatedToolCallId);
+      if (related) related.status = "awaiting_answer";
+    }
     return this.syncPatch(state);
   }
   onChildPause(state, loopId, payload) {
@@ -1877,6 +1887,7 @@ function projectHistory(entries) {
   const latestAssistantByLoop = /* @__PURE__ */ new Map();
   const pendingChildPauses = /* @__PURE__ */ new Map();
   const answeredToolCallIds = /* @__PURE__ */ new Set();
+  const nativeToolTurns = /* @__PURE__ */ new Set();
   let latestRootMessage = null;
   let latestRootAssistant = null;
   let sequence = 0;
@@ -1936,10 +1947,27 @@ function projectHistory(entries) {
         duration_ms: numberOrZero(message._duration_ms),
         started_at: typeof entry.timestamp === "string" ? entry.timestamp : void 0
       };
+      const isNativeToolEntry = role === "assistant" && Array.isArray(message.tool_calls) && message.tool_calls.some(
+        (toolCall) => isRecord4(toolCall) && typeof toolCall.provider === "string"
+      );
+      const previousTurn = turns[turns.length - 1];
+      if (role === "assistant" && previousTurn?.role === "assistant" && previousTurn.loop_id === loopId && nativeToolTurns.has(previousTurn)) {
+        previousTurn.blocks.push(...turn.blocks);
+        previousTurn.tool_calls.push(...turn.tool_calls);
+        previousTurn.status = turn.status;
+        previousTurn.model = turn.model ?? previousTurn.model;
+        previousTurn.usage = turn.usage ?? previousTurn.usage;
+        previousTurn.duration_ms = turn.duration_ms || previousTurn.duration_ms;
+        if (turn.started_at) previousTurn.started_at = turn.started_at;
+        if (loopId === "root") latestRootMessage = previousTurn;
+        latestAssistantByLoop.set(loopId, previousTurn);
+        continue;
+      }
       if (loopId === "root" && (role === "user" || role === "assistant")) {
         latestRootMessage = turn;
       }
       turns.push(turn);
+      if (isNativeToolEntry) nativeToolTurns.add(turn);
       if (role === "assistant") latestAssistantByLoop.set(loopId, turn);
       if (loopId === "root") {
         if (role === "assistant") latestRootAssistant = turn;
@@ -1948,6 +1976,21 @@ function projectHistory(entries) {
       for (const toolCall of toolCalls) toolOwners.set(toolCall.id, turn);
       continue;
     }
+    if (kind === "tool_result") {
+      const data = entry.data ?? {};
+      const toolCallId = String(data.tool_call_id ?? "");
+      const owner = toolOwners.get(toolCallId) ?? findLatestAssistant(turns, loopId);
+      if (owner && toolCallId) {
+        const rawContent = data.content;
+        const content = typeof rawContent === "string" ? rawContent : Array.isArray(rawContent) ? rawContent : String(rawContent ?? "");
+        const status = data.is_error === true ? "error" : inferToolStatus(typeof content === "string" ? content : JSON.stringify(content));
+        setToolCallStatus(owner, toolCallId, status, content, null);
+        if (!owner.blocks.some((block) => block.type === "tool_result" && block.tool_call_id === toolCallId)) {
+          owner.blocks.push({ type: "tool_result", content, tool_call_id: toolCallId });
+        }
+      }
+      continue;
+    }
     if (kind === "mode_change") {
       turns.push(markerTurn(id, ++sequence, loopId, "mode_change", entry.data ?? {}));
     } else if (kind === "workspace_change") {
@@ -2196,7 +2239,8 @@ function toCreateSessionPayload(request) {
     solution_snapshot_session_id: request.solution_snapshot_session_id ?? null,
     compaction_ratio: request.compaction_ratio ?? null,
     ...request.runtime_type ? { runtime_type: request.runtime_type } : {},
-    ...request.daemon_id ? { daemon_id: request.daemon_id } : {}
+    ...request.daemon_id ? { daemon_id: request.daemon_id } : {},
+    ...request.agent_runtime_id ? { agent_runtime_id: request.agent_runtime_id } : {}
   };
 }
 var builtinSolutionIds = /* @__PURE__ */ new Set([
@@ -3519,6 +3563,7 @@ var AgentSession = class _AgentSession {
   replayQueue = Promise.resolve();
   pendingReplayMessage = null;
   pendingReplayMode;
+  pendingReplayKbIds;
   recentChatEndAt = 0;
   disposed = false;
   getAppContext = null;
@@ -3687,6 +3732,7 @@ var AgentSession = class _AgentSession {
     if (!options.replayDecision) {
       this.pendingReplayMessage = content;
       this.pendingReplayMode = options.mode;
+      this.pendingReplayKbIds = options.kbIds;
     }
     const payload = {
       session_id: this.sessionId,
@@ -3697,7 +3743,8 @@ var AgentSession = class _AgentSession {
       thinking_override: options.thinkingOverride,
       whatif: options.whatif,
       replay_decision: options.replayDecision,
-      app_context: appContext
+      app_context: appContext,
+      kb_ids: options.kbIds && options.kbIds.length > 0 ? options.kbIds : void 0
     };
     try {
       await this.ensureJoined();
@@ -3733,6 +3780,7 @@ var AgentSession = class _AgentSession {
   rollbackOptimisticSend(optimisticTurnId, askUserAnswer, optimisticAskAnswer, previousAskAnswer, message) {
     this.pendingReplayMessage = null;
     this.pendingReplayMode = void 0;
+    this.pendingReplayKbIds = void 0;
     this.update((current) => {
       let next = optimisticTurnId ? withTurns(
         current,
@@ -3914,7 +3962,7 @@ var AgentSession = class _AgentSession {
   _handleTurnEvents(events) {
     if (!events.length) return;
     const terminalChatEndStatus = extractTerminalChatEndStatus(events);
-    if (!terminalChatEndStatus && hasChatRunEvent2(events)) {
+    if (!terminalChatEndStatus && hasChatRunEvent2(events) && this.state.status !== "waiting_for_input") {
       this.markRunningIfIdle();
     }
     if (this.state.turns.length) {
@@ -3986,8 +4034,10 @@ var AgentSession = class _AgentSession {
   _handleReplayMismatch(data) {
     const message = this.pendingReplayMessage ?? data.actual_message ?? "";
     const mode = this.pendingReplayMode;
+    const kbIds = this.pendingReplayKbIds;
     this.pendingReplayMessage = null;
     this.pendingReplayMode = void 0;
+    this.pendingReplayKbIds = void 0;
     const respond = (decision) => {
       void this.runtime.emitWithAck(
         "chat:send",
@@ -3995,7 +4045,8 @@ var AgentSession = class _AgentSession {
           session_id: this.sessionId,
           message,
           mode,
-          replay_decision: decision
+          replay_decision: decision,
+          kb_ids: kbIds && kbIds.length > 0 ? kbIds : void 0
         },
         3e4
       ).then((response) => {
@@ -4650,7 +4701,7 @@ function resolveAuthToken(options) {
 
 // src/version.ts
 var SDK_NAME = "agent-client";
-var SDK_VERSION = true ? "2610.0.0-beta.27" : "1.1.1";
+var SDK_VERSION = true ? "2610.0.0-beta.28" : "1.1.1";
 
 // src/socket.ts
 function withSdkIdentity(auth) {