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

drift/__main__.py

@@ -19,7 +19,7 @@ def main() -> None:
         from drift.engine_launcher import ensure_engine
 
         if not ensure_engine():
-            print("drift: Failed to start engine. Set DRIFT_BACKEND_URL or run: npm install -g drift-ml", file=sys.stderr)
+            print("drift: Failed to start engine. Check ~/.drift/bin/.engine-stderr.log or set DRIFT_BACKEND_URL", file=sys.stderr)
             sys.exit(1)
         base_url = f"http://127.0.0.1:{os.environ.get('DRIFT_ENGINE_PORT', '8000')}"
     run_repl(base_url=base_url)

drift/engine_launcher.py

@@ -1,11 +1,10 @@
 """
 Download and start the drift engine binary when no backend is running.
-Makes pipx install drift-ml work standalone (no npm needed).
+Open source — no tokens, no auth. Downloads from public GitHub Releases.
 """
 
 import os
 import platform
-import shutil
 import subprocess
 import sys
 from pathlib import Path
@@ -16,9 +15,8 @@ try:
 except ImportError:
     requests = None
 
-ENGINE_TAG = "v0.1.4"
-GITHUB_REPO = "lakshitsachdeva/drift"
-GITHUB_API = f"https://api.github.com/repos/{GITHUB_REPO}/releases/tags/{ENGINE_TAG}"
+GITHUB_REPO = "lakshitsachdeva/drift"  # Engine binaries (v0.2.0+ also in intent2model)
+ENGINE_TAG = "v0.2.0"  # Pinned — direct URL, no API, no rate limits
 ENGINE_PORT = os.environ.get("DRIFT_ENGINE_PORT", "8000")
 HEALTH_URL = f"http://127.0.0.1:{ENGINE_PORT}/health"
 
@@ -60,47 +58,26 @@ def _engine_running() -> bool:
         return False
 
 
-def _get_asset_download_url(asset_name: str) -> str:
-    """Resolve GitHub release asset to download URL. Prefer direct URL (no API)."""
+def _get_asset_url(asset_name: str) -> str:
+    """Direct download URL. No API — avoids rate limits."""
     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}"
 
 
-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",
-        "Accept": "application/vnd.github+json",
-    }
-    if token:
-        headers["Authorization"] = f"Bearer {token}"
-    r = requests.get(GITHUB_API, headers=headers, timeout=15)
-    if r.status_code == 404:
-        raise RuntimeError(
-            f"Release {ENGINE_TAG} not found. "
-            "If the repo is private, set DRIFT_GITHUB_TOKEN with repo read access."
-        )
-    r.raise_for_status()
-    data = r.json()
-    for a in data.get("assets", []):
-        if a.get("name") == asset_name:
-            url = a.get("browser_download_url") or a.get("url")
-            if url:
-                return url
-    raise RuntimeError(f"Asset {asset_name} not found in release {ENGINE_TAG}")
-
-
 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": "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}"
+    headers = {"User-Agent": "Drift/1.0"}
+
+    def try_curl():
+        result = subprocess.run(
+            ["curl", "-fsSL", "-o", str(dest), url],
+            capture_output=True,
+            timeout=120,
+        )
+        if result.returncode != 0:
+            raise RuntimeError("curl failed")
 
     def try_requests():
         r = requests.get(url, headers=headers, stream=True, timeout=60)
@@ -116,40 +93,28 @@ def _download_file(url: str, dest: Path) -> None:
             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}")
+    raise RuntimeError(f"Download failed. Try: curl -L -o {dest} {url}")
 
 
 def ensure_engine() -> bool:
-    """
-    If engine not running: download (if needed), start it, wait for health.
-    Returns True if engine is ready, False on failure.
-    """
+    """Download (if needed), start engine, wait for health."""
     if _engine_running():
         return True
 
     if not requests:
-        print("drift: 'requests' required for engine download. pip install requests", file=sys.stderr)
+        print("drift: 'requests' required. pip install requests", file=sys.stderr)
         return False
 
     bin_path = _get_engine_path()
     bin_dir = _get_engine_dir()
     if not bin_path or not bin_dir:
-        print("drift: Could not resolve engine dir (~/.drift/bin). Set HOME or USERPROFILE.", file=sys.stderr)
+        print("drift: Could not resolve ~/.drift/bin. Set HOME or USERPROFILE.", file=sys.stderr)
         return False
 
     bin_dir.mkdir(parents=True, exist_ok=True)
