Customer Support Chatbot with Pydantic AI

from pydantic_ai import Agent, RunContext

agent = Agent(
    "openai:gpt-4o",
    system_prompt="You are a helpful customer support agent. Use search_faq to find answers from the knowledge base before responding. Always cite the source FAQ.",
)

@agent.tool
async def search_faq(ctx: RunContext, query: str) -> str:
    """Search the FAQ knowledge base for relevant answers."""
    results = await faq_store.asimilarity_search(query, k=3)
    return "\n\n".join(
        f"Q: {r.metadata['question']}\nA: {r.page_content}"
        for r in results
    )

result = await agent.run("How do I reset my password?")
print(result.output)