Fix openai#976 MCP filtering, make agent and run ctx optional (openai#977)

njbrake · web-flow · commit db85a6d0310d · 2025-07-08T13:28:16.000-04:00
This is a patch to address the most pressing issue of openai#976 while we discuss whether or not these parameters should exist in `list_tools` or should be migrated elsewhere. Resolves openai#976
diff --git a/src/agents/mcp/server.py b/src/agents/mcp/server.py
@@ -52,8 +52,8 @@ async def cleanup(self):
     @abc.abstractmethod
     async def list_tools(
         self,
-        run_context: RunContextWrapper[Any],
-        agent: Agent[Any],
+        run_context: RunContextWrapper[Any] | None = None,
+        agent: Agent[Any] | None = None,
     ) -> list[MCPTool]:
         """List the tools available on the server."""
         pass
@@ -243,8 +243,8 @@ async def connect(self):
 
     async def list_tools(
         self,
-        run_context: RunContextWrapper[Any],
-        agent: Agent[Any],
+        run_context: RunContextWrapper[Any] | None = None,
+        agent: Agent[Any] | None = None,
     ) -> list[MCPTool]:
         """List the tools available on the server."""
         if not self.session:
@@ -263,6 +263,8 @@ async def list_tools(
         # Filter tools based on tool_filter
         filtered_tools = tools
         if self.tool_filter is not None:
+            if run_context is None or agent is None:
+                raise UserError("run_context and agent are required for dynamic tool filtering")
             filtered_tools = await self._apply_tool_filter(filtered_tools, run_context, agent)
         return filtered_tools
 
