drift-ml 0.1.0

package/README.md

@@ -0,0 +1,78 @@
+# drift-ml
+
+**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.
+
+---
+
+## Exactly what to do
+
+1. **Install drift**
+   ```bash
+   npm install -g drift-ml
+   ```
+   Requires Node.js ≥ 18 and Python 3 on your PATH.
+
+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?
+
+- **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.
+
+---
+
+## Install (details)
+
+```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.
+
+### Alternative: pipx (Python only)
+
+```bash
+pipx install drift
+drift
+```
+
+---
+
+## Example usage
+
+```text
+drift › load iris.csv
+drift › predict sepal.length
+drift › try something stronger
+drift › why is accuracy capped
+drift › quit
+```
+
+---
+
+## License
+
+MIT

package/bin/drift.js

@@ -0,0 +1,175 @@
+#!/usr/bin/env node
+
+const { spawnSync, spawn } = require("child_process");
+const fs = require("fs");
+const path = require("path");
+const https = require("https");
+const http = require("http");
+
+const ENGINE_PORT = process.env.DRIFT_ENGINE_PORT || "8000";
+const ENGINE_BASE = process.env.DRIFT_ENGINE_BASE_URL || "https://github.com/lakshitsachdeva/drift/releases/latest/download";
+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; // 30s
+
+function which(cmd) {
+  const { status, stdout } = spawnSync("which", [cmd], { encoding: "utf8" });
+  return status === 0 ? (stdout || "").trim() : null;
+}
+
+function getPlatformKey() {
+  const p = process.platform;
+  const a = process.arch;
+  const plat = p === "darwin" ? "macos" : p === "win32" ? "windows" : "linux";
+  const arch = a === "arm64" || a === "aarch64" ? "arm64" : "x64";
+  return { plat, arch };
+}
+
+function getEngineDir() {
+  const home = process.env.HOME || process.env.USERPROFILE || process.env.HOMEPATH;
+  if (!home) return null;
+  return path.join(home, ".drift", "bin");
+}
+
+function getEnginePath() {
+  const dir = getEngineDir();
+  if (!dir) return null;
+  const { plat, arch } = getPlatformKey();
+  const ext = process.platform === "win32" ? ".exe" : "";
+  return path.join(dir, `drift-engine-${plat}-${arch}${ext}`);
+}
+
+function fetchOk(url) {
+  return new Promise((resolve, reject) => {
+    const client = url.startsWith("https") ? https : http;
+    const req = client.get(url, { timeout: HEALTH_TIMEOUT_MS }, (res) => {
+      const redirect = res.statusCode >= 301 && res.statusCode <= 302 && res.headers.location;
+      if (redirect) {
+        fetchOk(redirect).then(resolve).catch(reject);
+        return;
+      }
+      resolve(res.statusCode === 200);
+    });
+    req.on("error", () => resolve(false));
+    req.on("timeout", () => { req.destroy(); resolve(false); });
+  });
+}
+
+function downloadFile(url, destPath) {
+  return new Promise((resolve, reject) => {
+    const client = url.startsWith("https") ? https : http;
+    const req = client.get(url, (res) => {
+      const redirect = res.statusCode >= 301 && res.statusCode <= 302 && res.headers.location;
+      if (redirect) {
+        downloadFile(redirect, destPath).then(resolve).catch(reject);
+        return;
+      }
+      if (res.statusCode !== 200) {
+        reject(new Error(`Download failed: ${res.statusCode}`));
+        return;
+      }
+      const dir = path.dirname(destPath);
+      if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+      const file = fs.createWriteStream(destPath);
+      res.pipe(file);
+      file.on("finish", () => { file.close(); resolve(); });
+      file.on("error", reject);
+    });
+    req.on("error", reject);
+  });
+}
+
+function engineRunning() {
+  return fetchOk(HEALTH_URL);
+}
+
+function waitForEngine() {
+  return new Promise((resolve) => {
+    let n = 0;
+    const t = setInterval(() => {
+      n++;
+      fetchOk(HEALTH_URL).then((ok) => {
+        if (ok) { clearInterval(t); resolve(true); }
+        else if (n >= HEALTH_POLL_MAX) { clearInterval(t); resolve(false); }
+      });
+    }, HEALTH_POLL_MS);
+  });
+}
+
+async function ensureEngine() {
+  const binPath = getEnginePath();
+  const binDir = getEngineDir();
+  if (!binPath || !binDir) {
+    console.error("drift: Could not resolve engine directory (~/.drift/bin). Set HOME or USERPROFILE.");
+    return false;
+  }
+  if (!fs.existsSync(binPath)) {
+    const { plat, arch } = getPlatformKey();
+    const ext = process.platform === "win32" ? ".exe" : "";
+    const asset = `drift-engine-${plat}-${arch}${ext}`;
+    const url = `${ENGINE_BASE}/${asset}`;
+    process.stderr.write(`drift: Downloading engine (${asset})...\n`);
+    try {
+      await downloadFile(url, binPath);
+    } catch (e) {
+      console.error("drift: Download failed.", e.message);
+      console.error("drift: Run the engine manually or set DRIFT_ENGINE_BASE_URL.");
+      return false;
+    }
+    if (process.platform !== "win32") {
+      try { fs.chmodSync(binPath, 0o755); } catch (_) {}
+    }
+  }
+  const child = spawn(binPath, [], {
+    detached: true,
+    stdio: "ignore",
+    cwd: binDir,
+    env: { ...process.env, DRIFT_ENGINE_PORT: ENGINE_PORT },
+  });
+  child.unref();
+  return waitForEngine();
+}
+
+async function main() {
+  const userBackend = process.env.DRIFT_BACKEND_URL;
+  if (userBackend) {
+    // User runs engine elsewhere; skip download/start and connect to their URL.
+  } else if (await engineRunning()) {
+    // Engine already running locally.
+  } else {
+    const engineStarted = await ensureEngine().catch(() => false);
+    if (!engineStarted) {
+      console.error("drift: Engine did not start. Run the engine manually or set DRIFT_BACKEND_URL to a running engine URL.");
+      process.exit(1);
+    }
+  }
+
+  const python = which("python3") || which("python");
+  if (!python) {
+    console.error("drift: Python is required for the chat CLI. Install Python 3 and ensure `python` or `python3` is on your PATH.");
+    process.exit(1);
+  }
+
+  const pip = spawnSync(python, ["-m", "pip", "install", "--quiet", "--upgrade", "drift"], {
+    encoding: "utf8",
+    stdio: ["inherit", "pipe", "inherit"],
+  });
+  if (pip.status !== 0) {
+    console.error("drift: Failed to install or upgrade the drift Python package.");
+    process.exit(pip.status === null ? 1 : pip.status);
+  }
+
+  const backendUrl = userBackend || `http://127.0.0.1:${ENGINE_PORT}`;
+  const run = spawnSync(python, ["-m", "drift"], {
+    encoding: "utf8",
+    stdio: "inherit",
+    env: { ...process.env, DRIFT_BACKEND_URL: backendUrl },
+  });
+  process.exit(run.status === null ? (run.signal ? 128 + 9 : 1) : run.status);
+}
+
+main().catch((e) => {
+  console.error("drift:", e.message);
+  process.exit(1);
+});

package/package.json

@@ -0,0 +1,13 @@
+{
+  "name": "drift-ml",
+  "version": "0.1.0",
+  "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).",
+  "bin": {
+    "drift": "./bin/drift.js"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "keywords": ["drift", "automl", "cli", "ml", "terminal", "local-first"],
+  "license": "MIT"
+}