drift-ml 0.1.15 → 0.2.1

package/README.md

@@ -1,86 +1,60 @@
-# drift-ml
+# drift
 
-**Drift** by Lakshit Sachdeva — terminal-first, chat-based AutoML. Same engine as the web app. Local-first: the engine runs on your machine; no commands to memorize.
+**Terminal-first, chat-based AutoML.** Open source. No tokens. No auth. Works fully local.
 
 ---
 
-## Exactly what to do
+## Install
 
-1. **Install drift** (pipx recommended — works standalone)
-   ```bash
-   pipx install drift-ml
-   ```
-   Or: `npm install -g drift-ml` (requires `pipx install drift-ml` for the CLI).
-
-2. **Run drift**
-   ```bash
-   drift
-   ```
-   On first run, the CLI detects your OS and architecture, downloads the correct engine binary into `~/.drift/bin/`, and starts the engine in the background. You’ll see a short welcome and instructions in the terminal.
-
-3. **Use a local LLM**
-   Training and planning use an LLM. You need one of:
-   - **Gemini CLI** — install it and set `GEMINI_API_KEY` or have `gemini` on your PATH.
-   - **Ollama** — run `ollama run llama2` (or another model).
-   - Another local LLM the engine supports.
-
-4. **Chat**
-   - `load path/to/data.csv`
-   - `predict price` (or any column)
-   - `try something stronger`
-   - `why is accuracy capped`
-   - `quit` to exit.
-
-That’s it. The engine runs locally. The web app (if you use it) can be hosted on Vercel; the engine stays on your machine.
-
----
-
-## What is drift?
+```bash
+pipx install drift-ml
+```
 
-- **Local-first** — The engine runs on your machine. Training and planning stay local; you never send data to our servers.
-- **Terminal-first, chat-based** — Same engine as the web app. No commands to memorize; chat in natural language.
-- **Engine** — On first run the CLI downloads and starts the engine from `~/.drift/bin/`. Or set `DRIFT_BACKEND_URL` to a running engine URL.
+Or with npm (also requires pipx for the CLI):
 
