style: apply ruff formatting

Author: MuhammadHassaanArain

src/agents/mcp/server.py

@@ -16,7 +16,14 @@
 from mcp.client.sse import sse_client
 from mcp.client.streamable_http import GetSessionIdCallback, streamablehttp_client
 from mcp.shared.message import SessionMessage
-from mcp.types import CallToolResult, GetPromptResult, InitializeResult, ListPromptsResult, ListResourcesResult, ReadResourceResult
+from mcp.types import (
+    CallToolResult,
+    GetPromptResult,
+    InitializeResult,
+    ListPromptsResult,
+    ListResourcesResult,
+    ReadResourceResult,
+)
 from typing_extensions import NotRequired, TypedDict
 
 from ..exceptions import UserError
@@ -80,15 +87,17 @@ async def call_tool(self, tool_name: str, arguments: dict[str, Any] | None) -> C
         pass
 
     @abc.abstractmethod
-    async def list_resources(self)-> ListResourcesResult:
+    async def list_resources(self) -> ListResourcesResult:
         """List resources available on the server."""
         pass
+
     @abc.abstractmethod
-    async def read_resource(self, uri:str)-> ReadResourceResult:
+    async def read_resource(self, uri: str) -> ReadResourceResult:
         """Read the content of a resource by URI."""
         pass
+
     @abc.abstractmethod
-    async def subscribe_resource(self, uri:str, **kwargs) -> AsyncIterator[SessionMessage]:
+    async def subscribe_resource(self, uri: str, **kwargs) -> AsyncIterator[SessionMessage]:
         """Subscribe to resource updates."""
         pass
 
@@ -105,7 +114,7 @@ async def get_prompt(
     ) -> GetPromptResult:
         """Get a specific prompt from the server."""
         pass
-    
+
 
 class _MCPServerWithClientSession(MCPServer, abc.ABC):
     """Base class for MCP servers that use a `ClientSession` to communicate with the server."""
@@ -301,10 +310,6 @@ async def connect(self):
             await self.cleanup()
             raise
 
-
-
-
-
     async def list_tools(
         self,
         run_context: RunContextWrapper[Any] | None = None,
@@ -343,29 +348,27 @@ async def call_tool(self, tool_name: str, arguments: dict[str, Any] | None) -> C
 
         return await self._run_with_retries(lambda: session.call_tool(tool_name, arguments))
 
-    async def list_resources(self)-> ListResourcesResult:
+    async def list_resources(self) -> ListResourcesResult:
         if not self.session:
             raise UserError("Server not initialized. Make sure you call `connect()` first.")
         session = self.session
         assert session is not None
         return await session.list_resources()
-        
-    async def read_resource(self,uri:str)-> ReadResourceResult:
+
+    async def read_resource(self, uri: str) -> ReadResourceResult:
         if not self.session:
             raise UserError("Server not initialized. Make sure you call `connect()` first.")
         session = self.session
         assert session is not None
         return await session.read_resource(uri)
-    
-    async def subscribe_resource(self, uri:str, **kwargs)-> AsyncIterator[SessionMessage]:
+
+    async def subscribe_resource(self, uri: str, **kwargs) -> AsyncIterator[SessionMessage]:
         if not self.session:
             raise UserError("Server not initialized. Make sure you call `connect()` first.")
         session = self.session
         assert session is not None
         async for msg in session.subscribe_resource(uri, **kwargs):
             yield msg
-        
-
 
     async def list_prompts(
         self,
@@ -634,8 +637,6 @@ class MCPServerStreamableHttpParams(TypedDict):
     """Custom HTTP client factory for configuring httpx.AsyncClient behavior."""
 
 
-
-
 class MCPServerStreamableHttp(_MCPServerWithClientSession):
     """MCP server implementation that uses the Streamable HTTP transport. See the [spec]
     (https://modelcontextprotocol.io/specification/2025-03-26/basic/transports#streamable-http)
@@ -731,4 +732,4 @@ def create_streams(
     @property
     def name(self) -> str:
         """A readable name for the server."""
-        return self._name
+        return self._name