Database Toolset for Pydantic AI

KI-Agenten mit SQL-Datenbankfähigkeiten ausstatten

SQL-Datenbank-Toolset für KI-Agenten zum Abfragen, Analysieren und Interagieren mit Datenbanken. Schema-Introspektion und sichere Abfrageausführung.

Installation

pip install database-pydantic-ai

GitHub PyPI

Fünf Tools geben Agenten volle Datenbankinteraktion: list_tables, get_schema, describe_table, explain_query und query. Unterstützt sowohl SQLite (via aiosqlite) als auch PostgreSQL (via asyncpg) mit einheitlichem Interface. Security-first: Read-only-Modus blockiert 15+ gefährliche SQL-Keywords inklusive DROP, DELETE und ALTER. Ressourcenkontrolle mit konfigurierbaren Query-Timeouts (30s Standard) und Zeilenlimits (100 Standard). Kommentar-bewusstes Parsing erkennt gefährliche Keywords hinter -- und /* */ Kommentaren.

Features

— SQL-Abfrageausführung

— Schema-Introspektion

— Ergebnisformatierung

— Standardmäßig sicher

Schnellstart

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)

Anwendungsfälle

Datenanalyse-Agenten

Agenten Schemas erkunden, Queries schreiben und Ergebnisse aus SQLite- oder PostgreSQL-Datenbanken in natürlicher Sprache analysieren lassen.

Berichterstellung

Agenten bauen, die Produktionsdatenbanken abfragen, Ergebnisse formatieren und automatisch lesbare Berichte generieren.

Datenbank-Administration

Assistenten erstellen, die bei Schema-Discovery, Query-Optimierung via EXPLAIN und Datenbankdokumentation helfen.

Business Intelligence

Nicht-technischen Benutzern natürlichsprachigen Datenzugriff mit Read-only-Sicherheitsgarantien und Query-Timeouts geben.

Bereit, deinen ersten Produktions-KI-Agenten zu bauen?

Open-Source-Tools, kampferprobte Patterns, null Boilerplate. Konfiguriere deinen Stack und shippe in Minuten — nicht Monaten.

Deine KI-App bauen