DEV Community

novatechnolab
novatechnolab

Posted on

Jupyter AI Extension - Multi-LLM Support

Jupyter AI Extension

Multi-LLM support for Jupyter notebooks with OpenAI, Claude, and Google Gemini.

Features

  • Support for GPT-4, Claude Opus, and Gemini 2.0 Flash
  • Simple one-line configuration
  • Chat and completion modes
  • Fully documented with examples

Installation

pip install jupyter-ai-extension
Enter fullscreen mode Exit fullscreen mode

Quick Start

from jupyter_ai_extension import JupyterAIExtension
ai = JupyterAIExtension()
ai.configure("openai")
response = ai.generate("Explain machine learning")
print(response)
Enter fullscreen mode Exit fullscreen mode

Documentation

Check out the repository for more examples and documentation!

Top comments (0)