DuckDB Tools

Query a remote IMDB movie CSV with SQL using DuckDbTools to compute average ratings.

duckdb_tools.py

"""
Duckdb Tools
=============================

Demonstrates duckdb tools.
"""

from agno.agent import Agent
from agno.tools.duckdb import DuckDbTools

# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------


agent = Agent(
    tools=[DuckDbTools()],
    instructions="Use this file for Movies data: https://agno-public.s3.amazonaws.com/demo_data/IMDB-Movie-Data.csv",
)

# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
if __name__ == "__main__":
    agent.print_response(
        "What is the average rating of movies?", markdown=True, stream=False
    )

Run the Example

1. Set up your virtual environment

   Mac

   uv venv --python 3.12
   source .venv/bin/activate
   
2. Install dependencies

   uv pip install -U agno duckdb openai
   
3. Export your OpenAI API key

   Mac/Linux

   export OPENAI_API_KEY="your_openai_api_key_here"
   
4. Run the example

   Save the code above as duckdb_tools.py, then run:

   python duckdb_tools.py
   

Full source: cookbook/91_tools/duckdb_tools.py