drift-ml 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

drift/__main__.py

@@ -1,13 +1,24 @@
 """
 Entry point for `python -m drift` or `drift` command.
-Runs the chat-based REPL.
+Runs the chat-based REPL. If no backend is running, downloads and starts the engine.
 """
 
+import os
+import sys
+
 from drift.cli.repl import run_repl
 
 
 def main() -> None:
-    run_repl()
+    base_url = os.environ.get("DRIFT_BACKEND_URL")
+    if not base_url:
+        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)
+            sys.exit(1)
+        base_url = f"http://127.0.0.1:{os.environ.get('DRIFT_ENGINE_PORT', '8000')}"
+    run_repl(base_url=base_url)
 
 
 if __name__ == "__main__":

drift/engine_launcher.py

@@ -0,0 +1,167 @@
+"""
+Download and start the drift engine binary when no backend is running.
+Makes pipx install drift-ml work standalone (no npm needed).
+"""
+
+import os
+import platform
+import shutil
+import subprocess
+import sys
+from pathlib import Path
+from typing import Optional, Tuple
+
+try:
+    import requests
+except ImportError:
+    requests = None
+
+ENGINE_TAG = "v0.1.3"
+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")
+HEALTH_URL = f"http://127.0.0.1:{ENGINE_PORT}/health"
+
+
+def _get_platform_key() -> Tuple[str, str]:
+    """Return (plat, arch) e.g. ('macos', 'arm64')."""
+    p = platform.system().lower()
+    a = platform.machine().lower()
+    plat = "macos" if p == "darwin" else "windows" if p == "windows" else "linux"
+    arch = "arm64" if a in ("arm64", "aarch64") else "x64"
+    if p == "windows" and "amd64" in a:
+        arch = "x64"
+    return plat, arch
+
+
+def _get_engine_dir() -> Optional[Path]:
+    home = os.environ.get("HOME") or os.environ.get("USERPROFILE")
+    if not home:
+        return None
+    return Path(home) / ".drift" / "bin"
+
+
+def _get_engine_path() -> Optional[Path]:
+    d = _get_engine_dir()
+    if not d:
+        return None
+    plat, arch = _get_platform_key()
+    ext = ".exe" if platform.system() == "Windows" else ""
+    return d / f"drift-engine-{plat}-{arch}{ext}"
+
+
+def _engine_running() -> bool:
+    if not requests:
+        return False
+    try:
+        r = requests.get(HEALTH_URL, timeout=2)
+        return r.status_code == 200
+    except Exception:
+        return False
+
+
+def _get_asset_download_url(asset_name: str) -> str:
+    """Resolve GitHub release asset to download URL."""
+    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:
+            # Prefer browser_download_url for public repos (no auth needed)
+            url = a.get("browser_download_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}")
+
+
+def _download_file(url: str, dest: Path) -> None:
+    token = os.environ.get("DRIFT_GITHUB_TOKEN") or os.environ.get("GITHUB_TOKEN")
+    headers = {"User-Agent": "Drift-Engine-Launcher/1.0"}
+    # API URLs need Accept: application/octet-stream; browser_download_url works with default
+    if "api.github.com" in url:
+        headers["Accept"] = "application/octet-stream"
+    if token:
+        headers["Authorization"] = f"Bearer {token}"
+    r = requests.get(url, headers=headers, stream=True, timeout=60)
+    r.raise_for_status()
+    dest.parent.mkdir(parents=True, exist_ok=True)
+    with open(dest, "wb") as f:
+        for chunk in r.iter_content(chunk_size=65536):
+            f.write(chunk)
+
+
+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.
+    """
+    if _engine_running():
+        return True
+
+    if not requests:
+        print("drift: 'requests' required for engine download. 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)
+        return False
+
+    bin_dir.mkdir(parents=True, exist_ok=True)
+
+    if not bin_path.exists():
+        plat, arch = _get_platform_key()
+        ext = ".exe" if platform.system() == "Windows" else ""
+        asset = f"drift-engine-{plat}-{arch}{ext}"
+        print(f"drift: Downloading engine ({asset})...", file=sys.stderr)
+        try:
+            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 platform.system() != "Windows":
+            bin_path.chmod(0o755)
+        if platform.system() == "Darwin":
+            try:
+                subprocess.run(["xattr", "-dr", "com.apple.quarantine", str(bin_path)], check=False, capture_output=True)
+            except Exception:
+                pass
+
+    # Start engine
+    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
+    for _ in range(60):
+        if _engine_running():
+            return True
+        import time
+        time.sleep(0.5)
+    return False

{drift_ml-0.1.5.dist-info → drift_ml-0.1.6.dist-info}/METADATA

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

{drift_ml-0.1.5.dist-info → drift_ml-0.1.6.dist-info}/RECORD

@@ -1,5 +1,6 @@
 drift/__init__.py,sha256=X0NUP5ZAZSz-rBFfjvmmS4IYWP2_CZtu187mqwpWwqk,127
-drift/__main__.py,sha256=MMUjNUbctbLHVe37ZCOW-66h8OZ8WIYiISxHJHRNxes,202
+drift/__main__.py,sha256=RnBMIqpDEMHknOKn-s5Q2MTUWm_UsKSkYuOArzShQeo,693
+drift/engine_launcher.py,sha256=mF1yNDJMV2ccWGUJNIxTIV8rpIOYXQowm2m0_hw2_5Q,5462
 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
@@ -8,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.5.dist-info/METADATA,sha256=DxTrkgNSt-kykg1ioaoR37V-A4MpvKv40lbbRG7Y8Wg,168
-drift_ml-0.1.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-drift_ml-0.1.5.dist-info/entry_points.txt,sha256=aCY7U9M8nhYj_tIfTXJmYkVmXY3ZoxF0tebDZzYswv8,46
-drift_ml-0.1.5.dist-info/top_level.txt,sha256=3u2KGqsciGZQ2uCoBivm55t3e8er8S4xnqkgdQ_8oeM,6
-drift_ml-0.1.5.dist-info/RECORD,,
+drift_ml-0.1.6.dist-info/METADATA,sha256=CIj7-l0RadnHdDOeKRPnH2RL-_MWfFzvQH-7TCUmWZA,168
+drift_ml-0.1.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+drift_ml-0.1.6.dist-info/entry_points.txt,sha256=aCY7U9M8nhYj_tIfTXJmYkVmXY3ZoxF0tebDZzYswv8,46
+drift_ml-0.1.6.dist-info/top_level.txt,sha256=3u2KGqsciGZQ2uCoBivm55t3e8er8S4xnqkgdQ_8oeM,6
+drift_ml-0.1.6.dist-info/RECORD,,