drift-ml 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

drift/engine_launcher.py

@@ -61,7 +61,12 @@ def _engine_running() -> bool:
 
 
 def _get_asset_download_url(asset_name: str) -> str:
-    """Resolve GitHub release asset to download URL."""
+    """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",
@@ -79,14 +84,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}")
 
 
@@ -135,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.7.dist-info → drift_ml-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drift-ml
-Version: 0.1.7
+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.7.dist-info → drift_ml-0.1.9.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=LoOoLR_V_EI7vmJOAyMdeHLDphCU9a4nTyfzJF1UGC8,5462
+drift/engine_launcher.py,sha256=gz9PQ9LwDdedwXDdZxlHFzIdYqhdgUkDNzesd3Dq7XM,5862
 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.7.dist-info/METADATA,sha256=jKJYNTlBG2MSvcqA9K8eT0yzE0I74KESeEBqoA51Usw,168
-drift_ml-0.1.7.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-drift_ml-0.1.7.dist-info/entry_points.txt,sha256=aCY7U9M8nhYj_tIfTXJmYkVmXY3ZoxF0tebDZzYswv8,46
-drift_ml-0.1.7.dist-info/top_level.txt,sha256=3u2KGqsciGZQ2uCoBivm55t3e8er8S4xnqkgdQ_8oeM,6
-drift_ml-0.1.7.dist-info/RECORD,,
+drift_ml-0.1.9.dist-info/METADATA,sha256=tds6F8gmV3g7qSqsaib6kn8KgdQz7nJ7iBTOWE7E7Gk,168
+drift_ml-0.1.9.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+drift_ml-0.1.9.dist-info/entry_points.txt,sha256=aCY7U9M8nhYj_tIfTXJmYkVmXY3ZoxF0tebDZzYswv8,46
+drift_ml-0.1.9.dist-info/top_level.txt,sha256=3u2KGqsciGZQ2uCoBivm55t3e8er8S4xnqkgdQ_8oeM,6
+drift_ml-0.1.9.dist-info/RECORD,,