Skip to main content

Sage Agent - Your Jupyter Notebook Assistant

Project description

Sage Agent — The AI-Powered Assistant for Jupyter Notebooks

Logo


What is Sage Agent?

Sage is an AI-native notebook assistant that supercharges your existing Jupyter workflows.

Built by leading AI and quant researchers from YC, Harvard, MIT, and Goldman Sachs, Sage brings real-time, context-aware assistance directly into JupyterLab.

Use natural language to clean data, write analysis code, debug errors, explore dataframes, and build models—faster and with fewer mistakes.

No hallucinated code. No context switching. Just faster insights.


Why Use Sage Agent in Jupyter?

Whether you’re a quant, data scientist, or analyst living in notebooks, Sage helps you:

✅ Clean and transform messy data in seconds

✅ Visualize trends, rollups, and anomalies from a prompt

✅ Connect your custom databases in one click and easily explore from notebooks

✅ Generate runnable Python or SQL that fits your current cell + variable context

✅ Auto-detect schema changes and debug downstream errors

✅ Stay private: run entirely local-first or in your own secure VPC

✅ Extend JupyterLab without changing how you work


Perfect For:

  • Data scientists cleaning huge CSVs
  • Quant researchers testing ML pipelines
  • Product and analytics teams tired of building dashboards and flaky notebooks
  • Anyone tired of LLM tools that break their code

Installation

📦 Requirements

  • JupyterLab >= 4.0.0
  • NodeJS (for development)

🧠 Install Sage Agent:

pip install jupyterlab signalpilot_ai

❌ Uninstall:

pip uninstall signalpilot_ai

How to Get Started

To unlock full functionality, you’ll need Sage API credentials.

👉 Request your API key or email us at contact@sagebook.ai


Why Sage

  • ✅ Context-aware code gen: understands variables, dataframes, imports, and prior cells
  • ✅ AI that fixes schema issues and silent join bugs
  • ✅ Inline review + diffs before you run any code
  • ✅ Visualizations via natural language (matplotlib, plotly, seaborn supported)
  • ✅ BYO LLM: Anthropic, OpenAI, vLLM, Ollama, or HF endpoints
  • ✅ Built to run in air-gapped / enterprise environments

Local Development Instructions

To contribute or develop locally:

# Clone the repo and enter the directory
git clone https://github.com/sagebook/signalpilot_ai.git
cd signalpilot_ai

# Install in editable mode
pip install -e "."

# Link extension to JupyterLab
jupyter labextension develop . --overwrite

# Rebuild on changes
jlpm build

For auto-rebuild while editing:

# Watch source
jlpm watch

# Run JupyterLab in parallel
jupyter lab

Uninstall in Dev Mode

pip uninstall signalpilot_ai
# Then manually remove labextension symlink from JupyterLab extensions dir.

Want to See Sage in Action?

🎥 Try the demo notebook or explore at https://sagebook.ai


Built for teams working with sensitive data:

  • Zero data retention by default
  • Optional BYO keys for Claude, OpenAI, or local models
  • Notebook-specific controls for what the model can “see”
  • Fine-grained telemetry settings

Contact

Questions? Ideas?

Email: fahim@sagebook.ai

Website: https://sagebook.ai


AI Jupyter Notebook, JupyterLab Extension, Jupyter Assistant, Data Science Assistant, Jupyter LLM, AI code generation, dataframe cleaning, Jupyter AI, Sagebook, Sage Agent, AI for dataframes, Jupyter SQL assistant, notebook extension

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

signalpilot_ai-0.1.0.tar.gz (1.7 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

signalpilot_ai-0.1.0-py3-none-any.whl (5.5 kB view details)

Uploaded Python 3

File details

Details for the file signalpilot_ai-0.1.0.tar.gz.

File metadata

  • Download URL: signalpilot_ai-0.1.0.tar.gz
  • Upload date:
  • Size: 1.7 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.13.5

File hashes

Hashes for signalpilot_ai-0.1.0.tar.gz
Algorithm Hash digest
SHA256 b96227ea5d6dc35b109cc383f44d3d5da88329ee8c94cf6ff6ec9d331c5ad136
MD5 4eb6f82a66cda53d2251a09c61a317c4
BLAKE2b-256 a6e7028bd68dd3aa21db61c73a743c7e92ffe2211db58ec055f84dad7d5f2912

See more details on using hashes here.

File details

Details for the file signalpilot_ai-0.1.0-py3-none-any.whl.

File metadata

  • Download URL: signalpilot_ai-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 5.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.13.5

File hashes

Hashes for signalpilot_ai-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 5712d15e6dbcd16224f5218f02875914a30ab4e4d366c4da538e254e217b81ec
MD5 22df546ebb921de8fa6c303c493cf244
BLAKE2b-256 666817f77bdc70577fab5ee5c531158bb72ecea22d4b275008f32f4fa9f842c7

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page