Delay import of mcp servers

Author: tconley1428

temporalio/contrib/openai_agents/_temporal_openai_agents.py

@@ -52,15 +52,11 @@
 )
 from temporalio.worker.workflow_sandbox import SandboxedWorkflowRunner
 
-# MCP only supported on python >=3.10
-try:
-    from temporalio.contrib.openai_agents._mcp import TemporalMCPServer
-except ImportError:
-    pass
-
 if typing.TYPE_CHECKING:
     from agents.mcp import MCPServer
 
+    from temporalio.contrib.openai_agents._mcp import TemporalMCPServer
+
 
 @contextmanager
 def set_open_ai_agent_temporal_overrides(
@@ -273,12 +269,18 @@ def __init__(
         self._model_params = model_params
         self._model_provider = model_provider
 
-        self._mcp_servers = [
-            server
-            if isinstance(server, TemporalMCPServer)
-            else TemporalMCPServer(server)
-            for server in mcp_servers
-        ]
+        if mcp_servers:
+            # Delayed import as mcp servers only work on python >=3.10
+            from temporalio.contrib.openai_agents._mcp import TemporalMCPServer
+
+            self._mcp_servers = [
+                server
+                if isinstance(server, TemporalMCPServer)
+                else TemporalMCPServer(server)
+                for server in mcp_servers
+            ]
+        else:
+            self._mcp_servers = []
 
     def init_client_plugin(self, next: temporalio.client.Plugin) -> None:
         """Set the next client plugin"""