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

drift/engine_launcher.py

@@ -101,20 +101,38 @@ def _download_file(url: str, dest: Path) -> None:
         headers["Accept"] = "application/octet-stream"
     if token:
         headers["Authorization"] = f"Bearer {token}"
-    try:
+
+    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)
-    except Exception:
-        # Fallback: urllib (stdlib, no deps, works when requests has issues)
+
+    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:
     """
@@ -145,13 +163,11 @@ def ensure_engine() -> bool:
         try:
             _download_file(url, bin_path)
         except Exception as 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)
-                print(f"drift: Try manually: curl -L -o {bin_path} {url}", 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.10.dist-info → drift_ml-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drift-ml
-Version: 0.1.10
+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.10.dist-info → drift_ml-0.1.11.dist-info}/RECORD

@@ -1,6 +1,6 @@
 drift/__init__.py,sha256=X0NUP5ZAZSz-rBFfjvmmS4IYWP2_CZtu187mqwpWwqk,127
 drift/__main__.py,sha256=r2mhZ3sCvDY7iEEiZoGDvUPei_CY24tfMQgs7Oal2yk,856
-drift/engine_launcher.py,sha256=GlqT5KU_01mPP4kMO4UF__trbjtdcRHl3lmMwjAyEco,6168
+drift/engine_launcher.py,sha256=DlnSRNeHdov_XfVQ-QvDyTTN0J-wQdTlSnx8N_6N_7o,6648
 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.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,,
+drift_ml-0.1.11.dist-info/METADATA,sha256=NMdqz6WMo0nNe81-4M3T96-FglpZOZouWyCM31q8YPE,169
+drift_ml-0.1.11.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+drift_ml-0.1.11.dist-info/entry_points.txt,sha256=aCY7U9M8nhYj_tIfTXJmYkVmXY3ZoxF0tebDZzYswv8,46
+drift_ml-0.1.11.dist-info/top_level.txt,sha256=3u2KGqsciGZQ2uCoBivm55t3e8er8S4xnqkgdQ_8oeM,6
+drift_ml-0.1.11.dist-info/RECORD,,