Skip to content

plmbr/notebook-intelligence

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1,011 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Notebook Intelligence

Notebook Intelligence (NBI) is an AI coding assistant and extensible AI framework for JupyterLab. It adds chat, inline edit, auto-complete, and an agent that can drive notebooks — backed by GitHub Copilot, an OpenAI-compatible or LiteLLM-compatible endpoint, local Ollama models, or Anthropic's Claude Code CLI.

NBI is free and open-source. Connect it to a free or paid LLM provider of your choice — GitHub Copilot, any OpenAI- or LiteLLM-compatible endpoint, Ollama (local), or Anthropic Claude (via the Claude Code CLI). Provider charges, when applicable, are paid directly to the provider.

Contents

Requirements

  • Python 3.10+
  • JupyterLab 4.x
  • Node.js — only required for Claude mode (the Claude Code CLI) and for MCP servers that launch via npx.
  • A fresh virtualenv or conda env is recommended so NBI doesn't conflict with system Python.

Quick start

pip install notebook-intelligence
jupyter lab     # restart JupyterLab if it was already running

After restart:

  1. Click the NBI icon in the left sidebar to open the chat panel.
  2. Open NBI Settings (gear icon in the chat panel, or Settings → Notebook Intelligence Settings).
  3. Sign into your provider — for GitHub Copilot, click Sign in; for an OpenAI- or LiteLLM-compatible endpoint, paste an API key; for Ollama, point at your local daemon. To use Claude, enable Claude mode (see below).
  4. Type a message in the chat panel and press Enter.

If the panel stays empty or login does nothing, see Troubleshooting.

Concepts

A short glossary you'll see referenced throughout these docs.

  • LLM Provider — the service that runs the model. NBI ships with four provider adapters: GitHub Copilot, OpenAI-compatible, LiteLLM-compatible, and Ollama. Anthropic Claude is available through Claude mode, not as a top-level provider.
  • Chat Participant — a @mention-able persona inside the chat panel (@workspace, @mcp, …). Participants route the request to a specific tool surface.
  • Default mode vs Claude modeDefault uses the configured LLM Provider for chat, inline chat, and auto-complete. Claude mode uses the Claude Code CLI for the chat panel (gaining its tools, skills, MCP servers, and custom commands) and Claude models via the Anthropic API for inline chat and auto-complete. Requires the Claude Code CLI on PATH.
  • Claude Code vs the Anthropic API — the Anthropic API (api.anthropic.com) is the HTTPS endpoint NBI calls directly for inline chat and auto-complete in Claude mode. Claude Code is Anthropic's local CLI agent that NBI shells out to for the chat panel; it talks to Anthropic itself.
  • MCPModel Context Protocol. A way for the LLM to call out to external tools (read files, hit APIs, run scripts).
  • Ruleset — markdown files in ~/.jupyter/nbi/rules/ that get injected into the system prompt to enforce conventions, coding standards, or domain rules.
  • Skill: a directory under ~/.claude/skills/ (or <project>/.claude/skills/) holding a SKILL.md plus helper files. Claude can invoke it like a callable plugin scoped to a workspace.
  • Claude plugin: a unit packaged for claude plugin install, distributed through a marketplace (typically a GitHub repo that publishes a manifest of plugins). Distinct from NBI's own labextension; plugins run inside Claude Code sessions.

Feature highlights

Claude mode

NBI provides a dedicated mode for Claude Code integration. In Claude mode, NBI uses the Claude Code CLI for the chat panel, and Claude models (via the Anthropic API) for inline chat and auto-complete suggestions. This brings Claude Code's tools, skills, MCP servers, and custom commands into JupyterLab.

Claude mode

Configure via the NBI Settings dialog (gear icon in the chat panel, or Settings → Notebook Intelligence Settings). Toggle Enable Claude mode, then:

  • Chat model — the Claude model used for the chat panel and inline chat.
  • Auto-complete model — the Claude model used for auto-complete suggestions.
  • Chat Agent setting sources — user, project, or both, mirroring Claude Code's settings.
  • Chat Agent tools — which tool sets to activate. Claude Code tools are always on. Jupyter UI tools are NBI's own (authoring notebooks, running cells, etc.).
  • API key and Base URL — point at Anthropic or a self-hosted endpoint.

