ipychat: An AI extension for IPython that makes it work like Cursor
ipychat is an AI extension for IPython that helps you understand, debug, and write Python faster.
- Context-Aware: Automatically uses relevant context from your IPython session
- Rich Output: Markdown-formatted responses with syntax highlighting
- Multiple AI Models: Supports GPT-4o, Claude 3.5 Sonnet, and Gemini. Local models coming soon.
- Interactive Configuration: Easy model switching and configuration through magic commands
You can install ipychat
using pip:
$ pip install ipychat
You can start the REPL by running the ipychat
command:
$ ipychat
Welcome to ipychat! Use %ask to chat with gpt-4o.
You can change models using %models.
In [1]:
Alternatively, you can load the ipychat
extension in an IPython session:
In [1]: %load_ext ipychat
You can then start asking the model any question using the %ask
magic.
In [1]: %ask what can I do with the cities dataframe
You can change the current model using the %models
magic.
In [1]: %models
Based on the model you want to use, either set OPENAI_API_KEY
, or ANTHROPIC_API_KEY
, or both environment variables. You can also run ipychat config
to configure ipychat
interactively.
Contributions are welcome! Please feel free to submit a pull request.
ipychat
uses Semantic Versioning. For the available versions, see the tags on the GitHub repository.
This project is licensed under the Apache License, see the LICENSE file for details.