@@ -159,14 +124,12 @@ 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_url(asset)
             _download_file(url, bin_path)
         except Exception as e:
             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)
+            print(f"drift: Run: mkdir -p ~/.drift/bin && curl -L -o ~/.drift/bin/{asset} <url>", file=sys.stderr)
             return False
         if platform.system() != "Windows":
             bin_path.chmod(0o755)
@@ -176,23 +139,46 @@ def ensure_engine() -> bool:
             except Exception:
                 pass
 
-    # Start engine
+    if platform.system() == "Darwin" and bin_path:
+        try:
+            bin_path.chmod(0o755)
+            subprocess.run(["xattr", "-dr", "com.apple.quarantine", str(bin_path)], check=False, capture_output=True)
+        except Exception:
+            pass
+
     env = {**os.environ, "DRIFT_ENGINE_PORT": ENGINE_PORT}
-    proc = subprocess.Popen(
-        [str(bin_path)],
-        cwd=str(bin_dir),
-        env=env,
-        stdout=subprocess.DEVNULL,
-        stderr=subprocess.DEVNULL,
-        start_new_session=True,
-    )
-    proc.wait()  # wait briefly to avoid race
-    del proc
-
-    # Poll for health
+    stderr_file = bin_dir / ".engine-stderr.log"
+    proc = None
+    try:
+        with open(stderr_file, "w") as errf:
+            proc = subprocess.Popen(
+                [str(bin_path)],
+                cwd=str(bin_dir),
+                env=env,
+                stdout=subprocess.DEVNULL,
+                stderr=errf,
+                start_new_session=True,
+            )
+        try:
+            proc.wait(timeout=3)
+            if proc.returncode and proc.returncode != 0:
+                err = stderr_file.read_text().strip() if stderr_file.exists() else ""
+                print(f"drift: Engine exited with code {proc.returncode}", file=sys.stderr)
+                if err:
+                    print(f"drift: {err[-400:]}", file=sys.stderr)
+                return False
+        except subprocess.TimeoutExpired:
+            pass  # Engine still running
+    except Exception as e:
+        print(f"drift: Engine spawn failed: {e}", file=sys.stderr)
+        return False
+
+    import time
     for _ in range(60):
         if _engine_running():
             return True
-        import time
         time.sleep(0.5)
+    err = stderr_file.read_text().strip() if stderr_file.exists() else ""
+    if err:
+        print(f"drift: Engine log: {err[-400:]}", file=sys.stderr)
     return False

drift_ml-0.2.2.dist-info/METADATA

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: drift-ml
+Version: 0.2.2
+Summary: drift — terminal-first, chat-based AutoML. Open source. No tokens. No auth.
+Project-URL: Homepage, https://github.com/lakshitsachdeva/intent2model
+Project-URL: Repository, https://github.com/lakshitsachdeva/intent2model
+Requires-Python: >=3.10
+License-File: LICENSE
+Requires-Dist: requests>=2.28.0
+Dynamic: license-file

{drift_ml-0.1.11.dist-info → drift_ml-0.2.2.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=DlnSRNeHdov_XfVQ-QvDyTTN0J-wQdTlSnx8N_6N_7o,6648
+drift/__main__.py,sha256=b0sqvVmeQhI-iDKQ_G_nB_n77whL5bnnPp-Llt556Ws,865
+drift/engine_launcher.py,sha256=HmpnjTiOj_Pftj43uvrhY5akEXWPhIIFslqlmqFcCy8,6119
 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,9 @@ 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.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,,
+drift_ml-0.2.2.dist-info/licenses/LICENSE,sha256=MdZfE1ctfpz-RISaUozKjRDuwTLqlMKcoHDzT3m43zk,1073
+drift_ml-0.2.2.dist-info/METADATA,sha256=qJA4kDoZbcJDw98c_82Gi3tvFxmTOH23AUw5-4Ycir0,383
+drift_ml-0.2.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+drift_ml-0.2.2.dist-info/entry_points.txt,sha256=aCY7U9M8nhYj_tIfTXJmYkVmXY3ZoxF0tebDZzYswv8,46
+drift_ml-0.2.2.dist-info/top_level.txt,sha256=3u2KGqsciGZQ2uCoBivm55t3e8er8S4xnqkgdQ_8oeM,6
+drift_ml-0.2.2.dist-info/RECORD,,

drift_ml-0.2.2.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Lakshit Sachdeva
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

drift_ml-0.1.11.dist-info/METADATA

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