If the Claude Code CLI is on PATH, NBI launches it automatically. To override the location, set the NBI_CLAUDE_CLI_PATH environment variable before starting JupyterLab.

Claude settings

Resuming a previous Claude session

When Claude mode is on, the chat sidebar shows a history icon next to the gear. Click it to list the Claude Code sessions recorded for the current working directory (the same transcripts the Claude Code CLI stores under ~/.claude/projects/). Selecting a session reconnects via resume, so the next message you send continues that transcript with full prior context.

Claude Code launcher tile

When the Claude CLI is on PATH, the JupyterLab launcher (the panel that opens with new tabs) shows a Claude Code tile alongside the standard kernel launchers. Clicking it opens a session picker; search across past transcripts and resume one in a fresh terminal, or start a new session in the file browser's active subdirectory. Session IDs are copyable from the picker for paste into a claude --resume <id> command.

Other coding-agent launcher tiles

When any of the following CLIs are on PATH, the launcher adds a tile for each. Clicking a tile opens a terminal at the file browser's current directory and runs the CLI:

  • opencode (override path with NBI_OPENCODE_CLI_PATH)
  • Pi (override path with NBI_PI_CLI_PATH)
  • GitHub Copilot CLI (override path with NBI_GITHUB_COPILOT_CLI_PATH)
  • OpenAI Codex (override path with NBI_CODEX_CLI_PATH)

Tiles add and remove themselves as CLIs become available or unavailable; they do not require Claude mode.

Agent mode

In Agent mode, the built-in AI agent creates, edits, and executes notebooks for you interactively. It can detect issues in cells and fix them.

Agent mode

Code generation with inline chat

Use the sparkle icon on the cell toolbar or the keyboard shortcut to show the inline chat popover.

Ctrl+G / Cmd+G opens the popover. Ctrl+Enter / Cmd+Enter accepts the suggestion. Esc closes it. The accept shortcut overrides JupyterLab's default run cell binding only while the popover is open — outside the popover, Ctrl+Enter / Cmd+Enter still runs the active cell.

Generate code

Auto-complete

Auto-complete suggestions are shown as you type. Tab accepts. NBI provides auto-complete in code cells and Python file editors.

Auto-complete

Chat interface

Chat interface

You can paste or attach images alongside a chat prompt — the image goes to the model as input when the active model supports vision.

Cell output actions

Right-click a cell output (or hover for the toolbar) to send it straight into the chat as context:

  • Explain cell errors — surfaces a "Troubleshoot errors in output" entry on cells that raised; opens a chat turn with the traceback attached.
  • Ask about cell outputs — attaches the output as structured context for a follow-up question. Includes images for vision-capable models.
  • Show output toolbar — the floating toolbar above each output with quick Explain / Ask / Troubleshoot actions.

Each is per-user toggleable from Settings (saved as enable_explain_error, enable_output_followup, enable_output_toolbar in config.json, default on) and admin-lockable via NBI_EXPLAIN_ERROR_POLICY / NBI_OUTPUT_FOLLOWUP_POLICY / NBI_OUTPUT_TOOLBAR_POLICY.

Notebook toolbar generation

Active notebooks show a sparkle icon on the toolbar. Click it to open a popover that scopes the generation request to that specific notebook — handy for multi-notebook sessions where you don't want the chat sidebar to compete for context.

Configuration

Configure your provider, model, and API key from NBI Settings — the gear icon in the chat panel, the /settings chat command, or the JupyterLab command palette. For background, see the provider blog post.

Settings dialog

Configuration files

NBI saves configuration at ~/.jupyter/nbi/config.json. It also supports an environment-wide base configuration at <env-prefix>/share/jupyter/nbi/config.json — organizations can ship default configuration there, and user changes save as overrides on top.

