sage-ai-cli 0.1.0__py3-none-any.whl

sage_ai_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,214 @@
+Metadata-Version: 2.4
+Name: sage-ai-cli
+Version: 0.1.0
+Summary: SageCLI - AI Engineering Agent for Machine Learning & Deep Learning
+Author-email: zaheerjklabs <zaheerjklabs@gmail.com>
+License: MIT
+Keywords: ai,cli,machine-learning,deep-learning,agent,engineering
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: rich>=13.0.0
+Requires-Dist: typer>=0.9.0
+Requires-Dist: prompt-toolkit>=3.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Dynamic: license-file
+
+# SageCLI
+
+```text
+ ███████╗    █████╗     ██████╗    ███████╗
+ ██╔════╝   ██╔══██╗   ██╔════╝    ██╔════╝
+ ███████╗   ███████║   ██║  ███╗   █████╗
+ ╚════██║   ██╔══██║   ██║   ██║   ██╔══╝
+ ███████║   ██║  ██║   ╚██████╔╝   ███████╗
+ ╚══════╝   ╚═╝  ╚═╝    ╚═════╝    ╚══════╝
+
+              S A G E C L I
+        AI ENGINEERING AGENT
+```
+
+> **Build • Train • Debug • Evaluate • Deploy**
+
+SageCLI is a terminal-native autonomous AI coding agent designed specifically for **Machine Learning, Deep Learning, and AI Engineering**. Running directly inside your terminal, SageCLI autonomously inspects repositories and datasets, plans modular architectures, writes and executes code, runs tests, self-debugs tracebacks, computes metrics, and sets up inference endpoints.
+
+---
+
+## 🚀 Autonomous Agent Workflow
+
+```text
+User Prompt
+    ↓
+Context / Repository & Dataset Ingestion
+    ↓
+Planning & Architecture Phase
+    ↓
+Code Generation (PyTorch, Scikit-Learn, LightGBM, FastAPI)
+    ↓
+Execution & Pipeline Run
+    ↓
+Automated Testing & Autonomous Traceback Debugging
+    ↓
+Model Evaluation & Metric Tracking
+    ↓
+Documentation & Git Checkpoints
+```
+
+---
+
+## 🛠️ Key Capabilities
+
+* **First-Run Interactive Setup Wizard (`sage setup`)**: Multi-provider auto-detection, masked credential entry, model selection, mode choice, and live connection verification.
+* **Provider-Agnostic Engine**: Native tool-calling support for:
+  * **Google Gemini** (`gemini-2.5-pro`, `gemini-2.5-flash`, `gemini-2.0-flash`)
+  * **OpenAI** (`gpt-4o`, `gpt-4o-mini`, `o3-mini`, `o1`)
+  * **Anthropic Claude** (`claude-3-7-sonnet`, `claude-3-5-sonnet`, `claude-3-5-haiku`)
+  * **Groq** (`llama-3.3-70b-versatile`, `deepseek-r1-distill-llama-70b`)
+  * **OpenRouter** (`deepseek/deepseek-r1`, etc.)
+  * **Ollama (Local)** (`llama3.2`, `deepseek-r1`, `qwen2.5-coder`)
+  * **Custom OpenAI-compatible endpoints** (vLLM, LocalAI, LM Studio)
+* **ML/DL Engineering Tool Suite**:
+  * `inspect_dataset`: Deep EDA on CSV, Parquet, JSON, NumPy, Excel files (shapes, missing values, statistics, class distribution).
+  * `read_file` / `write_file` / `patch_file` / `list_directory`: Surgical and sandboxed workspace file operations.
+  * `execute_command`: Shell command and Python script execution with timeouts and security checks.
+  * `run_tests`: Pytest runner with failure analysis and traceback extraction.
+  * `git_checkpoint` / `git_diff` / `git_rollback`: Version control safety and rollback support.
+* **Self-Healing & Autonomous Debugging**: When scripts fail with errors (e.g., shape mismatches, missing modules, runtime exceptions), SageCLI intercepts the stderr/stack trace, diagnoses the cause, patches the code, and re-executes until verified.
+* **Execution Modes**:
+  * `Safe` (Default) — Requests user approval before running write or shell operations.
+  * `Auto` — Autonomously executes safe workspace steps, prompting only for destructive actions.
+  * `Plan` — Generates comprehensive plans and diffs without modifying files on disk.
+* **State & Memory Management**: Persists conversation history, task memory, dataset metadata, and ML experiment metrics in `.sage/` (automatically added to `.gitignore`).
+
+---
+
+## 📦 Installation
+
+### From PyPI
+```bash
+pip install sage-ai-cli
+```
+
+### From Source (Development)
+```bash
+# Clone the repository
+git clone https://github.com/zaheerjklabs/SageCLI.git
+cd SageCLI
+
+# Install in editable mode
+pip install -e .
+```
+
+---
+
+## ⚡ Quickstart
+
+### 1. Launch SageCLI
+```bash
+sage
+```
+On first launch, the interactive setup wizard will guide you through provider selection and credential validation.
+
+### 2. Interactive Engineering Prompt
+```text
+sage ❯ Build a customer churn prediction system using data/raw/customers.csv with XGBoost and a FastAPI endpoint.
+```
+
+SageCLI will:
+1. 🔍 **Inspect** `data/raw/customers.csv` and report feature distributions.
+2. 📐 **Plan** modular components (`src/data.py`, `src/model.py`, `src/train.py`, `src/app.py`).
+3. 💻 **Implement** clean preprocessing, feature engineering, and model training.
+4. ⚡ **Execute** the pipeline and self-debug any runtime errors.
+5. 📊 **Evaluate** ROC-AUC, Precision/Recall, and log metrics.
+6. 🚀 **Deliver** test suites and a FastAPI inference service.
+
+---
+
+## 🎮 Interactive Slash Commands (`/`)
+
+SageCLI provides an intuitive, Claude Code-inspired command palette with autocompletion and instant dropdown help:
+
+| Command | Action | Description |
+| :--- | :--- | :--- |
+| `/help` | Detailed help manual | View all commands, shortcuts, and capabilities |
+| `/setup` | Setup wizard | Interactive multi-provider and API key configuration |
+| `/config` | View/edit settings | Show active configuration or update with `/config set <key> <val>` |
+| `/mode <safe\|auto\|plan>` | Switch execution mode | `Safe` (confirm writes), `Auto` (full autonomous), `Plan` (read-only) |
+| `/model [name]` | Switch LLM model | Switch model or view recommended options |
+| `/provider [name]` | Switch AI provider | `gemini`, `openai`, `anthropic`, `groq`, `openrouter`, `ollama` |
+| `/doctor` | Environment diagnostics | Inspect Python, GPU/CUDA, Git repo, disk space, and API latency |
+| `/cost` / `/usage` | Token & cost tracking | Display session token usage and estimated USD cost breakdown |
+| `/context` | Workspace summary | Review detected directory tree, datasets, and ML stack |
+| `/memory` | Session task memory | Display or reset active task history and persistent state |
+| `/dataset <path>` | EDA & Data Profiling | Analyze dataset shape, missing values, statistics, and distributions |
+| `/diff` | Git diff | View uncommitted code modifications |
+| `/checkpoint <msg>` | Git checkpoint | Create an immediate snapshot commit |
+| `/rollback` | Git rollback | Revert workspace changes safely |
+| `/compact` | Memory compaction | Compact session history to save token context window |
+| `/ascii` | Toggle ASCII mode | Switch between Unicode and pure ASCII terminal glyphs |
+| `/demo` | Status language demo | Preview SageCLI's branded status symbols and styling |
+| `/clear` | Clear screen | Clear terminal view while preserving session state |
+| `/exit` / `/quit` | Exit | Exit SageCLI session |
+
+---
+
+## 🏗️ Architecture
+
+```text
+src/sagecli/
+├── __init__.py           # Package version and brand definitions
+├── __main__.py           # python -m sagecli entry point
+├── cli.py                # Typer application, CLI flags, and interactive REPL
+├── config.py             # Multi-provider hierarchical configuration
+├── security.py           # Secret masking, sandboxing, and safety checks
+├── wizard.py             # First-run interactive setup wizard
+├── core/
+│   ├── agent.py          # Autonomous agent execution loop
+│   ├── state.py          # Project memory, task history, and metrics in .sage/
+│   ├── context.py        # Workspace scanner, dataset detector, and environment specs
+│   └── orchestrator.py   # Multi-role prompt engineering for ML/DL pipelines
+├── llm/
+│   ├── base.py           # Provider interface and message models
+│   ├── factory.py        # Dynamic provider factory
+│   ├── gemini_provider.py# Google Gemini provider
+│   ├── openai_provider.py# OpenAI / Groq / OpenRouter / Ollama provider
+│   └── anthropic_provider.py # Anthropic Claude provider
+├── tools/
+│   ├── base.py           # Tool base class and schemas
+│   ├── registry.py       # Tool registry and permission dispatcher
+│   ├── file_ops.py       # Read, write, patch, list files
+│   ├── shell_ops.py      # Terminal command and script execution
+│   ├── dataset_ops.py    # Exploratory data analysis & schema inspection
+│   ├── test_eval_ops.py  # Pytest execution and failure parser
+│   └── git_ops.py        # Git checkpoints, diffs, and rollbacks
+└── ui/
+    ├── banner.py         # Startup banner, splash, metadata table
+    ├── console.py        # Branded visual language formatters
+    ├── logo.py           # ASCII logo and responsive renderers
+    ├── prompts.py        # Interactive prompt_toolkit session
+    ├── symbols.py        # Unicode & ASCII symbols
+    └── theme.py          # SageCLI palette and Rich styles
+```
+
+---
+
+## 🧪 Testing
+
+Run the full automated test suite with:
+
+```bash
+python3 -m pytest -v
+```

