Ai Jup

ai-jup: AI-Powered Prompt Cells for JupyterLab

A JupyterLab extension that adds AI-powered prompt cells with special syntax for referencing kernel variables and exposing functions as AI-callable tools. Inspired by [fast.ai's Solveit](https://www.fast.ai/posts/2025-11-07-solveit-features.html). This project was a ["vibe-coding" experiment](https://x.com/HamelHusain/status/2006440720001835135?s=20) by Hamel Husain, and only has a small subset of Solveit's features. It's not designed to be a project for long-term development and maintenance. If you do wish to use it, please be sure to follow the GPL v3 license, including releasing all your source code if you modify and use this yourself, including hosting it as an application.

[](https://github.com/hamelsmu/ai-jup/actions/workflows/ci.yml)   

Features

• $`variable` - Reference kernel variables in prompts (values substituted automatically)
• &`function` - Expose Python functions as AI-callable tools
• Streaming responses - Real-time SSE streaming from Anthropic's Claude
• Full context - AI sees all preceding code cells and kernel state
• Tool loop - AI can call multiple tools in sequence to complete tasks
• Built-in file tools - Includes fastcore.tools for file exploration and editing

Quick Start

1. Install

pip install ai_jup

Verify the extension is installed:

jupyter labextension list | grep ai-jup
# Should show: ai-jup v0.1.0 enabled OK

2. Set your API key

export ANTHROPIC_API_KEY="your-key-here"

To persist across sessions, add to your shell profile:

# For zsh (macOS default)
echo 'export ANTHROPIC_API_KEY="your-key-here"' >> ~/.zshrc

# For bas