Add name argument

Author: tconley1428

temporalio/contrib/openai_agents/_mcp.py

@@ -141,23 +141,25 @@ class StatelessMCPServerProvider:
 
     def __init__(
         self,
+        name: str,
         server_factory: Union[
             Callable[[], MCPServer], Callable[[Optional[Any]], MCPServer]
         ],
     ):
         """Initialize the stateless temporal MCP server.
 
         Args:
+            name: The name of the MCP server.
             server_factory: A function which will produce MCPServer instances. It should return a new server each time
-                so that state is not shared between workflow runs
+                so that state is not shared between workflow runs.
         """
         self._server_factory = server_factory
 
         # Cache whether the server factory needs to be provided with arguments
         sig = inspect.signature(self._server_factory)
         self._server_accepts_arguments = len(sig.parameters) != 0
 
-        self._name = self._create_server(None).name + "-stateless"
+        self._name = name + "-stateless"
         super().__init__()
 
     def _create_server(self, factory_argument: Optional[Any]) -> MCPServer:
@@ -417,16 +419,18 @@ class StatefulMCPServerProvider:
 
     def __init__(
         self,
+        name: str,
         server_factory: Callable[[Optional[Any]], MCPServer],
     ):
         """Initialize the stateful temporal MCP server.
 
         Args:
+            name: The name of the MCP server.
             server_factory: A function which will produce MCPServer instances. It should return a new server each time
                 so that state is not shared between workflow runs
         """
         self._server_factory = server_factory
-        self._name = server_factory(None).name + "-stateful"
+        self._name = name + "-stateful"
         self._connect_handle: Optional[ActivityHandle] = None
         self._servers: dict[str, MCPServer] = {}
         super().__init__()

tests/contrib/openai_agents/test_openai.py

@@ -2464,9 +2464,9 @@ async def test_mcp_server(
 
     tracking_server = get_tracking_server(name="HelloServer")
     server: Union[StatefulMCPServerProvider, StatelessMCPServerProvider] = (
-        StatefulMCPServerProvider(lambda _: tracking_server)
+        StatefulMCPServerProvider("HelloServer", lambda _: tracking_server)
         if stateful
-        else StatelessMCPServerProvider(lambda _: tracking_server)
+        else StatelessMCPServerProvider("HelloServer", lambda _: tracking_server)
     )
 
     new_config = client.config()
@@ -2577,9 +2577,9 @@ def factory(args: Optional[Any]) -> MCPServer:
         return get_tracking_server("HelloServer")
 
     server: Union[StatefulMCPServerProvider, StatelessMCPServerProvider] = (
-        StatefulMCPServerProvider(factory)
+        StatefulMCPServerProvider("HelloServer", factory)
         if stateful
-        else StatelessMCPServerProvider(factory)
+        else StatelessMCPServerProvider("HelloServer", factory)
     )
 
     new_config = client.config()
@@ -2625,6 +2625,7 @@ async def test_stateful_mcp_server_no_worker(client: Client):
     from temporalio.contrib.openai_agents import StatefulMCPServerProvider
 
     server = StatefulMCPServerProvider(
+        "Filesystem-Server",
         lambda _: MCPServerStdio(
             name="Filesystem-Server",
             params={
@@ -2635,7 +2636,7 @@ async def test_stateful_mcp_server_no_worker(client: Client):
                     os.path.dirname(os.path.abspath(__file__)),
                 ],
             },
-        )
+        ),
     )
 
     # Override the connect activity to not actually start a worker