-**Custom / private repo:** Engine binaries are hosted at [lakshitsachdeva/drift](https://github.com/lakshitsachdeva/drift) (public). For a private fork, set `DRIFT_GITHUB_TOKEN` with a token that has repo read access.
+```bash
+pipx install drift-ml
+npm install -g drift-ml
+```
 
 ---
 
-## Install (details)
+## Run
 
 ```bash
-npm install -g drift-ml
 drift
 ```
 
-The `drift` command installs or upgrades the chat CLI (Python) and runs it. You get the welcome and instructions every time.
+That's it. On first run, drift downloads and starts the engine automatically. No backend setup. No config.
 
-### Alternative: pipx (Python only — macOS, Linux, Windows)
+---
 
-```bash
-pipx install drift-ml
-drift
-```
+## How it works
 
-**Update (pipx):**
-```bash
-pipx upgrade drift-ml
-```
-(PowerShell on Windows: same command.)
+- **Local-first** — Engine runs on your machine. Data never leaves.
+- **Chat-based** — `load data.csv`, `predict price`, `try something stronger`
+- **Auto-start** — Engine downloads and starts in the background. You never touch it.
+- **No tokens** — No API keys for drift. (You need an LLM for training: Ollama, Gemini CLI, etc.)
 
 ---
 
-## Example usage
+## Example
 
 ```text
 drift › load iris.csv
-drift › predict sepal.length
+drift › predict variety
 drift › try something stronger
-drift › why is accuracy capped
 drift › quit
 ```
 
 ---
 
+## Philosophy
+
+drift should feel like `git`, `docker`, `brew` — a tool you trust immediately. Zero friction. Open source.
+
+---
+
 ## License
 
 MIT

package/bin/drift.js

@@ -1,11 +1,7 @@
 #!/usr/bin/env node
 //
-// REMOVED: Any pip/pipx/python -m pip install or upgrade logic.
-// WHY: PEP 668 and user envs; we must not modify Python. User installs CLI via pipx once.
-//
-// FINAL FLOW: (1) If no DRIFT_BACKEND_URL, ensure engine at 127.0.0.1:8000 (download + start if needed).
-//             (2) Locate Python drift ONLY in ~/.local/bin (macOS/Linux) or %USERPROFILE%\.local\bin (Windows).
-//             (3) Spawn that binary with DRIFT_BACKEND_URL set; never run drift.cmd/drift.ps1/drift.js (self).
+// drift — terminal-first AutoML. Engine auto-starts locally.
+// No tokens. No auth. No backend setup.
 //
 
 const { spawnSync, spawn } = require("child_process");
@@ -16,16 +12,13 @@ const http = require("http");
 
 const isWindows = process.platform === "win32";
 const ENGINE_PORT = process.env.DRIFT_ENGINE_PORT || "8000";
-// Pinned tag: draft releases are invisible to /releases/latest.
-const ENGINE_TAG = "v0.1.3";
-const GITHUB_REPO = "lakshitsachdeva/drift";
-const ENGINE_BASE_URL = process.env.DRIFT_ENGINE_BASE_URL || `https://github.com/${GITHUB_REPO}/releases/download/${ENGINE_TAG}`;
-const GITHUB_API_RELEASE = `https://api.github.com/repos/${GITHUB_REPO}/releases/tags/${ENGINE_TAG}`;
-const GITHUB_TOKEN = process.env.DRIFT_GITHUB_TOKEN || process.env.GITHUB_TOKEN;
+const GITHUB_REPO = "lakshitsachdeva/drift";  // Engine binaries
+const ENGINE_TAG = "v0.2.0";  // Pinned — direct URL, no API, no rate limits
+const ENGINE_BASE_URL = `https://github.com/${GITHUB_REPO}/releases/download/${ENGINE_TAG}`;
 const HEALTH_URL = `http://127.0.0.1:${ENGINE_PORT}/health`;
 const HEALTH_TIMEOUT_MS = 2000;
 const HEALTH_POLL_MS = 500;
-const HEALTH_POLL_MAX = 60; // 30 seconds total
+const HEALTH_POLL_MAX = 60;
 const isMac = process.platform === "darwin";
 
 function getPlatformKey() {
@@ -66,55 +59,32 @@ function fetchOk(url) {
   });
 }
 
-const API_HEADERS = {
-  "User-Agent": "Drift-Engine-Launcher/1.0",
-  Accept: "application/vnd.github+json",
-  ...(GITHUB_TOKEN && { Authorization: `Bearer ${GITHUB_TOKEN}` }),
-};
-const DOWNLOAD_HEADERS = {
-  "User-Agent": "Drift-Engine-Launcher/1.0",
-  Accept: "application/octet-stream",
-  ...(GITHUB_TOKEN && { Authorization: `Bearer ${GITHUB_TOKEN}` }),
-};
+function getAssetUrl(assetName) {
+  const baseUrl = process.env.DRIFT_ENGINE_BASE_URL;
+  if (baseUrl) {
+    return `${baseUrl.replace(/\/$/, "")}/${assetName}`;
+  }
+  return `${ENGINE_BASE_URL}/${assetName}`;
+}
 
