drift-ml 0.1.8__py3-none-any.whl → 0.1.10__py3-none-any.whl

drift/__main__.py

@@ -10,6 +10,10 @@ from drift.cli.repl import run_repl
 
 
 def main() -> None:
+    if "--version" in sys.argv or "-v" in sys.argv:
+        from importlib.metadata import version
+        print(f"drift-ml {version('drift-ml')}")
+        return
     base_url = os.environ.get("DRIFT_BACKEND_URL")
     if not base_url:
         from drift.engine_launcher import ensure_engine

drift/engine_launcher.py

@@ -61,14 +61,15 @@ def _engine_running() -> bool:
 
 
 def _get_asset_download_url(asset_name: str) -> str:
-    """Resolve GitHub release asset to download URL."""
-    # Direct URL works for public repos, no API rate limits
-    direct_url = f"https://github.com/{GITHUB_REPO}/releases/download/{ENGINE_TAG}/{asset_name}"
-    r = requests.head(direct_url, timeout=10, allow_redirects=True)
-    if r.status_code == 200:
-        return r.url  # follow redirects to final URL
-
-    # Fallback: API (needed for private repos or if direct fails)
+    """Resolve GitHub release asset to download URL. Prefer direct URL (no API)."""
+    base = os.environ.get("DRIFT_ENGINE_BASE_URL")
+    if base:
+        return f"{base.rstrip('/')}/{asset_name}"
+    return f"https://github.com/{GITHUB_REPO}/releases/download/{ENGINE_TAG}/{asset_name}"
+
+
+def _get_asset_download_url_via_api(asset_name: str) -> str:
+    """Fallback: get URL via GitHub API (for private repos)."""
     token = os.environ.get("DRIFT_GITHUB_TOKEN") or os.environ.get("GITHUB_TOKEN")
     headers = {
         "User-Agent": "Drift-Engine-Launcher/1.0",
@@ -93,19 +94,26 @@ def _get_asset_download_url(asset_name: str) -> str:
 
 
 def _download_file(url: str, dest: Path) -> None:
+    dest.parent.mkdir(parents=True, exist_ok=True)
     token = os.environ.get("DRIFT_GITHUB_TOKEN") or os.environ.get("GITHUB_TOKEN")
-    headers = {"User-Agent": "Drift-Engine-Launcher/1.0"}
-    # API URLs need Accept: application/octet-stream; browser_download_url works with default
+    headers = {"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36"}
     if "api.github.com" in url:
         headers["Accept"] = "application/octet-stream"
     if token:
         headers["Authorization"] = f"Bearer {token}"
-    r = requests.get(url, headers=headers, stream=True, timeout=60)
-    r.raise_for_status()
-    dest.parent.mkdir(parents=True, exist_ok=True)
-    with open(dest, "wb") as f:
-        for chunk in r.iter_content(chunk_size=65536):
-            f.write(chunk)
+    try:
+        r = requests.get(url, headers=headers, stream=True, timeout=60)
+        r.raise_for_status()
+        with open(dest, "wb") as f:
+            for chunk in r.iter_content(chunk_size=65536):
+                f.write(chunk)
+    except Exception:
+        # Fallback: urllib (stdlib, no deps, works when requests has issues)
+        import urllib.request
+        req = urllib.request.Request(url, headers=headers)
+        with urllib.request.urlopen(req, timeout=60) as resp:
+            with open(dest, "wb") as f:
+                f.write(resp.read())
 
 
 def ensure_engine() -> bool:
@@ -133,12 +141,17 @@ def ensure_engine() -> bool:
         ext = ".exe" if platform.system() == "Windows" else ""
         asset = f"drift-engine-{plat}-{arch}{ext}"
         print(f"drift: Downloading engine ({asset})...", file=sys.stderr)
+        url = _get_asset_download_url(asset)
         try:
-            url = _get_asset_download_url(asset)
             _download_file(url, bin_path)
         except Exception as e:
-            print(f"drift: Download failed: {e}", file=sys.stderr)
-            return False
+            try:
+                url = _get_asset_download_url_via_api(asset)
+                _download_file(url, bin_path)
+            except Exception as e2:
+                print(f"drift: Download failed: {e2}", file=sys.stderr)
+                print(f"drift: Try manually: curl -L -o {bin_path} {url}", file=sys.stderr)
+                return False
         if platform.system() != "Windows":
             bin_path.chmod(0o755)
         if platform.system() == "Darwin":

{drift_ml-0.1.8.dist-info → drift_ml-0.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drift-ml
-Version: 0.1.8
+Version: 0.1.10
 Summary: Terminal-first AutoML CLI - chat-based ML engineer
 Requires-Python: >=3.10
 Requires-Dist: requests>=2.28.0

{drift_ml-0.1.8.dist-info → drift_ml-0.1.10.dist-info}/RECORD

@@ -1,6 +1,6 @@
 drift/__init__.py,sha256=X0NUP5ZAZSz-rBFfjvmmS4IYWP2_CZtu187mqwpWwqk,127
-drift/__main__.py,sha256=RnBMIqpDEMHknOKn-s5Q2MTUWm_UsKSkYuOArzShQeo,693
-drift/engine_launcher.py,sha256=YFSqXFuwvVB6JHbwAsbsbD2uHx8fdwPnIBhTD07oiQc,5623
+drift/__main__.py,sha256=r2mhZ3sCvDY7iEEiZoGDvUPei_CY24tfMQgs7Oal2yk,856
+drift/engine_launcher.py,sha256=GlqT5KU_01mPP4kMO4UF__trbjtdcRHl3lmMwjAyEco,6168
 drift/cli/__init__.py,sha256=OQt7M06e98e_L_60Qz-HsmSqGXKWdH53SvgETJ_BZZ0,172
 drift/cli/client.py,sha256=hGxz-J8fC_gGV9BZnpCBwkQZg4Q0_s9-Agk2PuXQ5MQ,5527
 drift/cli/repl.py,sha256=hRnx4nfP_StnXDbdRZXnnjA8aNvrdQSV-uTNsuaaO60,8262
@@ -9,8 +9,8 @@ drift/llm_adapters/__init__.py,sha256=y1UhZWlC8Ik_OKfLcOp0JZP-FKR3MBBCemWwsL6Tnk
 drift/llm_adapters/base.py,sha256=KlZUPYpvCI8pafklBWel0GHHLNtVKVwSHA92MZt3VsI,1331
 drift/llm_adapters/gemini_cli.py,sha256=Z61wY3yFiZqPrQrpJAQrBtMbBkr2qLWBkni-A4p9lZo,2163
 drift/llm_adapters/local_llm.py,sha256=Z6j6z1CXk2LMeQ5ZnY4o38PiYkHYmcIkgJGkHdU50M8,2279
-drift_ml-0.1.8.dist-info/METADATA,sha256=lN6INF5pRo1iuVmyG3QeLmVEFX4bDkNMSNKWG2g24EI,168
-drift_ml-0.1.8.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-drift_ml-0.1.8.dist-info/entry_points.txt,sha256=aCY7U9M8nhYj_tIfTXJmYkVmXY3ZoxF0tebDZzYswv8,46
-drift_ml-0.1.8.dist-info/top_level.txt,sha256=3u2KGqsciGZQ2uCoBivm55t3e8er8S4xnqkgdQ_8oeM,6
-drift_ml-0.1.8.dist-info/RECORD,,
+drift_ml-0.1.10.dist-info/METADATA,sha256=T8JG1IH8YWkfWQjzwxV30ICNP3MRZrqWgrps5sORy9o,169
+drift_ml-0.1.10.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+drift_ml-0.1.10.dist-info/entry_points.txt,sha256=aCY7U9M8nhYj_tIfTXJmYkVmXY3ZoxF0tebDZzYswv8,46
+drift_ml-0.1.10.dist-info/top_level.txt,sha256=3u2KGqsciGZQ2uCoBivm55t3e8er8S4xnqkgdQ_8oeM,6
+drift_ml-0.1.10.dist-info/RECORD,,