fix(mcp): ensure database and memory manager sync across server instances (#229)

…ction and disconnection handling

Author: cgoinglove

src/app/api/mcp/list/route.ts

@@ -12,6 +12,26 @@ export async function GET() {
     memoryClients.map(({ id, client }) => [id, client] as const),
   );
 
+  const addTargets = servers.filter((server) => !memoryMap.has(server.id));
+
+  const serverIds = new Set(servers.map((s) => s.id));
+  const removeTargets = memoryClients.filter(({ id }) => !serverIds.has(id));
+
+  if (addTargets.length > 0) {
+    // no need to wait for this
+    Promise.allSettled(
+      addTargets.map((server) => mcpClientsManager.refreshClient(server.id)),
+    );
+  }
+  if (removeTargets.length > 0) {
+    // no need to wait for this
+    Promise.allSettled(
+      removeTargets.map((client) =>
+        mcpClientsManager.disconnectClient(client.id),
+      ),
+    );
+  }
+
   const result = servers.map((server) => {
     const mem = memoryMap.get(server.id);
     const info = mem?.getInfo();

src/lib/ai/mcp/create-mcp-clients-manager.ts

@@ -185,13 +185,16 @@ export class MCPClientsManager {
    * Removes a client by name, disposing resources and removing from storage
    */
   async removeClient(id: string) {
-    const client = this.clients.get(id);
-
     if (this.storage) {
       if (await this.storage.has(id)) {
         await this.storage.delete(id);
       }
     }
+    this.disconnectClient(id);
+  }
+
+  async disconnectClient(id: string) {
+    const client = this.clients.get(id);
     this.clients.delete(id);
     if (client) {
       void client.client.disconnect();