download db from internet

Author: xzrderek

eval_protocol/mcp_servers/tau2/airplane_environment/airline_environment.py

@@ -21,7 +21,41 @@
 
 logger = logging.getLogger(__name__)
 
-from vendor.tau2.domains.airline.utils import AIRLINE_DB_PATH
+
+def _get_airline_db_path():
+    """Get airline database path, downloading if necessary."""
+    import os
+    import tempfile
+    import urllib.request
+    from pathlib import Path
+
+    # Try local development path first
+    try:
+        from vendor.tau2.domains.airline.utils import AIRLINE_DB_PATH
+
+        if Path(AIRLINE_DB_PATH).exists():
+            return AIRLINE_DB_PATH
+    except ImportError:
+        pass
+
+    # Use a cache directory in user's temp/cache area
+    cache_dir = Path(tempfile.gettempdir()) / "tau2_bench_cache"
+    cache_dir.mkdir(exist_ok=True)
+    airline_db_path = cache_dir / "airline_db.json"
+
+    if not airline_db_path.exists():
+        print(f"📥 Downloading airline database to {airline_db_path}...")
+        url = "https://raw.githubusercontent.com/sierra-research/tau2-bench/main/data/tau2/domains/airline/db.json"
+        try:
+            urllib.request.urlretrieve(url, airline_db_path)
+            print("✅ Download complete!")
+        except Exception as e:
+            raise RuntimeError(f"Failed to download airline database: {e}")
+
+    return airline_db_path
+
+
+AIRLINE_DB_PATH = _get_airline_db_path()
 
 
 class AirlineEnvironment:

eval_protocol/mcp_servers/tau2/mock_environment/mock_environment.py

@@ -20,7 +20,41 @@
 
 logger = logging.getLogger(__name__)
 
-from vendor.tau2.domains.mock.utils import MOCK_DB_PATH
+
+def _get_mock_db_path():
+    """Get mock database path, downloading if necessary."""
+    import os
+    import tempfile
+    import urllib.request
+    from pathlib import Path
+
+    # Try local development path first
+    try:
+        from vendor.tau2.domains.mock.utils import MOCK_DB_PATH
+
+        if Path(MOCK_DB_PATH).exists():
+            return MOCK_DB_PATH
+    except ImportError:
+        pass
+
+    # Use a cache directory in user's temp/cache area
+    cache_dir = Path(tempfile.gettempdir()) / "tau2_bench_cache"
+    cache_dir.mkdir(exist_ok=True)
+    mock_db_path = cache_dir / "mock_db.json"
+
+    if not mock_db_path.exists():
+        print(f"📥 Downloading mock database to {mock_db_path}...")
+        url = "https://raw.githubusercontent.com/sierra-research/tau2-bench/main/data/tau2/domains/mock/db.json"
+        try:
+            urllib.request.urlretrieve(url, mock_db_path)
+            print("✅ Download complete!")
+        except Exception as e:
+            raise RuntimeError(f"Failed to download mock database: {e}")
+
+    return mock_db_path
+
+
+MOCK_DB_PATH = _get_mock_db_path()
 
 
 class MockEnvironment:

eval_protocol/mcp_servers/tau2/retail_environment/retail_environment.py

@@ -20,7 +20,41 @@
 
 logger = logging.getLogger(__name__)
 
-from vendor.tau2.domains.retail.utils import RETAIL_DB_PATH
+
+def _get_retail_db_path():
+    """Get retail database path, downloading if necessary."""
+    import os
+    import tempfile
+    import urllib.request
+    from pathlib import Path
+
+    # Try local development path first
+    try:
+        from vendor.tau2.domains.retail.utils import RETAIL_DB_PATH
+
+        if Path(RETAIL_DB_PATH).exists():
+            return RETAIL_DB_PATH
+    except ImportError:
+        pass
+
+    # Use a cache directory in user's temp/cache area
+    cache_dir = Path(tempfile.gettempdir()) / "tau2_bench_cache"
+    cache_dir.mkdir(exist_ok=True)
+    retail_db_path = cache_dir / "retail_db.json"
+
+    if not retail_db_path.exists():
+        print(f"📥 Downloading retail database to {retail_db_path}...")
+        url = "https://raw.githubusercontent.com/sierra-research/tau2-bench/40f46d3540dc95aca145ddecb0464fdd9a1e8c15/data/tau2/domains/retail/db.json"
+        try:
+            urllib.request.urlretrieve(url, retail_db_path)
+            print("✅ Download complete!")
+        except Exception as e:
+            raise RuntimeError(f"Failed to download retail database: {e}")
+
+    return retail_db_path
+
+
+RETAIL_DB_PATH = _get_retail_db_path()
 
 
 class RetailEnvironment: