Database Toolset for Pydantic AI

Daj agentom AI możliwości baz danych SQL

Zestaw narzędzi SQL umożliwiający agentom AI odpytywanie, analizę i interakcję z bazami danych. Obsługuje introspekcję schematu i bezpieczne wykonywanie zapytań.

Instalacja

pip install database-pydantic-ai

Pięć narzędzi daje agentom pełną interakcję z bazą danych: list_tables, get_schema, describe_table, explain_query i query. Obsługuje SQLite (przez aiosqlite) i PostgreSQL (przez asyncpg) z ujednoliconym interfejsem. Bezpieczeństwo na pierwszym miejscu: tryb read-only blokuje 15+ niebezpiecznych słów kluczowych SQL w tym DROP, DELETE i ALTER. Kontrola zasobów z konfigurowalnymi timeoutami zapytań (domyślnie 30s) i limitami wierszy (domyślnie 100). Parsowanie uwzględniające komentarze wykrywa niebezpieczne słowa kluczowe ukryte za -- i /* */.

Funkcje

Szybki Start

from pydantic_ai import Agent
from database_pydantic_ai import (
    SQLiteDatabase,
    SQLDatabaseDeps,
    SQLITE_SYSTEM_PROMPT,
    create_database_toolset,
)

async with SQLiteDatabase("data.db") as db:
    deps = SQLDatabaseDeps(database=db, read_only=True)
    agent = Agent(
        "openai:gpt-4o",
        deps_type=SQLDatabaseDeps,
        toolsets=[create_database_toolset()],
        system_prompt=SQLITE_SYSTEM_PROMPT,
    )
    result = await agent.run("What are the top 5 most expensive products?", deps=deps)

Przypadki Użycia