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

drift_ml-0.2.8/PKG-INFO

@@ -0,0 +1,129 @@
+Metadata-Version: 2.4
+Name: drift-ml
+Version: 0.2.8
+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
+Keywords: automl,ml,machine-learning,cli,chat,local-first
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: requests>=2.28.0
+Dynamic: license-file
+
+# drift
+
+**Terminal-first, chat-based AutoML.** Open source. No tokens. No auth.
+
+---
+
+## Install
+
+```bash
+pipx install drift-ml
+```
+
+Or:
+
+```bash
+npm install -g drift-ml
+```
+
+(Both require `pipx install drift-ml` for the Python CLI.)
+
+---
+
+## Run
+
+```bash
+drift
+```
+
+On first run, drift downloads and starts the engine automatically. No backend setup. No config. No tokens.
+
+---
+
+## How it works
+
+- **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 start a backend manually.
+- **No auth** — No API keys for drift. (You need an LLM for training: Ollama, Gemini CLI, etc.)
+
+---
+
+## Example (CLI)
+
+```text
+drift › load iris.csv
+drift › predict variety
+drift › try something stronger
+drift › quit
+```
+
+---
+
+## Use as library
+
+```bash
+pip install drift-ml
+```
+
+```python
+from drift import Drift
+
+d = Drift()
+d.load("iris.csv")
+d.chat("predict sepal length")
+result = d.train()
+print(result["metrics"])
+```
+
+Or with an existing engine:
+
+```python
+from drift import Drift
+
+d = Drift(base_url="http://localhost:8000")
+d.load("data.csv")
+reply = d.chat("predict price")
+print(d.get_last_reply(reply))
+```
+
+---
+
+## Philosophy
+
+drift should feel like `git`, `docker`, `brew` — a tool you trust immediately. Zero friction. Open source.
+
+---
+
+## Web UI (optional)
+
+This repo also includes a web UI. For development:
+
+```bash
+./start.sh
+```
+
+Then open http://localhost:3000
+
+---
+
+## Contributing
+
+See [CONTRIBUTING.md](CONTRIBUTING.md).
+
+---
+
+## License
+
+MIT

{drift_ml-0.2.7 → drift_ml-0.2.8}/README.md

@@ -39,7 +39,7 @@ On first run, drift downloads and starts the engine automatically. No backend se
 
 ---
 
