From 912a69422165d88cd6a5c36e612781b4d743f110 Mon Sep 17 00:00:00 2001
From: Mohamed Boudra <boudra.moha@gmail.com>
Date: Fri, 23 Jan 2026 13:41:33 +0700
Subject: [PATCH] Update files

---
 packages/server/src/server/agent/agent-manager.ts   | 6 ++++++
 packages/server/src/server/agent/mcp-server.test.ts | 3 ++-
 packages/server/src/server/agent/mcp-server.ts      | 2 +-
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/packages/server/src/server/agent/agent-manager.ts b/packages/server/src/server/agent/agent-manager.ts
index 1336dc925..7a9de18c5 100644
--- a/packages/server/src/server/agent/agent-manager.ts
+++ b/packages/server/src/server/agent/agent-manager.ts
@@ -395,6 +395,12 @@ export class AgentManager {
     this.emitState(agent);
   }
 
+  async setTitle(agentId: string, title: string): Promise<void> {
+    const agent = this.requireAgent(agentId);
+    await this.registry?.setTitle(agentId, title);
+    this.emitState(agent);
+  }
+
   async clearAgentAttention(agentId: string): Promise<void> {
     const agent = this.requireAgent(agentId);
     if (agent.attention.requiresAttention) {
diff --git a/packages/server/src/server/agent/mcp-server.test.ts b/packages/server/src/server/agent/mcp-server.test.ts
index 5a2ce5ff9..3588a4817 100644
--- a/packages/server/src/server/agent/mcp-server.test.ts
+++ b/packages/server/src/server/agent/mcp-server.test.ts
@@ -20,6 +20,7 @@ function createTestDeps(): TestDeps {
     waitForAgentEvent: vi.fn(),
     recordUserMessage: vi.fn(),
     setAgentMode: vi.fn(),
+    setTitle: vi.fn().mockResolvedValue(undefined),
     getAgent: vi.fn(),
     streamAgent: vi.fn(() => (async function* noop() {})()),
     respondToPermission: vi.fn(),
@@ -118,7 +119,7 @@ describe("create_agent MCP tool", () => {
 
     await tool.callback({ title: "  Fix auth  " });
 
-    expect(spies.agentRegistry.setTitle).toHaveBeenCalledWith(
+    expect(spies.agentManager.setTitle).toHaveBeenCalledWith(
       "agent-1",
       "Fix auth"
     );
diff --git a/packages/server/src/server/agent/mcp-server.ts b/packages/server/src/server/agent/mcp-server.ts
index 1b24358e7..1cecfc1de 100644
--- a/packages/server/src/server/agent/mcp-server.ts
+++ b/packages/server/src/server/agent/mcp-server.ts
@@ -938,7 +938,7 @@ export async function createAgentMcpServer(
         );
       }
 
-      await agentRegistry.setTitle(agent.id, normalizedTitle);
+      await agentManager.setTitle(agent.id, normalizedTitle);
       return {
         content: [],
         structuredContent: ensureValidJson({