jupyter-ai  by jupyterlab

JupyterLab extension for generative AI integration

created 2 years ago
3,724 stars

Top 13.3% on sourcepulse

GitHubView on GitHub
Project Summary

Jupyter AI integrates generative AI capabilities directly into the Jupyter ecosystem, targeting data scientists, researchers, and developers. It enhances productivity by providing a seamless interface for interacting with various AI models within notebooks and a dedicated chat UI in JupyterLab.

How It Works

Jupyter AI leverages the IPython kernel for its %%ai magic command, enabling reproducible generative AI experiments directly within notebooks. It supports a broad spectrum of AI model providers, including major cloud platforms and open-source models via local integrations like Ollama and GPT4All. This approach allows users to easily switch between different models and providers, facilitating exploration and comparison of AI capabilities within their existing Jupyter workflows.

Quick Start & Requirements

  • Install via pip: pip install jupyter-ai[all]
  • Requires Python 3.9-3.12 and JupyterLab 4 or Notebook 7.
  • Model provider credentials (API keys) are necessary for most services.
  • Official documentation: https://jupyter-ai.readthedocs.io/en/latest/

Highlighted Details

  • Supports a wide range of providers: AI21, Anthropic, AWS, Cohere, Gemini, Hugging Face, MistralAI, NVIDIA, OpenAI.
  • Enables local model usage via GPT4All and Ollama for privacy and cost-effectiveness.
  • %%ai magic command interpolates IPython expressions and generates rich output (HTML, math).
  • Native chat UI within JupyterLab for conversational AI interaction.

Maintenance & Community

  • Incubation project under the JupyterLab organization.
  • Contributor documentation available on ReadTheDocs.

Licensing & Compatibility

  • License: BSD-3-Clause.
  • Compatible with commercial and closed-source applications.

Limitations & Caveats

JupyterLab 3 support is deprecated, with critical issue backports ending December 31, 2024; upgrading to JupyterLab 4 is strongly recommended. Some model providers require specific, separately installed dependencies.

Health Check
Last commit

2 days ago

Responsiveness

1 day

Pull Requests (30d)
18
Issues (30d)
26
Star History
164 stars in the last 90 days

Explore Similar Projects

Starred by Chip Huyen Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems) and Carol Willing Carol Willing(Core Contributor to CPython, Jupyter).

genai by rgbkrk

0%
352
IPython extension for generative AI assistance in Jupyter notebooks
created 3 years ago
updated 1 year ago
Starred by Tobi Lutke Tobi Lutke(Cofounder of Shopify), Joe Walnes Joe Walnes(Head of Experimental Projects at Stripe), and
5 more.

LibreChat by danny-avila

0.8%
29k
Enhanced ChatGPT clone for self-hosting
created 2 years ago
updated 1 day ago
Feedback? Help us improve.