These config files store provider, model, and MCP configuration. API keys for custom LLM providers are also stored here in plaintext — never commit ~/.jupyter/nbi/config.json to git, share it, or sync it across users. If a key leaks, rotate it at the provider immediately.

Manual edits to config.json require a JupyterLab restart to take effect. Edits via the Settings dialog are picked up live.

Admin policies

Most settings panel toggles can be locked by org administrators. Two shapes:

Boolean policies use the *_POLICY suffix and accept three values: user-choice (default — user toggles freely), force-on (locked enabled), force-off (locked disabled). When forced, the panel control is disabled with a "Locked by your administrator" tooltip and any client-side write is ignored.

Env var Locks the Settings panel control for
NBI_EXPLAIN_ERROR_POLICY "Explain cell errors"
NBI_OUTPUT_FOLLOWUP_POLICY "Ask about cell outputs"
NBI_OUTPUT_TOOLBAR_POLICY "Show output toolbar"
NBI_CLAUDE_MODE_POLICY "Enable Claude mode"
NBI_CLAUDE_CONTINUE_CONVERSATION_POLICY "Remember conversation history"
NBI_CLAUDE_CODE_TOOLS_POLICY "Claude Code tools"
NBI_CLAUDE_JUPYTER_UI_TOOLS_POLICY "Jupyter UI tools"
NBI_CLAUDE_SETTING_SOURCE_USER_POLICY Setting source: User
NBI_CLAUDE_SETTING_SOURCE_PROJECT_POLICY Setting source: Project
NBI_STORE_GITHUB_ACCESS_TOKEN_POLICY "Remember my GitHub Copilot access token"
NBI_SKILLS_MANAGEMENT_POLICY The Skills tab (force-off hides it and 403s the API; also disables the managed-skills reconciler)
NBI_CLAUDE_MCP_MANAGEMENT_POLICY The Claude-mode MCP Servers tab (force-off hides it and 403s /claude-mcp/*; independent of the non-Claude MCP Servers tab)
NBI_CLAUDE_PLUGINS_MANAGEMENT_POLICY The Claude-mode Plugins tab (force-off hides it and 403s /plugins/*)
NBI_TERMINAL_DRAG_DROP_POLICY Terminal drag-drop file attach feature
NBI_REFRESH_OPEN_FILES_ON_DISK_CHANGE_POLICY "Refresh open files when changed on disk"

The first three also have matching traitlets on NotebookIntelligence (explain_error_policy, output_followup_policy, output_toolbar_policy); add the others as needed in the same shape:

c.NotebookIntelligence.claude_mode_policy = "force-on"
c.NotebookIntelligence.claude_jupyter_ui_tools_policy = "force-off"

Per-user preferences (default on for the cell-output features) live in config.json as enable_explain_error, enable_output_followup, enable_output_toolbar.

List-shaped denylists (LLM providers, built-in tools, coding-agent launcher tiles) use traitlets rather than *_POLICY env vars. See docs/admin-guide.md for the disabled_providers, disabled_tools, and disabled_coding_agent_launchers recipes.

Value-presence locks for non-boolean settings: setting the env var to a non-empty value pins the control to that value and disables it. Empty/unset = user-choice.

Env var Pins
NBI_CHAT_MODEL_PROVIDER General → Chat model → Provider
NBI_CHAT_MODEL_ID General → Chat model → Model
NBI_INLINE_COMPLETION_MODEL_PROVIDER General → Auto-complete model → Provider
NBI_INLINE_COMPLETION_MODEL_ID General → Auto-complete model → Model
NBI_CLAUDE_CHAT_MODEL Claude → Chat model
NBI_CLAUDE_INLINE_COMPLETION_MODEL Claude → Auto-complete model
ANTHROPIC_API_KEY Claude → API Key (input is locked + blanked; the SDK reads the env directly)
ANTHROPIC_BASE_URL Claude → Base URL

Provider IDs: github-copilot, openai-compatible, litellm-compatible, ollama, none. The *_MODEL_ID value is whatever the chosen provider exposes (e.g. gpt-4o, llama3:latest). Claude model IDs are the literal IDs from the Anthropic API (e.g. claude-opus-4-7, claude-sonnet-4-6); empty string = "Default (recommended)"; NBI_CLAUDE_INLINE_COMPLETION_MODEL also accepts none (no inline completion in Claude mode) or inherit (use the General-tab Auto-complete model).

Upload tunables govern the shared upload endpoint used by both chat-sidebar file attachments and terminal drag-drop:

Env var Default Behavior
NBI_UPLOAD_MAX_MB 50 Per-file size cap in megabytes. Requests over the cap return HTTP 413. Set to 0 to disable the cap entirely.
NBI_UPLOAD_RETENTION_HOURS 24 How long staged uploads survive in the temp directory before the next upload sweeps them. Set to 0 to keep only the atexit purge (uploads survive the session).

The same values are also configurable via the upload_max_mb and upload_retention_hours traitlets on NotebookIntelligence.

Remembering GitHub Copilot login

NBI can remember your GitHub Copilot login so you don't have to sign in again after a JupyterLab or system restart.

Caution

If you enable this, NBI encrypts the token and stores it in ~/.jupyter/nbi/user-data.json. Never share this file. The encryption uses a default password unless you set NBI_GH_ACCESS_TOKEN_PASSWORD to a custom value — on shared or multi-tenant systems, set a custom password before enabling this option.

NBI_GH_ACCESS_TOKEN_PASSWORD=my_custom_password

To enable, check Remember my GitHub Copilot access token in the Settings dialog.

Remember access token

If the stored token fails to authenticate (expired, revoked, password mismatch), NBI prompts you to sign in again.

Built-in tools

These tools are available in Agent mode and to MCP-enabled chats.

Tool What it does
Notebook Edit (nbi-notebook-edit) Edit notebooks via the JupyterLab notebook editor.
Notebook Execute (nbi-notebook-execute) Run notebooks in the JupyterLab UI.
Python File Edit (nbi-python-file-edit) Edit Python files via the JupyterLab file editor.
File Edit (nbi-file-edit) Edit files in the Jupyter root directory.
File Read (nbi-file-read) Read files in the Jupyter root directory.
Command Execute (nbi-command-execute) Execute shell commands using the embedded terminal in Agent UI or JupyterLab terminal.

In multi-tenant deployments, nbi-command-execute and nbi-file-edit are effectively arbitrary code execution as the user. See docs/admin-guide.md for guidance on disabling them.

Model Context Protocol (MCP) support

NBI integrates with MCP servers. It supports both stdio and Streamable HTTP transports. MCP server tools are supported; resources and prompts are not yet supported.

Add MCP servers by editing ~/.jupyter/nbi/mcp.json. An environment-wide base file at <env-prefix>/share/jupyter/nbi/mcp.json is also supported.

Note

MCP requires an LLM model with tool-calling capability. All GitHub Copilot models in NBI support this. For other providers, choose a tool-calling-capable model.

Caution

Most MCP servers run on the same machine as JupyterLab and can make irreversible changes or access private data. Only install MCP servers from trusted sources.

MCP config example

{
  "mcpServers": {
    "filesystem": {
      "command": "npx",
      "args": [
        "-y",
        "@modelcontextprotocol/server-filesystem",
        "/Users/mbektas/mcp-test"
      ]
    }
  }
}

For stdio servers you can pass extra environment variables under env:

"mcpServers": {
    "servername": {
        "command": "",
        "args": [],
        "env": {
            "ENV_VAR_NAME": "ENV_VAR_VALUE"
        }
    }
}

For Streamable HTTP servers you can also specify request headers:

"mcpServers": {
    "remoteservername": {
        "url": "http://127.0.0.1:8080/mcp",
        "headers": {
            "Authorization": "Bearer mysecrettoken"
        }
    }
}

To temporarily disable a configured server without removing it, set "disabled": true:

"mcpServers": {
    "servername2": {
        "command": "",
        "args": [],
        "disabled": true
    }
}

Rulesets

NBI's ruleset system lets you define guidelines and best practices that get injected into AI prompts automatically — for consistent coding standards, project conventions, or domain knowledge. Rules are markdown files in ~/.jupyter/nbi/rules/ and can scope by file pattern, kernel, directory, or chat mode.

A two-line example:

---
priority: 10
---

- Always use type hints in Python functions.
- Add docstrings to all public functions.

For full details (frontmatter reference, mode-specific rules, auto-reload), see docs/rulesets.md.

Claude Skills

The Settings panel exposes a top-level Skills tab for managing the skills that Claude can invoke. Skills are stored under ~/.claude/skills/ (user) or <project>/.claude/skills/ (project). You can create and edit skills inline, duplicate, rename, delete (with undo), or import from a public GitHub repo. The tab is visible in any mode; when Claude mode is off, a hint banner notes that skills only take effect inside Claude sessions (handy for authoring skills now and using them when you turn Claude mode on later).

For organization-wide deployments, your admin can ship a curated manifest of skills and keep them in sync by setting NBI_SKILLS_MANIFEST. Skills installed this way are marked Managed and are read-only in the UI. Admins who want to keep managed skills but disallow ad-hoc GitHub imports use NBI_ALLOW_GITHUB_SKILL_IMPORT=false.

For full details, see docs/skills.md.

Claude MCP Servers

When Claude mode is enabled and the Claude CLI is available, the Settings panel exposes an MCP Servers tab that manages the user, project, and local-scope MCP entries Claude Code reads from ~/.claude.json and the project's .mcp.json. This is a different tab from NBI's own MCP Servers tab (which manages the servers used by the non-Claude chat path); the two never appear together, and the Settings dialog shows whichever one applies to your current mode.

Reads come from Claude's JSON config files directly. Writes (add and remove) shell out to claude mcp add and claude mcp remove so Claude remains the source of truth for any side effects (project-trust prompts, OAuth bookkeeping). Admins can lock the tab with NBI_CLAUDE_MCP_MANAGEMENT_POLICY=force-off.

Claude Plugins

When Claude mode is enabled and the Claude CLI is available, the Settings panel exposes a Plugins tab wrapping claude plugin for install, uninstall, enable, disable, and marketplace add (for example: add a marketplace from a GitHub repo, then install plugins it publishes). Marketplaces hosted on GitHub reuse the same GITHUB_TOKEN / GH_TOKEN / gh auth token precedence as Skills imports; the token is passed via env to the subprocess and never appears in argv or DEBUG logs. See Anthropic's plugin docs for what a plugin is and how marketplaces work.

Admins can lock the entire tab with NBI_CLAUDE_PLUGINS_MANAGEMENT_POLICY=force-off, or keep the tab and block only GitHub-sourced marketplaces with NBI_ALLOW_GITHUB_PLUGIN_IMPORT=false.

Chat feedback

Enable thumbs-up/down feedback on AI responses by setting:

c.NotebookIntelligence.enable_chat_feedback = True

…or via CLI:

jupyter lab --NotebookIntelligence.enable_chat_feedback=true

The feedback fires an in-process telemetry event. Nothing leaves the process by default — see the admin guide for how to wire it into your observability stack.

Chat feedback

Documentation

Further reading

Roadmap

NBI 4.x is stable. New features land in minor releases (4.5, 4.6, …); breaking changes are reserved for the next major (5.x) and will be announced in the changelog.

License

Licensed under GPL-3.0.

About

A JupyterLab extension supporting Claude Code, Copilot, Ollama, and OpenAI-compatible LLMs, with MCP, skills, plugins, and notebook agents.

Topics

Resources

License

Contributing

Security policy

Stars

Watchers

Forks

Packages

 
 
 

Contributors