MCP client refactor - storage adapter

.changeset/plenty-friends-hope.md

@@ -0,0 +1,24 @@
+---
+"agents": minor
+---
+
+### Breaking Changes
+
+- **`getMcpServers()` is now async**: Changed from synchronous to asynchronous method to support storage operations
+- **`DurableObjectOAuthClientProvider` constructor**: Now accepts `OAuthClientStorage` interface instead of `DurableObjectStorage`
+
+### New Features
+
+- **`MCPClientManager` API changes**:
+  - New `registerServer()` method to register servers (replaces part of `connect()`)
+  - New `connectToServer()` method to establish connection (replaces part of `connect()`)
+  - `connect()` method deprecated (still works for backward compatibility)
+  - Requires `MCPClientStorage` interface implementation (provided via `AgentMCPClientStorage`)
+- **Storage abstraction layer**: New `MCPClientStorage` and `OAuthClientStorage` interfaces enable custom storage implementations beyond Durable Objects
+- **Connection state observability**: New `onServerStateChanged()` event for tracking all server state changes
+- **Improved reconnect logic**: `restoreConnectionsFromStorage()` handles failed connections
+
+### Bug Fixes
+
+- Fixed failed connections not being recreated on restore
+- Fixed redundant storage operations during connection restoration

packages/agents/src/ai-chat-agent.ts

@@ -190,10 +190,12 @@ export class AIChatAgent<Env = unknown, State = unknown> extends Agent<
   override async onRequest(request: Request): Promise<Response> {
     return this._tryCatchChat(() => {
       const url = new URL(request.url);
+
       if (url.pathname.endsWith("/get-messages")) {
         const messages = this._loadMessagesFromDb();
         return Response.json(messages);
       }
+
       return super.onRequest(request);
     });
   }