-// Resolve GitHub release asset to API download URL (browser_download_url returns 404 for scripts).
-function getGitHubAssetUrl(assetName) {
+function downloadWithCurl(url, destPath) {
   return new Promise((resolve, reject) => {
-    const req = https.get(GITHUB_API_RELEASE, { headers: API_HEADERS }, (res) => {
-      if (res.statusCode === 404) {
-        reject(new Error(
-          "Release not found (404). If the repo is private, set DRIFT_GITHUB_TOKEN with a token that has repo read access."
-        ));
-        return;
-      }
-      if (res.statusCode !== 200) {
-        reject(new Error(`Release not found: ${res.statusCode}`));
-        return;
-      }
-      let body = "";
-      res.on("data", (chunk) => { body += chunk; });
-      res.on("end", () => {
-        try {
-          const data = JSON.parse(body);
-          const asset = (data.assets || []).find((a) => a.name === assetName);
-          if (!asset || !asset.url) {
-            reject(new Error(`Asset not found: ${assetName}`));
-            return;
-          }
-          resolve(asset.url);
-        } catch (e) {
-          reject(e);
-        }
-      });
+    const result = spawnSync("curl", ["-fsSL", "-o", destPath, url], {
+      stdio: "pipe",
+      timeout: 120000,
     });
-    req.on("error", reject);
+    if (result.status !== 0) {
+      reject(new Error("Download failed"));
+      return;
+    }
+    resolve();
   });
 }
 
 function downloadFile(url, destPath) {
   return new Promise((resolve, reject) => {
     const client = url.startsWith("https") ? https : http;
-    const req = client.get(url, { headers: DOWNLOAD_HEADERS }, (res) => {
+    const req = client.get(url, { headers: { "User-Agent": "Drift/1.0" } }, (res) => {
       const redirect = res.statusCode >= 301 && res.statusCode <= 302 && res.headers.location;
       if (redirect) {
         downloadFile(redirect, destPath).then(resolve).catch(reject);
@@ -166,16 +136,12 @@ async function ensureEngine() {
     const { plat, arch } = getPlatformKey();
     const ext = isWindows ? ".exe" : "";
     const asset = `drift-engine-${plat}-${arch}${ext}`;
-    const isDefaultGitHub = !process.env.DRIFT_ENGINE_BASE_URL;
-    const url = isDefaultGitHub
-      ? await getGitHubAssetUrl(asset)
-      : `${ENGINE_BASE_URL.replace(/\/$/, "")}/${asset}`;
+    const url = getAssetUrl(asset);
     process.stderr.write(`drift: Downloading engine (${asset})...\n`);
     try {
-      await downloadFile(url, binPath);
+      await downloadWithCurl(url, binPath).catch(() => downloadFile(url, binPath));
     } catch (e) {
       console.error("drift: Download failed.", e.message);
-      console.error("drift: Set DRIFT_ENGINE_BASE_URL or run the engine manually.");
       return false;
     }
     if (!isWindows) {
@@ -187,7 +153,6 @@ async function ensureEngine() {
       } catch (_) {}
     }
   }
-  // On macOS, always ensure binary is executable and not quarantined before spawn (covers existing binaries).
   if (isMac && binPath) {
     try {
       fs.chmodSync(binPath, 0o755);
@@ -204,7 +169,6 @@ async function ensureEngine() {
   return waitForEngine();
 }
 
-// Locate Python drift ONLY in pipx bin dir. Never search PATH (avoids running drift.cmd/drift.ps1/drift.js).
 function findPythonDrift() {
   const home = process.env.HOME || process.env.USERPROFILE || "";
   if (!home) return null;
@@ -230,7 +194,6 @@ async function main() {
       });
       if (!started) {
         console.error("Failed to start drift engine.");
-        console.error("Please check permissions or download the engine manually.");
         process.exit(1);
       }
     }
@@ -239,9 +202,7 @@ async function main() {
   const driftPath = findPythonDrift();
   if (!driftPath) {
     console.error(`
-drift is not installed.
-
-Install it with:
+drift is not installed. Install it with:
 
   pipx install drift-ml
 `);

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "drift-ml",
-  "version": "0.1.15",
-  "description": "Drift — terminal-first, chat-based AutoML. Same engine as the web app. On first run: downloads and starts the engine locally (never exposes engine source).",
+  "version": "0.2.1",
+  "description": "drift — terminal-first, chat-based AutoML. Open source. No tokens. No auth.",
   "bin": {
     "drift": "bin/drift.js"
   },
@@ -9,7 +9,7 @@
   "engines": {
     "node": ">=18"
   },
-  "keywords": ["drift", "automl", "cli", "ml", "terminal", "local-first"],
+  "keywords": ["drift", "automl", "cli", "ml", "terminal", "local-first", "ai", "open-source"],
   "license": "MIT",
   "author": "Lakshit Sachdeva",
   "repository": {