-## Example
+## Example (CLI)
 
 ```text
 drift › load iris.csv
@@ -50,6 +50,35 @@ drift › quit
 
 ---
 
+## Use as library
+
+```bash
+pip install drift-ml
+```
+
+```python
+from drift import Drift
+
+d = Drift()
+d.load("iris.csv")
+d.chat("predict sepal length")
+result = d.train()
+print(result["metrics"])
+```
+
+Or with an existing engine:
+
+```python
+from drift import Drift
+
+d = Drift(base_url="http://localhost:8000")
+d.load("data.csv")
+reply = d.chat("predict price")
+print(d.get_last_reply(reply))
+```
+
+---
+
 ## Philosophy
 
 drift should feel like `git`, `docker`, `brew` — a tool you trust immediately. Zero friction. Open source.

drift_ml-0.2.8/drift/__init__.py

@@ -0,0 +1,25 @@
+"""
+drift — terminal-first, chat-based AutoML. Use as CLI or library.
+
+CLI:
+    pip install drift-ml
+    drift
+    drift › load data.csv
+    drift › predict price
+
+Library:
+    from drift import Drift
+
+    d = Drift()
+    d.load("data.csv")
+    d.chat("predict price")
+    result = d.train()
+"""
+
+__version__ = "0.2.8"
+
+from drift.api import Drift
+from drift.cli.client import BackendClient, BackendError
+from drift.engine_launcher import ensure_engine
+
+__all__ = ["Drift", "BackendClient", "BackendError", "ensure_engine", "__version__"]

drift_ml-0.2.8/drift/api.py

@@ -0,0 +1,116 @@
+"""
+Programmatic API for drift — use in your own Python scripts.
+
+Example:
+    from drift import Drift
+
+    d = Drift()
+    d.load("data.csv")
+    d.chat("predict price")
+    result = d.train()
+    print(result["metrics"])
+"""
+
+import os
+from typing import Any, Dict, Optional
+
+from drift.cli.client import BackendClient, BackendError
+
+
+class Drift:
+    """
+    High-level client for drift. Load data, chat, train — in your code.
+
+    Uses a local engine (auto-started if needed) or DRIFT_BACKEND_URL.
+    """
+
+    def __init__(self, base_url: Optional[str] = None):
+        """
+        Args:
+            base_url: Engine URL. If None, uses DRIFT_BACKEND_URL or auto-starts engine.
+        """
+        if base_url:
+            self._client = BackendClient(base_url=base_url)
+        else:
+            url = os.environ.get("DRIFT_BACKEND_URL")
+            if not url:
+                from drift.engine_launcher import ensure_engine
+
+                if not ensure_engine():
+                    raise BackendError(
+                        "Could not start engine. Set DRIFT_BACKEND_URL or check ~/.drift/bin/.engine-stderr.log"
+                    )
+                port = os.environ.get("DRIFT_ENGINE_PORT", "8000")
+                url = f"http://127.0.0.1:{port}"
+            self._client = BackendClient(base_url=url)
+        self._session_id: Optional[str] = None
+        self._dataset_id: Optional[str] = None
+
+    def load(self, path: str) -> "Drift":
+        """
+        Load a CSV file. Returns self for chaining.
+
+        Args:
+            path: Path to CSV file.
+
+        Returns:
+            self
+        """
+        out = self._client.upload_csv(path)
+        self._session_id = out["session_id"]
+        self._dataset_id = out["dataset_id"]
+        return self
+
+    def chat(self, message: str) -> Dict[str, Any]:
+        """
+        Send a chat message. Returns chat_history and trigger_training.
+
+        Args:
+            message: Natural language, e.g. "predict price", "try something stronger".
+
+        Returns:
+            {"chat_history": [...], "trigger_training": bool}
+        """
+        if not self._session_id:
+            raise BackendError("Load a dataset first: drift.load('data.csv')")
+        return self._client.chat(self._session_id, message)
+
+    def train(self) -> Dict[str, Any]:
+        """
+        Run one training attempt. Blocking.
+
+        Returns:
+            run_id, metrics, refused, agent_message, etc.
+        """
+        if not self._session_id:
+            raise BackendError("Load a dataset first: drift.load('data.csv')")
+        return self._client.train(self._session_id)
+
+    def get_last_reply(self, chat_result: Optional[Dict[str, Any]] = None) -> Optional[str]:
+        """Get the last agent reply from chat_history."""
+        hist = (chat_result or {}).get("chat_history") if chat_result else None
+        if not hist and self._session_id:
+            sess = self._client.get_session(self._session_id)
+            hist = sess.get("chat_history") or []
+        if not hist:
+            return None
+        for m in reversed(hist):
+            if m.get("role") == "agent" and m.get("content"):
+                return m["content"]
+        return None
+
+    def download_notebook(self, run_id: str) -> Optional[bytes]:
+        """Download training notebook as bytes."""
+        return self._client.download_notebook(run_id)
+
+    def download_model(self, run_id: str) -> Optional[bytes]:
+        """Download trained model pickle as bytes."""
+        return self._client.download_model(run_id)
+
+    def health(self) -> Dict[str, Any]:
+        """Check engine health and LLM status."""
+        return self._client.health()
+
+    @property
+    def session_id(self) -> Optional[str]:
+        return self._session_id

{drift_ml-0.2.7 → drift_ml-0.2.8}/drift/engine_launcher.py

@@ -16,7 +16,7 @@ except ImportError:
     requests = None
 
 GITHUB_REPO = "lakshitsachdeva/intent2model"  # Engine binaries (same repo)
-ENGINE_TAG = "v0.2.7"  # Pinned — direct URL, no API, no rate limits
+ENGINE_TAG = "v0.2.8"  # 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"
 

drift_ml-0.2.8/drift_ml.egg-info/PKG-INFO

@@ -0,0 +1,129 @@
+Metadata-Version: 2.4
+Name: drift-ml
+Version: 0.2.8
+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
+Keywords: automl,ml,machine-learning,cli,chat,local-first
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: requests>=2.28.0
+Dynamic: license-file
+
+# drift
+
+**Terminal-first, chat-based AutoML.** Open source. No tokens. No auth.
+
+---
+
+## Install
+
+```bash
+pipx install drift-ml
+```
+
+Or:
+
+```bash
+npm install -g drift-ml
+```
+
+(Both require `pipx install drift-ml` for the Python CLI.)
+
+---
+
+## Run
+
+```bash
+drift
+```
+
+On first run, drift downloads and starts the engine automatically. No backend setup. No config. No tokens.
+
+---
+
+## How it works
+
+- **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 start a backend manually.
+- **No auth** — No API keys for drift. (You need an LLM for training: Ollama, Gemini CLI, etc.)
+
+---
+
+## Example (CLI)
+
+```text
+drift › load iris.csv
+drift › predict variety
+drift › try something stronger
+drift › quit
+```
+
+---
+
+## Use as library
+
+```bash
+pip install drift-ml
+```
+
+```python
+from drift import Drift
+
+d = Drift()
+d.load("iris.csv")
+d.chat("predict sepal length")
+result = d.train()
+print(result["metrics"])
+```
+
+Or with an existing engine:
+
+```python
+from drift import Drift
+
+d = Drift(base_url="http://localhost:8000")
+d.load("data.csv")
+reply = d.chat("predict price")
+print(d.get_last_reply(reply))
+```
+
+---
+
+## Philosophy
+
+drift should feel like `git`, `docker`, `brew` — a tool you trust immediately. Zero friction. Open source.
+
+---
+
+## Web UI (optional)
+
+This repo also includes a web UI. For development:
+
+```bash
+./start.sh
+```
+
+Then open http://localhost:3000
+
+---
+
+## Contributing
+
+See [CONTRIBUTING.md](CONTRIBUTING.md).
+
+---
+
+## License
+
+MIT

{drift_ml-0.2.7 → drift_ml-0.2.8}/drift_ml.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ README.md
 pyproject.toml
 drift/__init__.py
 drift/__main__.py
+drift/api.py
 drift/engine_launcher.py
 drift/cli/__init__.py
 drift/cli/client.py

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

@@ -4,12 +4,24 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "drift-ml"
-version = "0.2.7"
+version = "0.2.8"
 description = "drift — terminal-first, chat-based AutoML. Open source. No tokens. No auth."
 requires-python = ">=3.10"
 dependencies = [
     "requests>=2.28.0",
 ]
+readme = "README.md"
+keywords = ["automl", "ml", "machine-learning", "cli", "chat", "local-first"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+]
 
 [project.urls]
 Homepage = "https://github.com/lakshitsachdeva/intent2model"

drift_ml-0.2.7/PKG-INFO

@@ -1,10 +0,0 @@
-Metadata-Version: 2.4
-Name: drift-ml
-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
-Requires-Python: >=3.10
-License-File: LICENSE
-Requires-Dist: requests>=2.28.0
-Dynamic: license-file

drift_ml-0.2.7/drift/__init__.py

@@ -1,5 +0,0 @@
-"""
-drift — terminal-first, chat-based CLI for the same AutoML planner + executor as the web app.
-"""
-
-__version__ = "0.1.0"

drift_ml-0.2.7/drift_ml.egg-info/PKG-INFO

@@ -1,10 +0,0 @@
-Metadata-Version: 2.4
-Name: drift-ml
-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
-Requires-Python: >=3.10
-License-File: LICENSE
-Requires-Dist: requests>=2.28.0
-Dynamic: license-file