drift-ml 0.1.9__tar.gz → 0.1.11__tar.gz

{drift_ml-0.1.9 → drift_ml-0.1.11}/PKG-INFO

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

{drift_ml-0.1.9 → drift_ml-0.1.11}/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_ml-0.1.9 → drift_ml-0.1.11}/drift/engine_launcher.py

@@ -62,6 +62,9 @@ def _engine_running() -> bool:
 
 def _get_asset_download_url(asset_name: str) -> str:
     """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}"
 
 
@@ -91,19 +94,44 @@ def _get_asset_download_url_via_api(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)
+
+    def try_requests():
+        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)
+
+    def try_urllib():
+        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 try_curl():
+        result = subprocess.run(
+            ["curl", "-fsSL", "-o", str(dest), url],
+            capture_output=True,
+            timeout=120,
+        )
+        if result.returncode != 0:
+            raise RuntimeError(f"curl failed: {result.stderr.decode() or result.returncode}")
+
+    for attempt in [try_curl, try_requests, try_urllib]:
+        try:
+            attempt()
+            return
+        except Exception:
+            continue
+    raise RuntimeError(f"Download failed. Try manually: curl -L -o {dest} {url}")
 
 
 def ensure_engine() -> bool:
@@ -131,20 +159,15 @@ 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:
-            if "404" in str(e) or "Not Found" in str(e):
-                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)
-                    return False
-            else:
-                print(f"drift: Download failed: {e}", file=sys.stderr)
-                return False
+            print(f"drift: Download failed: {e}", file=sys.stderr)
+            print(f"drift: Run this manually, then try again:", file=sys.stderr)
+            print(f"  mkdir -p ~/.drift/bin && curl -L -o ~/.drift/bin/{asset} {url}", file=sys.stderr)
+            print(f"  chmod +x ~/.drift/bin/{asset}", file=sys.stderr)
+            return False
         if platform.system() != "Windows":
             bin_path.chmod(0o755)
         if platform.system() == "Darwin":

{drift_ml-0.1.9 → drift_ml-0.1.11}/drift_ml.egg-info/PKG-INFO

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

{drift_ml-0.1.9 → drift_ml-0.1.11}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "drift-ml"
-version = "0.1.9"
+version = "0.1.11"
 description = "Terminal-first AutoML CLI - chat-based ML engineer"
 requires-python = ">=3.10"
 dependencies = [