from pathlib import Path
import httpx
from agno.agent import Agent
from agno.tools.csv_toolkit import CsvTools
url = "https://agno-public.s3.amazonaws.com/demo_data/IMDB-Movie-Data.csv"
response = httpx.get(url)
imdb_csv = Path(__file__).parent.joinpath("imdb.csv")
imdb_csv.parent.mkdir(parents=True, exist_ok=True)
imdb_csv.write_bytes(response.content)
agent = Agent(
tools=[CsvTools(csvs=[imdb_csv])],
markdown=True,
show_tool_calls=True,
instructions=[
"首先,始终获取文件列表",
"然后,检查文件中的列",
"然后,运行查询以回答问题",
],
)
agent.cli_app(stream=False)
创建虚拟环境
打开 Terminal
并创建一个 python 虚拟环境。
python3 -m venv .venv
source .venv/bin/activate
设置你的 API 密钥
export OPENAI_API_KEY=xxx
安装库
pip install -U httpx openai agno
运行 Agent
python cookbook/tools/csv_tools.py