Pydantic

git clone https://github.com/SnowLeopard-AI/snowy-examples.git
cd snowy-examples/quickstart/pydantic-ai

export ANTHROPIC_API_KEY=your-anthropic-key
export SNOWLEOPARD_API_KEY=your-snowleopard-key
export SNOWLEOPARD_DATAFILE_ID=your-datafile-id

uv run clai --agent agent:agent

clai ➤ How many superheroes are there?

uv add snowleopard

from pydantic_ai import Agent, RunContext
from snowleopard import SnowLeopardClient

# Your existing agent
agent = Agent(
    'anthropic:claude-sonnet-4-5',
    instructions='Be concise, reply with one sentence.',
)

# Initialize the Snow Leopard client
snowy = SnowLeopardClient(api_key="{your-snowleopard-api-key}")

# Add the data retrieval tool
@agent.tool
def get_data(ctx: RunContext[str], user_query: str) -> str:
    """
    Retrieve data from the database.
    Describe your data here - this becomes part of the agent's context.
    """
    response = snowy.retrieve(user_query=user_query, datafile_id="{your-datafile-id}")
    return str(response)

@agent.tool
def get_data(ctx: RunContext[str], user_query: str) -> str:
    """
    Retrieve customer order data.
    Contains order history, products, and customer information.
    Use this to answer questions about sales, orders, and customers.
    """
    response = snowy.retrieve(user_query=user_query, datafile_id=datafile_id)
    return str(response)