import error

Author: xzrderek

eval_protocol/mcp_servers/tau2/mock_environment/mock_environment.py

@@ -20,7 +20,7 @@
 
 logger = logging.getLogger(__name__)
 
-MOCK_DB_PATH = Path(__file__).parent / "db.json"
+from vendor.tau2.domains.mock.utils import MOCK_DB_PATH
 
 
 class MockEnvironment:

eval_protocol/mcp_servers/tau2/retail_environment/retail_environment.py

@@ -20,7 +20,7 @@
 
 logger = logging.getLogger(__name__)
 
-RETAIL_DB_PATH = Path(__file__).parent / "db.json"
+from vendor.tau2.domains.retail.utils import RETAIL_DB_PATH
 
 
 class RetailEnvironment:

eval_protocol/mcp_servers/tau2/server.py

@@ -17,8 +17,13 @@
 import sys
 from pathlib import Path
 
-# Add root directory to path so we can import eval_protocol
-sys.path.insert(0, str(Path(__file__).parent.parent.parent))
+# Add current directory first for local imports (tau2_mcp)
+sys.path.insert(0, str(Path(__file__).parent))
+
+# Add eval_protocol parent to path, but use append to avoid priority conflicts
+parent_dir = str(Path(__file__).parent.parent.parent)
+if parent_dir not in sys.path:
+    sys.path.append(parent_dir)
 
 from tau2_mcp import AirlineDomainMcp, MockDomainMcp, RetailDomainMcp
 

eval_protocol/mcp_servers/tau2/tau2_mcp.py

@@ -11,11 +11,19 @@
 import os
 from typing import Annotated, Any, Dict, List, Optional
 
-from eval_protocol.mcp_servers.tau2.airplane_environment.airline_environment import AirlineEnvironment
+try:
+    # Try relative imports first (when run as module)
+    from .airplane_environment.airline_environment import AirlineEnvironment
+    from .mock_environment.mock_environment import MockEnvironment
+    from .retail_environment.retail_environment import RetailEnvironment
+except ImportError:
+    # Fallback to direct imports (when run as script)
+    from airplane_environment.airline_environment import AirlineEnvironment
+    from mock_environment.mock_environment import MockEnvironment
+    from retail_environment.retail_environment import RetailEnvironment
+
 from mcp.server.fastmcp import Context
-from eval_protocol.mcp_servers.tau2.mock_environment.mock_environment import MockEnvironment
 from pydantic import Field
-from eval_protocol.mcp_servers.tau2.retail_environment.retail_environment import RetailEnvironment
 
 from eval_protocol.mcp import EnvironmentAdapter, McpGym
 from eval_protocol.mcp.mcpgym import control_plane_endpoint