Add policy files to download and clean up package data

Author: xzrderek

eval_protocol/mcp_servers/tau2/airplane_environment/airline_environment.py

@@ -21,41 +21,7 @@
 
 logger = logging.getLogger(__name__)
 
-
-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, FileNotFoundError):
-        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()
+from vendor.tau2.domains.airline.utils import AIRLINE_DB_PATH
 
 
 class AirlineEnvironment:

eval_protocol/mcp_servers/tau2/mock_environment/mock_environment.py

@@ -20,41 +20,7 @@
 
 logger = logging.getLogger(__name__)
 
-
-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, FileNotFoundError):
-        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()
+from vendor.tau2.domains.mock.utils import MOCK_DB_PATH
 
 
 class MockEnvironment:

eval_protocol/mcp_servers/tau2/retail_environment/retail_environment.py

@@ -20,41 +20,7 @@
 
 logger = logging.getLogger(__name__)
 
-
-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, FileNotFoundError):
-        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()
+from vendor.tau2.domains.retail.utils import RETAIL_DB_PATH
 
 
 class RetailEnvironment:

pyproject.toml

@@ -148,6 +148,7 @@ include = ["eval_protocol*", "development*", "vendor*"]
 "eval_protocol.mcp_servers.tau2" = ["*.md", "tests/system_prompts/*.md"]
 "eval_protocol.benchmarks" = ["data/*.jsonl"]
 
+
 [tool.versioneer]
 VCS = "git"
 style = "pep440"