basic.py
"""
TokenLab Basic
==============
Cookbook example for `tokenlab/basic.py`.
"""
import asyncio
from agno.agent import Agent, RunOutput # noqa
from agno.models.tokenlab import TokenLab
# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------
agent = Agent(
model=TokenLab(id="gpt-5.4-mini"),
markdown=True,
)
# Get the response in a variable
# run: RunOutput = agent.run("Share a 2 sentence sci-fi story")
# print(run.content)
# Print the response in the terminal
# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
if __name__ == "__main__":
# --- Sync ---
agent.print_response("Share a 2 sentence sci-fi story")
# --- Sync + Streaming ---
agent.print_response("Share a 2 sentence sci-fi story", stream=True)
# --- Async ---
asyncio.run(agent.aprint_response("Share a 2 sentence sci-fi story"))
# --- Async + Streaming ---
asyncio.run(agent.aprint_response("Share a 2 sentence sci-fi story", stream=True))
Run the Example
1
Set up your virtual environment
uv venv --python 3.12
source .venv/bin/activate
uv venv --python 3.12
.venv\Scripts\activate
2
Install dependencies
uv pip install -U agno openai
3
Export your API keys
export TOKENLAB_API_KEY="your_tokenlab_api_key_here"
$Env:TOKENLAB_API_KEY="your_tokenlab_api_key_here"
4
Run the example
Save the code above as
basic.py, then run:python basic.py