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

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

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

{drift_ml-0.1.8 → drift_ml-0.1.9}/drift/engine_launcher.py

@@ -61,14 +61,12 @@ 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)."""
+    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",
@@ -137,8 +135,16 @@ def ensure_engine() -> bool:
             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
+            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
         if platform.system() != "Windows":
             bin_path.chmod(0o755)
         if platform.system() == "Darwin":

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

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

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

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