Notebooks
Overview
Athena Notebooks integrate Jupyter notebook functionality with additional AI features, making it easier for you to collaborate with Athena and get work done.
Video Overview:
Features
Chat with Athena
You can collaborate with Athena while you work. She can "see" your notebook and your currently selected cell. So if you are stuck on something, just pop open the sidebar and ask for help or to write the code for you!
Athena Jupyter Command
In Jupyer notebooks the double percentage signs "%%" are cell magic commands. You can use Athena directly by using the %%ai athena
command followed by your prompt your request.
You can use the -f code
flag to have Athena insert the output of your into your notebook as a cell. The full command then looks like:
%%ai athena -f code
Athena SDK
You can use the Athena SDK from directly within the Athena Notebooks.
The Athena SDK enables you to:
Generate Large-Language Model responses
Search and Browse the Web
Scrape URLs
Use tool-based workflows
and more!
You can started using the Athena SDK by initializing it with your API Key.
Your Athena API key is automatically injected into your Jupyter notebook. Simply initialize the Athena client like the code below and you are ready to go!
from athena.client import Athena
from athena import Model, Tools, LlmModel
import pandas as pd
import os
import time
# Your API key is automatically set in the environment for you
ATHENA_API_KEY = os.environ['ATHENA_API_KEY']
# This initializes the Athena Client
athena = Athena(
api_key=ATHENA_API_KEY,
)