linkshell-cli 0.2.113 → 0.2.115

package/src/runtime/acp/agent-workspace.ts

@@ -723,6 +723,73 @@ interface ProviderRuntimeCapabilities {
 const ALL_REASONING_EFFORTS = ["none", "minimal", "low", "medium", "high", "xhigh"] as const;
 const ALL_PERMISSION_MODES = ["read_only", "workspace_write", "full_access"] as const;
 const CODEX_COMMAND_NAMES = ["plan", "exit-plan", "compact", "clear", "status", "review", "subagents"] as const;
+const CLAUDE_REMOTE_HIDDEN_COMMANDS = new Set([
+  "add-dir",
+  "agents",
+  "allowed-tools",
+  "android",
+  "app",
+  "bashes",
+  "branch",
+  "bug",
+  "checkpoint",
+  "chrome",
+  "color",
+  "config",
+  "continue",
+  "copy",
+  "cost",
+  "desktop",
+  "diff",
+  "doctor",
+  "exit",
+  "export",
+  "extra-usage",
+  "feedback",
+  "focus",
+  "fork",
+  "hooks",
+  "ide",
+  "init",
+  "install-github-app",
+  "install-slack-app",
+  "ios",
+  "keybindings",
+  "login",
+  "logout",
+  "mcp",
+  "memory",
+  "migrate-installer",
+  "mobile",
+  "model",
+  "passes",
+  "permissions",
+  "plugin",
+  "powerup",
+  "pr-comments",
+  "privacy-settings",
+  "quit",
+  "rc",
+  "release-notes",
+  "remote-control",
+  "remote-env",
+  "resume",
+  "rewind",
+  "settings",
+  "statusline",
+  "stickers",
+  "tasks",
+  "teleport",
+  "terminal-setup",
+  "theme",
+  "tp",
+  "tui",
+  "undo",
+  "upgrade",
+  "vim",
+  "voice",
+  "web-setup",
+]);
 const CLAUDE_BUILT_IN_COMMANDS: Array<{ name: string; description: string; argsMode?: AgentCommandDescriptor["argsMode"]; destructive?: boolean }> = [
   { name: "add-dir", description: "Add a working directory for file access", argsMode: "required" },
   { name: "agents", description: "Manage agent configurations", argsMode: "none" },
@@ -827,6 +894,10 @@ const CLAUDE_BUILT_IN_COMMANDS: Array<{ name: string; description: string; argsM
   { name: "web-setup", description: "Connect your GitHub account to Claude Code on the web", argsMode: "none" },
 ];
 
+function isClaudeRemoteFriendlyCommand(name: string): boolean {
+  return !CLAUDE_REMOTE_HIDDEN_COMMANDS.has(name.replace(/^\/+/, ""));
+}
+
 function commandId(provider: AgentProvider, name: string, source: AgentCommandSource = "built_in"): string {
   return `${provider}:${source}:${name.replace(/^\/+/, "")}`;
 }
@@ -946,15 +1017,17 @@ function defaultProviderCommands(provider: AgentProvider, cwd: string, enabled:
     }));
   }
   if (provider === "claude") {
-    const builtIns = CLAUDE_BUILT_IN_COMMANDS.map((entry) => makeCommand({
-      provider,
-      name: entry.name,
-      description: entry.description,
-      argsMode: entry.argsMode,
-      destructive: entry.destructive,
-      disabledReason,
-      executionKind: "prompt",
-    }));
+    const builtIns = CLAUDE_BUILT_IN_COMMANDS
+      .filter((entry) => isClaudeRemoteFriendlyCommand(entry.name))
+      .map((entry) => makeCommand({
+        provider,
+        name: entry.name,
+        description: entry.description,
+        argsMode: entry.argsMode,
+        destructive: entry.destructive,
+        disabledReason,
+        executionKind: "prompt",
+      }));
     const custom = customClaudeCommands(cwd).map((command) => ({
       ...command,
       disabledReason: command.disabledReason ?? disabledReason,
@@ -980,7 +1053,12 @@ function mergeCommands(...groups: Array<AgentCommandDescriptor[] | undefined>):
   for (const group of groups) {
     for (const command of group ?? []) {
       const key = `${command.provider ?? ""}:${command.name}`;
-      map.set(key, { ...map.get(key), ...command });
+      const existing = map.get(key);
+      map.set(key, {
+        ...existing,
+        ...command,
+        disabledReason: command.disabledReason ?? existing?.disabledReason,
+      });
     }
   }
   return [...map.values()].sort((a, b) => a.name.localeCompare(b.name));
@@ -998,6 +1076,7 @@ function runtimeCommands(provider: AgentProvider, value: unknown): AgentCommandD
   return commandsValue
     .map((entry) => {
       if (typeof entry === "string") {
+        if (provider === "claude" && !isClaudeRemoteFriendlyCommand(entry)) return undefined;
         return makeCommand({
           provider,
           name: entry,
@@ -1010,6 +1089,7 @@ function runtimeCommands(provider: AgentProvider, value: unknown): AgentCommandD
       const record = asRecord(entry);
       const name = firstString(record, ["name", "command", "id"]);
       if (!name) return undefined;
+      if (provider === "claude" && !isClaudeRemoteFriendlyCommand(name)) return undefined;
       return makeCommand({
         provider,
         name,
@@ -1127,6 +1207,7 @@ export class AgentWorkspaceProxy {
   private error: string | undefined;
   private activeConversationId: string | undefined;
   private currentTurnIds = new Map<string, string>();
+  private turnConversationIds = new Map<string, string>();
   private conversations = new Map<string, AgentConversation>();
   private conversationByAgentSessionId = new Map<string, string>();
   private timelines = new Map<string, AgentTimelineItem[]>();
@@ -1136,6 +1217,7 @@ export class AgentWorkspaceProxy {
   private permissionSources = new Map<string, string>();
   private pendingStructuredInputs = new Map<string, { conversationId: string; input: AgentStructuredInput }>();
   private structuredInputWaiters = new Map<string, PendingStructuredInputWaiter>();
+  private itemConversationIds = new Map<string, string>();
   private toolConversationIds = new Map<string, string>();
 
   constructor(
@@ -1197,7 +1279,7 @@ export class AgentWorkspaceProxy {
           sessionId: conversation?.agentSessionId,
           turnId: this.currentTurnIds.get(payload.conversationId),
         });
-        this.currentTurnIds.delete(payload.conversationId);
+        this.forgetCurrentTurn(payload.conversationId);
         this.updateConversationStatus(payload.conversationId, "idle");
         this.emitStatus(payload.conversationId, "idle", "已停止");
         break;
@@ -1689,7 +1771,7 @@ export class AgentWorkspaceProxy {
         this.conversationByAgentSessionId.set(nextAgentSessionId, conversation.id);
       }
       const turnId = this.extractTurnId(result);
-      if (turnId) this.currentTurnIds.set(conversation.id, turnId);
+      if (turnId) this.rememberTurnConversationId(conversation.id, turnId);
       if (conversation.status === "running" && protocol !== "codex-app-server") {
         this.updateConversationStatus(conversation.id, "idle");
       }
@@ -1927,7 +2009,7 @@ export class AgentWorkspaceProxy {
       process.stderr.write(`[agent:v2] ${method} ${stringify(params).slice(0, 500)}\n`);
     }
     if (method === "initialized") {
-      const conversationId = this.conversationIdFromParams(params) ?? this.activeConversationId;
+      const conversationId = this.conversationIdFromParams(params) ?? this.fallbackConversationId();
       const provider = conversationId ? this.conversations.get(conversationId)?.provider : this.input.availableProviders[0];
       if (provider) {
         const commands = runtimeCommands(provider, params);
@@ -1953,7 +2035,7 @@ export class AgentWorkspaceProxy {
       return;
     }
 
-    const conversationId = this.conversationIdFromParams(params) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(params) ?? this.fallbackConversationId();
     if (method === "item/tool/requestUserInput" || method === "tool/requestUserInput") {
       this.handleStructuredInput(params);
       return;
@@ -1978,14 +2060,14 @@ export class AgentWorkspaceProxy {
     if (method === "turn/started") {
       if (conversationId) {
         const turnId = this.extractTurnId(params);
-        if (turnId) this.currentTurnIds.set(conversationId, turnId);
+        if (turnId) this.rememberTurnConversationId(conversationId, turnId);
         this.updateConversationStatus(conversationId, "running");
       }
       return;
     }
     if (method === "turn/completed") {
       if (conversationId) {
-        this.currentTurnIds.delete(conversationId);
+        this.forgetCurrentTurn(conversationId, this.extractTurnId(params));
         this.updateConversationStatus(conversationId, "idle");
       }
       return;
@@ -2044,7 +2126,7 @@ export class AgentWorkspaceProxy {
   private handleAgentMessageDelta(params: unknown): void {
     const raw = asRecord(params);
     if (!raw) return;
-    const conversationId = this.conversationIdFromParams(raw) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(raw) ?? this.fallbackConversationId();
     if (!conversationId) return;
     const itemId = firstString(raw, ["itemId", "id", "messageId"]) ?? id("msg");
     const delta = firstString(raw, ["delta", "text", "content"]);
@@ -2068,7 +2150,7 @@ export class AgentWorkspaceProxy {
 
   private handlePlanUpdated(params: unknown): void {
     const raw = asRecord(params);
-    const conversationId = this.conversationIdFromParams(raw) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(raw) ?? this.fallbackConversationId();
     if (!conversationId) return;
     const plan = Array.isArray(raw?.plan) ? raw.plan : [];
     const steps = plan
@@ -2097,7 +2179,7 @@ export class AgentWorkspaceProxy {
   private handlePlanDelta(params: unknown): void {
     const raw = asRecord(params);
     if (!raw) return;
-    const conversationId = this.conversationIdFromParams(raw) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(raw) ?? this.fallbackConversationId();
     if (!conversationId) return;
     const itemId = firstString(raw, ["itemId", "id"]) ?? "plan";
     const delta = firstString(raw, ["delta", "text"]);
@@ -2119,23 +2201,25 @@ export class AgentWorkspaceProxy {
   private handleItemStarted(params: unknown): void {
     const item = extractItem(params);
     if (!item) return;
-    const itemType = firstString(item, ["type"]);
+    const sourceConversationId = this.conversationIdFromParams(params);
+    const routedItem = sourceConversationId ? { ...item, conversationId: sourceConversationId } : item;
+    const itemType = firstString(routedItem, ["type"]);
     const normalizedItemType = normalizedIdentifier(itemType);
     if (normalizedItemType === "agentmessage" || normalizedItemType === "assistantmessage") {
-      this.handleCompletedMessageItem(item, true);
+      this.handleCompletedMessageItem(routedItem, true);
       return;
     }
     if (normalizedItemType === "plan") {
-      this.handlePlanUpdated({ plan: [item] });
+      this.handlePlanUpdated({ plan: [routedItem], conversationId: sourceConversationId });
       return;
     }
     if (isSubagentItemType(itemType)) {
-      this.handleSubagentItem(item, "running", true);
+      this.handleSubagentItem(routedItem, "running", true);
       return;
     }
-    if (this.handleSemanticSystemItem(item, "running", true)) return;
-    const conversationId = this.conversationIdFromParams(item) ?? this.activeConversationId;
-    const toolCall = this.toolCallFromItem(item, "running");
+    if (this.handleSemanticSystemItem(routedItem, "running", true)) return;
+    const conversationId = this.conversationIdFromParams(routedItem) ?? this.fallbackConversationId();
+    const toolCall = this.toolCallFromItem(routedItem, "running");
     if (!conversationId || !toolCall) return;
     this.toolConversationIds.set(toolCall.id, conversationId);
     this.upsertTool(conversationId, toolCall);
@@ -2144,23 +2228,25 @@ export class AgentWorkspaceProxy {
   private handleItemCompleted(params: unknown): void {
     const item = extractItem(params);
     if (!item) return;
-    const itemType = firstString(item, ["type"]);
+    const sourceConversationId = this.conversationIdFromParams(params);
+    const routedItem = sourceConversationId ? { ...item, conversationId: sourceConversationId } : item;
+    const itemType = firstString(routedItem, ["type"]);
     const normalizedItemType = normalizedIdentifier(itemType);
     if (normalizedItemType === "agentmessage" || normalizedItemType === "assistantmessage") {
-      this.handleCompletedMessageItem(item, false);
+      this.handleCompletedMessageItem(routedItem, false);
       return;
     }
     if (normalizedItemType === "plan") {
-      this.handlePlanDelta({ ...item, delta: firstString(item, ["text", "content", "message"]) });
+      this.handlePlanDelta({ ...routedItem, delta: firstString(routedItem, ["text", "content", "message"]) });
       return;
     }
     if (isSubagentItemType(itemType)) {
-      this.handleSubagentItem(item, normalizeToolStatus(item.status, true), false);
+      this.handleSubagentItem(routedItem, normalizeToolStatus(routedItem.status, true), false);
       return;
     }
-    if (this.handleSemanticSystemItem(item, normalizeToolStatus(item.status, true), false)) return;
-    const conversationId = this.conversationIdFromParams(item) ?? this.activeConversationId;
-    const toolCall = this.toolCallFromItem(item, normalizeToolStatus(item.status, true));
+    if (this.handleSemanticSystemItem(routedItem, normalizeToolStatus(routedItem.status, true), false)) return;
+    const conversationId = this.conversationIdFromParams(routedItem) ?? this.fallbackConversationId();
+    const toolCall = this.toolCallFromItem(routedItem, normalizeToolStatus(routedItem.status, true));
     if (!conversationId || !toolCall) return;
     const bufferedOutput = this.toolOutputBuffers.get(toolCall.id);
     if (bufferedOutput && !toolCall.output) toolCall.output = bufferedOutput;
@@ -2176,7 +2262,8 @@ export class AgentWorkspaceProxy {
     const conversationId =
       this.conversationIdFromParams(raw) ??
       this.toolConversationIds.get(itemId) ??
-      this.activeConversationId;
+      this.itemConversationIds.get(itemId) ??
+      this.fallbackConversationId();
     if (!conversationId) return;
     const output = appendCapped(this.toolOutputBuffers.get(itemId), delta, 6000);
     this.toolOutputBuffers.set(itemId, output);
@@ -2198,7 +2285,8 @@ export class AgentWorkspaceProxy {
     const conversationId =
       this.conversationIdFromParams(raw) ??
       this.toolConversationIds.get(itemId) ??
-      this.activeConversationId;
+      this.itemConversationIds.get(itemId) ??
+      this.fallbackConversationId();
     if (!conversationId) return;
     const output =
       extractDiffText(raw) ??
@@ -2217,7 +2305,7 @@ export class AgentWorkspaceProxy {
   private handleTurnDiffUpdated(params: unknown): void {
     const raw = asRecord(params);
     if (!raw) return;
-    const conversationId = this.conversationIdFromParams(raw) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(raw) ?? this.fallbackConversationId();
     if (!conversationId) return;
     const diff = extractDiffText(raw);
     if (!diff) return;
@@ -2247,7 +2335,8 @@ export class AgentWorkspaceProxy {
     const conversationId =
       this.conversationIdFromParams(raw) ??
       this.toolConversationIds.get(processId) ??
-      this.activeConversationId;
+      this.itemConversationIds.get(processId) ??
+      this.fallbackConversationId();
     if (!conversationId) return;
     const output = appendCapped(this.toolOutputBuffers.get(processId), delta, 6000);
     this.toolOutputBuffers.set(processId, output);
@@ -2264,7 +2353,7 @@ export class AgentWorkspaceProxy {
 
   private handleAutoApprovalReview(params: unknown, streaming: boolean): void {
     const raw = asRecord(params) ?? {};
-    const conversationId = this.conversationIdFromParams(raw) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(raw) ?? this.fallbackConversationId();
     if (!conversationId) return;
     const itemId = firstString(raw, ["itemId", "id", "reviewId"]) ?? "auto-approval-review";
     const existing = this.findItem(conversationId, itemId);
@@ -2294,7 +2383,7 @@ export class AgentWorkspaceProxy {
   }
 
   private handleCompletedMessageItem(item: Record<string, unknown>, streaming: boolean): void {
-    const conversationId = this.conversationIdFromParams(item) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(item) ?? this.fallbackConversationId();
     if (!conversationId) return;
     const itemId = firstString(item, ["id"]) ?? id("msg");
     const existing = this.findItem(conversationId, itemId);
@@ -2321,7 +2410,7 @@ export class AgentWorkspaceProxy {
       firstString(raw, ["delta", "text", "content", "message"]) ??
       firstString(nested, ["delta", "text", "content", "message"]);
     if (!text) return;
-    const conversationId = this.conversationIdFromParams(raw) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(raw) ?? this.fallbackConversationId();
     if (!conversationId) return;
     if (firstString(raw, ["toolName", "tool", "name"])) {
       this.upsertTool(conversationId, {
@@ -2358,7 +2447,7 @@ export class AgentWorkspaceProxy {
   ): boolean {
     const itemType = firstString(item, ["type"]);
     const normalized = normalizedIdentifier(itemType);
-    const conversationId = this.conversationIdFromParams(item) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(item) ?? this.fallbackConversationId();
     if (!conversationId) return false;
     const itemId = firstString(item, ["id", "itemId"]) ?? id("item");
     const existing = this.findItem(conversationId, itemId);
@@ -2413,7 +2502,7 @@ export class AgentWorkspaceProxy {
     status: AgentToolCall["status"],
     streaming: boolean,
   ): void {
-    const conversationId = this.conversationIdFromParams(item) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(item) ?? this.fallbackConversationId();
     if (!conversationId) return;
     const subagent = decodeSubagentAction(item, status);
     if (!subagent) return;
@@ -2439,7 +2528,7 @@ export class AgentWorkspaceProxy {
 
   private handleStructuredInput(params: unknown, waitForResponse = false): Promise<unknown> | void {
     const raw = asRecord(params) ?? {};
-    const conversationId = this.conversationIdFromParams(raw) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(raw) ?? this.fallbackConversationId();
     const source = firstString(raw, ["method", "source", "requestMethod"]);
     const formatResponse = source === "mcpServer/elicitation/request"
       ? formatMcpElicitationResponse
@@ -2510,7 +2599,7 @@ export class AgentWorkspaceProxy {
     source?: string,
   ): Promise<unknown> | void {
     const raw = asRecord(params) ?? {};
-    const conversationId = this.conversationIdFromParams(raw) ?? this.activeConversationId;
+    const conversationId = this.conversationIdFromParams(raw) ?? this.fallbackConversationId();
     if (!conversationId) return waitForResponse ? Promise.resolve({ outcome: { outcome: "cancelled" } }) : undefined;
     const requestId = firstString(raw, ["requestId", "id", "permissionId"]) ?? id("perm");
     const rawToolCall = asRecord(raw.toolCall) ?? raw;
@@ -2646,6 +2735,7 @@ export class AgentWorkspaceProxy {
   }
 
   private addItem(conversationId: string, item: AgentTimelineItem): void {
+    this.rememberItemConversationId(conversationId, item);
     const timeline = this.timelines.get(conversationId) ?? [];
     timeline.push(item);
     timeline.sort((a, b) => a.createdAt - b.createdAt);
@@ -2657,6 +2747,7 @@ export class AgentWorkspaceProxy {
   }
 
   private upsertItem(conversationId: string, item: AgentTimelineItem): void {
+    this.rememberItemConversationId(conversationId, item);
     const timeline = this.timelines.get(conversationId) ?? [];
     const index = timeline.findIndex((entry) => entry.id === item.id);
     if (index >= 0) timeline[index] = item;
@@ -2684,6 +2775,8 @@ export class AgentWorkspaceProxy {
     };
     this.toolConversationIds.set(toolCall.id, conversationId);
     this.toolConversationIds.set(nextToolCall.id, conversationId);
+    this.itemConversationIds.set(toolCall.id, conversationId);
+    this.itemConversationIds.set(nextToolCall.id, conversationId);
     const kind: AgentTimelineKind = nextToolCall.name.includes("文件")
       ? "file_change"
       : nextToolCall.name.includes("命令")
@@ -2783,6 +2876,22 @@ export class AgentWorkspaceProxy {
         this.toolConversationIds.set(toolId, newId);
       }
     }
+    for (const [turnId, conversationId] of this.turnConversationIds) {
+      if (conversationId === oldId) {
+        this.turnConversationIds.set(turnId, newId);
+      }
+    }
+    const currentTurnId = this.currentTurnIds.get(oldId);
+    if (currentTurnId) {
+      this.currentTurnIds.delete(oldId);
+      this.currentTurnIds.set(newId, currentTurnId);
+      this.turnConversationIds.set(currentTurnId, newId);
+    }
+    for (const [itemId, conversationId] of this.itemConversationIds) {
+      if (conversationId === oldId) {
+        this.itemConversationIds.set(itemId, newId);
+      }
+    }
     if (this.activeConversationId === oldId) {
       this.activeConversationId = newId;
     }
@@ -2861,13 +2970,86 @@ export class AgentWorkspaceProxy {
 
   private conversationIdFromParams(params: unknown): string | undefined {
     const raw = asRecord(params);
-    const agentSessionId = this.extractSessionId(raw);
-    if (agentSessionId) return this.conversationByAgentSessionId.get(agentSessionId);
+    if (!raw) return undefined;
+    const directConversationId = firstString(raw, ["conversationId"]);
+    if (directConversationId && this.conversations.has(directConversationId)) {
+      return directConversationId;
+    }
     const threadId = firstString(raw, ["threadId", "sessionId", "agentSessionId"]);
-    if (threadId) return this.conversationByAgentSessionId.get(threadId);
+    if (threadId) {
+      const conversationId = this.conversationByAgentSessionId.get(threadId);
+      if (conversationId) return conversationId;
+    }
+    const agentSessionId = this.extractSessionId(raw);
+    if (agentSessionId) {
+      const conversationId = this.conversationByAgentSessionId.get(agentSessionId);
+      if (conversationId) return conversationId;
+    }
+    const turnId = this.extractTurnId(raw) ?? firstString(raw, ["turnId"]);
+    if (turnId) {
+      const conversationId = this.turnConversationIds.get(turnId);
+      if (conversationId) return conversationId;
+    }
+    const itemId = firstString(raw, [
+      "itemId",
+      "messageId",
+      "toolCallId",
+      "processId",
+      "callId",
+      "requestId",
+      "permissionId",
+      "id",
+    ]);
+    if (itemId) {
+      const conversationId =
+        this.itemConversationIds.get(itemId) ??
+        this.toolConversationIds.get(itemId);
+      if (conversationId) return conversationId;
+    }
+    for (const nested of [raw.params, raw.item, raw.message, raw.toolCall, raw.command, raw.event]) {
+      const nestedRecord = asRecord(nested);
+      if (!nestedRecord || nestedRecord === raw) continue;
+      const conversationId = this.conversationIdFromParams(nestedRecord);
+      if (conversationId) return conversationId;
+    }
     return undefined;
   }
 
+  private fallbackConversationId(): string | undefined {
+    const liveConversations = [...this.conversations.values()].filter((conversation) =>
+      conversation.status === "running" || conversation.status === "waiting_permission",
+    );
+    return liveConversations.length === 1 ? liveConversations[0]?.id : undefined;
+  }
+
+  private rememberTurnConversationId(conversationId: string, turnId: string): void {
+    this.currentTurnIds.set(conversationId, turnId);
+    this.turnConversationIds.set(turnId, conversationId);
+  }
+
+  private forgetCurrentTurn(conversationId: string, turnId?: string): void {
+    const currentTurnId = this.currentTurnIds.get(conversationId);
+    this.currentTurnIds.delete(conversationId);
+    if (turnId) this.turnConversationIds.delete(turnId);
+    if (currentTurnId && currentTurnId !== turnId) this.turnConversationIds.delete(currentTurnId);
+  }
+
+  private rememberItemConversationId(conversationId: string, item: AgentTimelineItem): void {
+    const keys = [
+      item.id,
+      item.itemId,
+      item.toolCall?.id,
+      item.permission?.requestId,
+      item.structuredInput?.requestId,
+    ].filter((key): key is string => Boolean(key));
+    for (const key of keys) {
+      this.itemConversationIds.set(key, conversationId);
+    }
+    if (item.turnId) {
+      this.turnConversationIds.set(item.turnId, conversationId);
+    }
+  }
+
   private handleProviderExit(provider: AgentProvider, message: string): void {
     this.clients.delete(provider);
     this.agentProtocols.delete(provider);