jupyter-ai  by jupyterlab

JupyterLab extension for generative AI integration

Created 2 years ago
3,805 stars

Top 12.8% 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

1 week ago

Responsiveness

1 day

Pull Requests (30d)
16
Issues (30d)
17
Star History
59 stars in the last 30 days

Explore Similar Projects

Starred by Chip Huyen Chip Huyen(Author of "AI Engineering", "Designing Machine Learning Systems"), Gabriel Almeida Gabriel Almeida(Cofounder of Langflow), and
1 more.

genai by rgbkrk

0.3%
353
IPython extension for generative AI assistance in Jupyter notebooks
Created 3 years ago
Updated 1 year ago
Feedback? Help us improve.