pydantic-ai-rag 0.0.0

RAG layer for pydantic-ai agents. Part of pydantic-ai-toolkits.

pydantic-ai-rag

This project is included in pydantic-ai-toolbox.

Install

pip install "pydantic-ai-toolbox[rag]"

Usage

from pydantic_ai import Agent
from pydantic_ai_toolbox import RAGToolset

rag = RAGToolset(
    embedder=my_embedder,
    chunk_size=1000,
    chunk_overlap=100,
    storage_path="./index",
    distance="cosine",
    max_results=20,
    namespace="default",
)
rag.add_text("The sky is green.", doc_id="d-sky")

agent = Agent("openai:gpt-4o", toolsets=[rag])
agent.run_sync("What color is the sky?")

embedder is any callable mapping list[str] -> list[list[float]] (OpenAI, a local model, or a stub for tests).

Full reference and a runnable example: RAG.md.