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

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

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

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

@@ -16,7 +16,7 @@ try:
 except ImportError:
     requests = None
 
-ENGINE_TAG = "v0.1.3"
+ENGINE_TAG = "v0.1.4"
 GITHUB_REPO = "lakshitsachdeva/drift"
 GITHUB_API = f"https://api.github.com/repos/{GITHUB_REPO}/releases/tags/{ENGINE_TAG}"
 ENGINE_PORT = os.environ.get("DRIFT_ENGINE_PORT", "8000")
@@ -62,6 +62,13 @@ 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)
     token = os.environ.get("DRIFT_GITHUB_TOKEN") or os.environ.get("GITHUB_TOKEN")
     headers = {
         "User-Agent": "Drift-Engine-Launcher/1.0",
@@ -79,14 +86,9 @@ def _get_asset_download_url(asset_name: str) -> str:
     data = r.json()
     for a in data.get("assets", []):
         if a.get("name") == asset_name:
-            # Prefer browser_download_url for public repos (no auth needed)
-            url = a.get("browser_download_url")
+            url = a.get("browser_download_url") or a.get("url")
             if url:
                 return url
-            # API URL requires Accept: application/octet-stream
-            api_url = a.get("url")
-            if api_url:
-                return api_url
     raise RuntimeError(f"Asset {asset_name} not found in release {ENGINE_TAG}")
 
 

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

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

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

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