from pydantic import BaseModel, EmailStr# Define the modelclass User(BaseModel):    name: str    age: int    email: EmailStr# Example inputuser_data = {    "name": "Alice",    "age": 25,    "email": "alice@example.com"}# Validate the inputuser = User(**user_data)print(user.name)  # Aliceprint(user.age)   # 25print(user.email) # alice@example.com

pip install 'pydantic-ai-slim[openai,vertexai,logfire]'

from dataclasses import dataclassfrom pydantic import BaseModel, Fieldfrom pydantic_ai import Agent, RunContextfrom bank_database import DatabaseConn@dataclassclass SupportDependencies:      customer_id: int    db: DatabaseConn  class SupportResult(BaseModel):      support_advice: str = Field(description='Advice returned to the customer')    block_card: bool = Field(description="Whether to block the customer's card")    risk: int = Field(description='Risk level of query', ge=0, le=10)support_agent = Agent(      'openai:gpt-4o',      deps_type=SupportDependencies,    result_type=SupportResult,      system_prompt=(          'You are a support agent in our bank, give the '        'customer support and judge the risk level of their query.'    ),)@support_agent.system_prompt  async def add_customer_name(ctx: RunContext[SupportDependencies]) -> str:    customer_name = await ctx.deps.db.customer_name(id=ctx.deps.customer_id)    return f"The customer's name is {customer_name!r}"@support_agent.tool  async def customer_balance(    ctx: RunContext[SupportDependencies], include_pending: bool) -> float:    """Returns the customer's current account balance."""      return await ctx.deps.db.customer_balance(        id=ctx.deps.customer_id,        include_pending=include_pending,    )async def main():    deps = SupportDependencies(customer_id=123, db=DatabaseConn())    result = await support_agent.run('What is my balance?', deps=deps)      print(result.data)      """    support_advice='Hello John, your current account balance,     including pending transactions, is $123.45.' block_card=False risk=1    """    result = await support_agent.run('I just lost my card!', deps=deps)    print(result.data)    """    support_advice="I'm sorry to hear that, John. We are temporarily blocking     your card to prevent unauthorized transactions." block_card=True risk=8    """