drift-ml 0.2.2__tar.gz → 0.2.7__tar.gz

{drift_ml-0.2.2/drift_ml.egg-info → drift_ml-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drift-ml
-Version: 0.2.2
+Version: 0.2.7
 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

{drift_ml-0.2.2 → drift_ml-0.2.7}/drift/__main__.py

@@ -19,7 +19,10 @@ def main() -> None:
         from drift.engine_launcher import ensure_engine
 
         if not ensure_engine():
-            print("drift: Failed to start engine. Check ~/.drift/bin/.engine-stderr.log or set DRIFT_BACKEND_URL", file=sys.stderr)
+            print("drift: Failed to start engine.", file=sys.stderr)
+            print("drift: Check ~/.drift/bin/.engine-stderr.log", file=sys.stderr)
+            print("drift: On macOS: System Settings → Privacy & Security → allow drift-engine", file=sys.stderr)
+            print("drift: Or set DRIFT_BACKEND_URL to a running engine", 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_ml-0.2.2 → drift_ml-0.2.7}/drift/engine_launcher.py

@@ -15,8 +15,8 @@ try:
 except ImportError:
     requests = None
 
-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
+GITHUB_REPO = "lakshitsachdeva/intent2model"  # Engine binaries (same repo)
+ENGINE_TAG = "v0.2.7"  # 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"
 
@@ -119,6 +119,20 @@ def ensure_engine() -> bool:
 
     bin_dir.mkdir(parents=True, exist_ok=True)
 
+    # Re-download if engine tag changed (e.g. after pipx upgrade)
+    version_file = bin_dir / ".engine-tag"
+    if bin_path.exists() and version_file.exists():
+        try:
+            if version_file.read_text().strip() != ENGINE_TAG:
+                bin_path.unlink()
+        except Exception:
+            pass
+    if bin_path.exists() and not version_file.exists():
+        try:
+            bin_path.unlink()
+        except Exception:
+            pass
+
     if not bin_path.exists():
         plat, arch = _get_platform_key()
         ext = ".exe" if platform.system() == "Windows" else ""
@@ -127,6 +141,10 @@ def ensure_engine() -> bool:
         try:
             url = _get_asset_url(asset)
             _download_file(url, bin_path)
+            try:
+                version_file.write_text(ENGINE_TAG)
+            except Exception:
+                pass
         except Exception as e:
             print(f"drift: Download failed: {e}", file=sys.stderr)
             print(f"drift: Run: mkdir -p ~/.drift/bin && curl -L -o ~/.drift/bin/{asset} <url>", file=sys.stderr)
@@ -143,16 +161,40 @@ def ensure_engine() -> bool:
         try:
             bin_path.chmod(0o755)
             subprocess.run(["xattr", "-dr", "com.apple.quarantine", str(bin_path)], check=False, capture_output=True)
+            # Ad-hoc sign so macOS Gatekeeper doesn't kill the binary (requires non-UPX build)
+            subprocess.run(["codesign", "-s", "-", "--force", str(bin_path)], check=False, capture_output=True)
         except Exception:
             pass
 
+    # macOS: use wrapper script via bash to avoid spawn -88
+    # Windows: use batch file so engine starts reliably (inherits PATH for gemini etc.)
+    if platform.system() == "Darwin":
+        wrapper = bin_dir / "run-engine.sh"
+        port = ENGINE_PORT
+        bin_name = bin_path.name
+        script = f'#!/bin/bash\ncd "$(dirname "$0")"\nexport DRIFT_ENGINE_PORT="{port}"\nexec ./{bin_name}\n'
+        if not wrapper.exists() or wrapper.read_text() != script:
+            wrapper.write_text(script)
+            wrapper.chmod(0o755)
+        launch_cmd = ["/bin/bash", str(wrapper)]
+    elif platform.system() == "Windows":
+        bat = bin_dir / "run-engine.bat"
+        port = ENGINE_PORT
+        bin_name = bin_path.name
+        script = f'@echo off\ncd /d "%~dp0"\nset DRIFT_ENGINE_PORT={port}\nstart /b "" {bin_name}\n'
+        if not bat.exists() or bat.read_text() != script:
+            bat.write_text(script)
+        launch_cmd = ["cmd", "/c", str(bat)]
+    else:
+        launch_cmd = [str(bin_path)]
+
     env = {**os.environ, "DRIFT_ENGINE_PORT": ENGINE_PORT}
     stderr_file = bin_dir / ".engine-stderr.log"
     proc = None
     try:
         with open(stderr_file, "w") as errf:
             proc = subprocess.Popen(
-                [str(bin_path)],
+                launch_cmd,
                 cwd=str(bin_dir),
                 env=env,
                 stdout=subprocess.DEVNULL,
@@ -174,7 +216,8 @@ def ensure_engine() -> bool:
         return False
 
     import time
-    for _ in range(60):
+    print("drift: Starting engine (first run may take 30s)...", file=sys.stderr)
+    for i in range(120):  # 60s — PyInstaller binary can take 30+ s to unpack
         if _engine_running():
             return True
         time.sleep(0.5)

{drift_ml-0.2.2 → drift_ml-0.2.7/drift_ml.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drift-ml
-Version: 0.2.2
+Version: 0.2.7
 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

{drift_ml-0.2.2 → drift_ml-0.2.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "drift-ml"
-version = "0.2.2"
+version = "0.2.7"
 description = "drift — terminal-first, chat-based AutoML. Open source. No tokens. No auth."
 requires-python = ">=3.10"
 dependencies = [