sage_ai_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,40 @@
+sage_ai_cli-0.1.0.dist-info/licenses/LICENSE,sha256=di-fUu_7bi14Zi_5tYq1WXaz_0ymSYCTOZ6lN-Km6h0,1071
+sagecli/__init__.py,sha256=fA9FcZiYMXastMgjt5bm7CqFhmTGIS-Wh7HgQ5Ju-BU,233
+sagecli/__main__.py,sha256=ov0gnHlj8ay-Txuxuli09IV84UJcVitsbp-wTKH9FkE,108
+sagecli/cli.py,sha256=eWdOVMHEQw21i-dZrxAntas3ATD-PKAaxkOr41j0WMc,21687
+sagecli/config.py,sha256=VEO8WlBVlgioPTT3b9vAL9Az05leU-Ue2wr5ERytUQ4,9289
+sagecli/security.py,sha256=wzlIbAzAoX_f9A579jjQGu-B42RLFk6pRS5_Y8qOL3o,3539
+sagecli/wizard.py,sha256=C3wV3YUxABZGqE6YzwLYoxtzHI4BkuznJVFbQuQhrf4,6007
+sagecli/core/__init__.py,sha256=VdIDOOTPKyNmj_73ztqB8lvLzMdbtUu0btvFYpEJa28,422
+sagecli/core/agent.py,sha256=P-2rISLneYOuSF_h9nPy9_C5dAqjRL6N5Fd0w-XKLKE,9503
+sagecli/core/context.py,sha256=a9MkGLfkA5mm2iMXxY8lfyVZMAzIWuMU44d5cVQhpOw,4232
+sagecli/core/doctor.py,sha256=vJtHK_kqWLc4FSScRlQYknI9SzrvoY1pYfXSuRpKzxE,4876
+sagecli/core/orchestrator.py,sha256=1r8xpNVVsVZh8b4SJmWHQtjol-p3s11iw7pBZrQvQAE,2506
+sagecli/core/state.py,sha256=0__PWoKnir-2iUDT5SG-PM4cJYKpI6muB2dczkV4poo,3027
+sagecli/core/tracker.py,sha256=gxZ6S17GlRLNzKHmN3lPm7nqiRtlLn3zgGMiwOYtJU4,2878
+sagecli/llm/__init__.py,sha256=fTAuJd394hiMRHO4byJzQl_6iH1XNK2Xq25H_bOdGWk,519
+sagecli/llm/anthropic_provider.py,sha256=YuP_JbwsDUdqWgPKrXKKssGiVTgLc2_PEQC9oN9kCkg,9999
+sagecli/llm/base.py,sha256=cep5Q1JNX0AKf_8UTM28xzxM6oNcaqbzFhIkKOVBpUk,2651
+sagecli/llm/factory.py,sha256=9k73NPbLbf5VdipqQ8EShbsDXKM1yhwXT5g6K3bl0OA,2492
+sagecli/llm/gemini_provider.py,sha256=q1sZlsnsboUK9fVjubDwqrt_5KxkttOZP1A059v33Xs,11716
+sagecli/llm/openai_provider.py,sha256=ZXT4T4XBFHxRq0EThLWxKVJBRR8ISEn4C8BSt7J2pDw,8897
+sagecli/tools/__init__.py,sha256=7NuPlYMuQiV_dqK3aAp7jfiKsott-jFDhTQ9ue_2S8Q,841
+sagecli/tools/base.py,sha256=pYcGs5VcYPBnwmAATduFtRg4MsJ973FCKCtjzxvDrOQ,1269
+sagecli/tools/dataset_ops.py,sha256=8qHnyPmC5_bpR3tzUa-3GZ4diZ3AvRGaKJ0idbgpSoQ,6729
+sagecli/tools/file_ops.py,sha256=XbxmhyIbrz7LW7KGY7b98Wio28qrbChPZhevV4LfXmU,8006
+sagecli/tools/git_ops.py,sha256=R8Z6Fe55TnYIXt9KXv3rjDwey9uDC5J31lmJKK6ugBk,4336
+sagecli/tools/registry.py,sha256=BOoOHcb1I5xPrDAOkLGQNPFO4-gAWgdmmS41wUA3GF0,3682
+sagecli/tools/shell_ops.py,sha256=FJcgFIsSOURntJDA7Oq9b5EPAIkVWvd703w8uVASHnE,2871
+sagecli/tools/test_eval_ops.py,sha256=hiLVhEfHndyTg1r_GiDtGK1QG9x78ZX_962ctcmbb_A,2439
+sagecli/ui/__init__.py,sha256=ufJQz-C-edIGDVlW1fj6NwwBZuPbodv1tJ4RpnZjP7s,1526
+sagecli/ui/banner.py,sha256=sfpMvQKmpSfcT1YKXzBjkdMzfAdtH28pO9SXxyUCgTo,10931
+sagecli/ui/console.py,sha256=lXwzFAagjzv-QHN8uU34Jlbgk06VNzji2Bwv_89LPrQ,6369
+sagecli/ui/logo.py,sha256=7S5MGxO2u4DZNA_pxiYHId2ZMA9ZJmTOppC0QS0i1qg,5133
+sagecli/ui/prompts.py,sha256=SROrsAVAWGWlcHEhqE66XUlO2FVqq_oqpUu9joFdez8,7073
+sagecli/ui/symbols.py,sha256=koLJqyFzQoA8QhVJucBdaUlmx_Y_Ks2Vid5_ypj6Ots,2131
+sagecli/ui/theme.py,sha256=AkRtTQmAGNRFJuhRcOXk10pjX79J7EFFmuV8h3mG97s,4125
+sage_ai_cli-0.1.0.dist-info/METADATA,sha256=fGagTOCoKyI262MzKEUu59yI1S9c85UL4lW4saQUiuE,9965
+sage_ai_cli-0.1.0.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+sage_ai_cli-0.1.0.dist-info/entry_points.txt,sha256=uLQRSbSQmmwXoXV0cqy1ez9BvaFOnf6TN-qYAW3agFU,67
+sage_ai_cli-0.1.0.dist-info/top_level.txt,sha256=fAtAH_nbWgIwOIToOIlOp2P5VYJ2DHBGPjmh7Ro6JoA,8
+sage_ai_cli-0.1.0.dist-info/RECORD,,

sage_ai_cli-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

sage_ai_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+sage = sagecli.cli:app
+sagecli = sagecli.cli:app

sage_ai_cli-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Zaheer JK Labs
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

sage_ai_cli-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+sagecli

sagecli/__init__.py

@@ -0,0 +1,7 @@
+"""SageCLI - AI Engineering Agent for Terminal."""
+
+__version__ = "0.1.0"
+__app_name__ = "SageCLI"
+__command__ = "sage"
+__tagline__ = "AI ENGINEERING AGENT"
+__secondary_tagline__ = "Build • Train • Debug • Evaluate • Deploy"

sagecli/__main__.py

@@ -0,0 +1,6 @@
+"""Entry point for python -m sagecli."""
+
+from sagecli.cli import app
+
+if __name__ == "__main__":
+    app()