Pydantic AI

from dotenv import load_dotenv
from pydantic_ai import Agent, RunContext
from pydantic_ai.models.google import GoogleModel


load_dotenv(override=True)

model = GoogleModel("gemini-2.5-flash")
agent = Agent(
    model=model,
    system_prompt=(
        "You are a helpful calculator assistant. You must use the tools "
        "according to the BODMAS rule."
    ),
    # this is required for OpenInference instrumentation to capture the agent trajectory.
    instrument=True,
)


@agent.tool
def add(ctx: RunContext, a: float, b: float) -> float:
    """Add two numbers."""
    return a + b


@agent.tool
def subtract(ctx: RunContext, a: float, b: float) -> float:
    """Subtract two numbers."""
    return a - b


@agent.tool
def multiply(ctx: RunContext, a: float, b: float) -> float:
    """Multiply two numbers."""
    return a * b


@agent.tool
def divide(ctx: RunContext, a: float, b: float) -> float:
    """Divide two numbers."""
    return a / b

from quraite.tracing import Framework, setup_tracing

tracer_provider = setup_tracing([Framework.PYDANTIC_AI])

from quraite.adapters import PydanticAIAdapter

pydantic_adapter = PydanticAIAdapter(
    agent=agent,
    tracer_provider=tracer_provider,
)

if __name__ == "__main__":
    from quraite import run_agent

    run_agent(
        pydantic_adapter,
        port=8080,
        host="0.0.0.0",
        tunnel="cloudflare",
    )