Database Toolset for Pydantic AI

Potencia agentes IA con capacidades de bases de datos SQL

Toolset de bases de datos SQL para agentes IA. Consultas, análisis e interacción con bases de datos. Introspección de esquema y ejecución segura.

Instalación

pip install database-pydantic-ai

Cinco herramientas dan a los agentes interacción completa con bases de datos: list_tables, get_schema, describe_table, explain_query y query. Soporta tanto SQLite (via aiosqlite) como PostgreSQL (via asyncpg) con interfaz unificada. Seguridad primero: el modo read-only bloquea 15+ palabras clave SQL peligrosas incluyendo DROP, DELETE y ALTER. Control de recursos con timeouts configurables (30s por defecto) y límites de filas (100 por defecto). Parsing consciente de comentarios detecta palabras clave peligrosas ocultas tras -- y /* */.

Características

Inicio Rápido

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)

Casos de Uso