AI-powered Jupyter Notebook

Thread is a Jupyter alternative that integrates an AI copilot into your Jupyter Notebook editing experience.

Thread runs locally and can be used for free with Ollama or your own API key. To start:

pip install thread-notebook

To start thread, run any of the following commands:

thread-notebook

or

thread

or

jupyter thread-notebook

Key features

1. Familiar Jupyter Notebook editing experience

2. Natural language code edits

3. Generate cells to answer natural language questions

4. Ask questions in a context aware chat sidebar

5. Automatically explain or fix errors

Demo

https://github.com/alishobeiri/thread/assets/18422723/b0ef0d7d-bae5-48ad-b293-217b940385fb

Development instructions

To run the repo in development mode, you need to run two terminal commands. One will run Jupyter Server, the other will run the NextJS front end.

To begin, run:

yarn install

Then in one terminal, run:

sh ./run_dev.sh

And in another, run:

yarn dev

Navigate to localhost:3000/thread and you should see your local version of Thread running.

If you would like to develop with the AI features, navigate to the proxy folder and run:

yarn install

Then:

yarn dev --port 5001

Using Thread with Ollama

You can use Ollama for a fully offline AI experience. To begin, install and run thread using the commands above.

Once you have run thread, in the bottom left, select the Settings icon:

Next, select the Model Settings setting:

This is what you will see:

Navigate to Ollama and enter your model details:

Use Ctrl / Cmd + K and try running a query to see how it looks!