pydantic-ai-examples 1.0.13__tar.gz → 1.49.0__tar.gz

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/.gitignore

@@ -10,7 +10,7 @@ env*/
 /TODO.md
 /postgres-data/
 .DS_Store
-examples/pydantic_ai_examples/.chat_app_messages.sqlite
+.chat_app_messages.sqlite
 .cache/
 .vscode/
 /question_graph_history.json
@@ -21,3 +21,5 @@ node_modules/
 /test_tmp/
 .mcp.json
 .claude/
+/.cursor/
+/.devcontainer/

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai-examples
-Version: 1.0.13
+Version: 1.49.0
 Summary: Examples of how to use Pydantic AI and what it can do.
 Author-email: Samuel Colvin <samuel@pydantic.dev>, Marcelo Trylesinski <marcelotryle@gmail.com>, David Montague <david@pydantic.dev>, Alex Hall <alex@pydantic.dev>, Douwe Maan <douwe@pydantic.dev>
 License-Expression: MIT
@@ -29,13 +29,13 @@ Requires-Dist: datasets>=4.0.0
 Requires-Dist: devtools>=0.12.2
 Requires-Dist: duckdb>=1.3.2
 Requires-Dist: fastapi>=0.115.4
-Requires-Dist: gradio>=5.9.0
+Requires-Dist: gradio>=5.31.0
 Requires-Dist: logfire[asyncpg,fastapi,httpx,sqlite3]>=3.14.1
-Requires-Dist: mcp[cli]>=1.4.1
+Requires-Dist: mcp[cli]<2.0,>=1.25.0
 Requires-Dist: modal>=1.0.4
 Requires-Dist: pandas>=2.2.3
-Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,groq,openai,vertexai]==1.0.13
-Requires-Dist: pydantic-evals==1.0.13
+Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,groq,openai,vertexai]==1.49.0
+Requires-Dist: pydantic-evals==1.49.0
 Requires-Dist: python-multipart>=0.0.17
 Requires-Dist: rich>=13.9.2
 Requires-Dist: uvicorn>=0.32.0

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/ag_ui/api/agentic_chat.py

@@ -6,9 +6,9 @@ from datetime import datetime
 from zoneinfo import ZoneInfo
 
 from pydantic_ai import Agent
+from pydantic_ai.ui.ag_ui.app import AGUIApp
 
-agent = Agent('openai:gpt-4o-mini')
-app = agent.to_ag_ui()
+agent = Agent('openai:gpt-5-mini')
 
 
 @agent.tool_plain
@@ -23,3 +23,6 @@ async def current_time(timezone: str = 'UTC') -> str:
     """
     tz: ZoneInfo = ZoneInfo(timezone)
     return datetime.now(tz=tz).isoformat()
+
+
+app = AGUIApp(agent)

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/ag_ui/api/agentic_generative_ui.py

@@ -9,6 +9,7 @@ from pydantic import BaseModel, Field
 
 from ag_ui.core import EventType, StateDeltaEvent, StateSnapshotEvent
 from pydantic_ai import Agent
+from pydantic_ai.ui.ag_ui.app import AGUIApp
 
 StepStatus = Literal['pending', 'completed']
 
@@ -26,7 +27,9 @@ class Step(BaseModel):
 class Plan(BaseModel):
     """Represents a plan with multiple steps."""
 
-    steps: list[Step] = Field(default_factory=list, description='The steps in the plan')
+    steps: list[Step] = Field(
+        default_factory=list[Step], description='The steps in the plan'
+    )
 
 
 class JSONPatchOp(BaseModel):
@@ -48,7 +51,7 @@ class JSONPatchOp(BaseModel):
 
 
 agent = Agent(
-    'openai:gpt-4o-mini',
+    'openai:gpt-5-mini',
     instructions=dedent(
         """
         When planning use tools only, without any other messages.
@@ -116,4 +119,4 @@ async def update_plan_step(
     )
 
 
-app = agent.to_ag_ui()
+app = AGUIApp(agent)

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/ag_ui/api/human_in_the_loop.py

@@ -8,9 +8,10 @@ from __future__ import annotations
 from textwrap import dedent
 
 from pydantic_ai import Agent
+from pydantic_ai.ui.ag_ui.app import AGUIApp
 
 agent = Agent(
-    'openai:gpt-4o-mini',
+    'openai:gpt-5-mini',
     instructions=dedent(
         """
         When planning tasks use tools only, without any other messages.
@@ -23,4 +24,4 @@ agent = Agent(
     ),
 )
 
-app = agent.to_ag_ui()
+app = AGUIApp(agent)

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/ag_ui/api/predictive_state_updates.py

@@ -8,7 +8,8 @@ from pydantic import BaseModel
 
 from ag_ui.core import CustomEvent, EventType
 from pydantic_ai import Agent, RunContext
-from pydantic_ai.ag_ui import StateDeps
+from pydantic_ai.ui import StateDeps
+from pydantic_ai.ui.ag_ui.app import AGUIApp
 
 
 class DocumentState(BaseModel):
@@ -17,7 +18,7 @@ class DocumentState(BaseModel):
     document: str = ''
 
 
-agent = Agent('openai:gpt-4o-mini', deps_type=StateDeps[DocumentState])
+agent = Agent('openai:gpt-5-mini', deps_type=StateDeps[DocumentState])
 
 
 # Tools which return AG-UI events will be sent to the client as part of the
@@ -74,4 +75,4 @@ async def story_instructions(ctx: RunContext[StateDeps[DocumentState]]) -> str:
     )
 
 
-app = agent.to_ag_ui(deps=StateDeps(DocumentState()))
+app = AGUIApp(agent, deps=StateDeps(DocumentState()))

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/ag_ui/api/shared_state.py

@@ -2,17 +2,18 @@
 
 from __future__ import annotations
 
-from enum import StrEnum
+from enum import Enum
 from textwrap import dedent
 
 from pydantic import BaseModel, Field
 
 from ag_ui.core import EventType, StateSnapshotEvent
 from pydantic_ai import Agent, RunContext
-from pydantic_ai.ag_ui import StateDeps
+from pydantic_ai.ui import StateDeps
+from pydantic_ai.ui.ag_ui.app import AGUIApp
 
 
-class SkillLevel(StrEnum):
+class SkillLevel(str, Enum):
     """The level of skill required for the recipe."""
 
     BEGINNER = 'Beginner'
@@ -20,7 +21,7 @@ class SkillLevel(StrEnum):
     ADVANCED = 'Advanced'
 
 
-class SpecialPreferences(StrEnum):
+class SpecialPreferences(str, Enum):
     """Special preferences for the recipe."""
 
     HIGH_PROTEIN = 'High Protein'
@@ -32,7 +33,7 @@ class SpecialPreferences(StrEnum):
     VEGAN = 'Vegan'
 
 
-class CookingTime(StrEnum):
+class CookingTime(str, Enum):
     """The cooking time of the recipe."""
 
     FIVE_MIN = '5 min'
@@ -61,18 +62,18 @@ class Recipe(BaseModel):
         description='The skill level required for the recipe',
     )
     special_preferences: list[SpecialPreferences] = Field(
-        default_factory=list,
+        default_factory=list[SpecialPreferences],
         description='Any special preferences for the recipe',
     )
     cooking_time: CookingTime = Field(
         default=CookingTime.FIVE_MIN, description='The cooking time of the recipe'
     )
     ingredients: list[Ingredient] = Field(
-        default_factory=list,
+        default_factory=list[Ingredient],
         description='Ingredients for the recipe',
     )
     instructions: list[str] = Field(
-        default_factory=list, description='Instructions for the recipe'
+        default_factory=list[str], description='Instructions for the recipe'
     )
 
 
@@ -84,7 +85,7 @@ class RecipeSnapshot(BaseModel):
     )
 
 
-agent = Agent('openai:gpt-4o-mini', deps_type=StateDeps[RecipeSnapshot])
+agent = Agent('openai:gpt-5-mini', deps_type=StateDeps[RecipeSnapshot])
 
 
 @agent.tool_plain
@@ -135,4 +136,4 @@ async def recipe_instructions(ctx: RunContext[StateDeps[RecipeSnapshot]]) -> str
     )
 
 
-app = agent.to_ag_ui(deps=StateDeps(RecipeSnapshot()))
+app = AGUIApp(agent, deps=StateDeps(RecipeSnapshot()))

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/ag_ui/api/tool_based_generative_ui.py

@@ -6,6 +6,7 @@ No special handling is required for this feature.
 from __future__ import annotations
 
 from pydantic_ai import Agent
+from pydantic_ai.ui.ag_ui.app import AGUIApp
 
-agent = Agent('openai:gpt-4o-mini')
-app = agent.to_ag_ui()
+agent = Agent('openai:gpt-5-mini')
+app = AGUIApp(agent)

pydantic_ai_examples-1.49.0/pydantic_ai_examples/bank_support.py

@@ -0,0 +1,101 @@
+"""Small but complete example of using Pydantic AI to build a support agent for a bank.
+
+Run with:
+
+    uv run -m pydantic_ai_examples.bank_support
+"""
+
+import sqlite3
+from dataclasses import dataclass
+
+from pydantic import BaseModel
+
+from pydantic_ai import Agent, RunContext
+
+
+@dataclass
+class DatabaseConn:
+    """A wrapper over the SQLite connection."""
+
+    sqlite_conn: sqlite3.Connection
+
+    async def customer_name(self, *, id: int) -> str | None:
+        res = cur.execute('SELECT name FROM customers WHERE id=?', (id,))
+        row = res.fetchone()
+        if row:
+            return row[0]
+        return None
+
+    async def customer_balance(self, *, id: int) -> float:
+        res = cur.execute('SELECT balance FROM customers WHERE id=?', (id,))
+        row = res.fetchone()
+        if row:
+            return row[0]
+        else:
+            raise ValueError('Customer not found')
+
+
+@dataclass
+class SupportDependencies:
+    customer_id: int
+    db: DatabaseConn
+
+
+class SupportOutput(BaseModel):
+    support_advice: str
+    """Advice returned to the customer"""
+    block_card: bool
+    """Whether to block their card or not"""
+    risk: int
+    """Risk level of query"""
+
+
+support_agent = Agent(
+    'openai:gpt-5',
+    deps_type=SupportDependencies,
+    output_type=SupportOutput,
+    instructions=(
+        'You are a support agent in our bank, give the '
+        'customer support and judge the risk level of their query. '
+        "Reply using the customer's name."
+    ),
+)
+
+
+@support_agent.instructions
+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]) -> str:
+    """Returns the customer's current account balance."""
+    balance = await ctx.deps.db.customer_balance(
+        id=ctx.deps.customer_id,
+    )
+    return f'${balance:.2f}'
+
+
+if __name__ == '__main__':
+    with sqlite3.connect(':memory:') as con:
+        cur = con.cursor()
+        cur.execute('CREATE TABLE customers(id, name, balance)')
+        cur.execute("""
+            INSERT INTO customers VALUES
+                (123, 'John', 123.45)
+        """)
+        con.commit()
+
+        deps = SupportDependencies(customer_id=123, db=DatabaseConn(sqlite_conn=con))
+        result = support_agent.run_sync('What is my balance?', deps=deps)
+        print(result.output)
+        """
+        support_advice='Hello John, your current account balance, including pending transactions, is $123.45.' block_card=False risk=1
+        """
+
+        result = support_agent.run_sync('I just lost my card!', deps=deps)
+        print(result.output)
+        """
+        support_advice="I'm sorry to hear that, John. We are temporarily blocking your card to prevent unauthorized transactions." block_card=True risk=8
+        """

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/chat_app.py

@@ -40,7 +40,7 @@ from pydantic_ai import (
 logfire.configure(send_to_logfire='if-token-present')
 logfire.instrument_pydantic_ai()
 
-agent = Agent('openai:gpt-4o')
+agent = Agent('openai:gpt-5')
 THIS_DIR = Path(__file__).parent
 
 

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/data_analyst.py

@@ -9,7 +9,7 @@ from pydantic_ai import Agent, ModelRetry, RunContext
 
 @dataclass
 class AnalystAgentDeps:
-    output: dict[str, pd.DataFrame] = field(default_factory=dict)
+    output: dict[str, pd.DataFrame] = field(default_factory=dict[str, pd.DataFrame])
 
     def store(self, value: pd.DataFrame) -> str:
         """Store the output in deps and return the reference such as Out[1] to be used by the LLM."""
@@ -26,7 +26,7 @@ class AnalystAgentDeps:
 
 
 analyst_agent = Agent(
-    'openai:gpt-4o',
+    'openai:gpt-5',
     deps_type=AnalystAgentDeps,
     instructions='You are a data analyst and your job is to analyze the data according to the user request.',
 )
@@ -47,7 +47,7 @@ def load_dataset(
     """
     # begin load data from hf
     builder = datasets.load_dataset_builder(path)  # pyright: ignore[reportUnknownMemberType]
-    splits: dict[str, datasets.SplitInfo] = builder.info.splits or {}  # pyright: ignore[reportUnknownMemberType]
+    splits: dict[str, datasets.SplitInfo] = builder.info.splits or {}
     if split not in splits:
         raise ModelRetry(
             f'{split} is not valid for dataset {path}. Valid splits are {",".join(splits.keys())}'
@@ -87,7 +87,7 @@ def run_duckdb(ctx: RunContext[AnalystAgentDeps], dataset: str, sql: str) -> str
     data = ctx.deps.get(dataset)
     result = duckdb.query_df(df=data, virtual_table_name='dataset', sql_query=sql)
     # pass the result as ref (because DuckDB SQL can select many rows, creating another huge dataframe)
-    ref = ctx.deps.store(result.df())  # pyright: ignore[reportUnknownMemberType]
+    ref = ctx.deps.store(result.df())
     return f'Executed SQL, result is `{ref}`'
 
 

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/evals/agent.py

@@ -21,7 +21,7 @@ class TimeRangeDeps:
 
 
 time_range_agent = Agent[TimeRangeDeps, TimeRangeResponse](
-    'gpt-4o',
+    'gpt-5',
     output_type=TimeRangeResponse,  # type: ignore  # we can't yet annotate something as receiving a TypeForm
     deps_type=TimeRangeDeps,
     system_prompt="Convert the user's request into a structured time range.",

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/evals/example_04_compare_models.py

@@ -28,8 +28,8 @@ def compare_models():
         dataset_path, custom_evaluator_types=CUSTOM_EVALUATOR_TYPES
     )
     with logfire.span('Comparing different models for time_range_agent'):
-        with time_range_agent.override(model='openai:gpt-4o'):
-            dataset.evaluate_sync(infer_time_range, name='openai:gpt-4o')
+        with time_range_agent.override(model='openai:gpt-5'):
+            dataset.evaluate_sync(infer_time_range, name='openai:gpt-5')
         with time_range_agent.override(model='openai:o1'):
             dataset.evaluate_sync(infer_time_range, name='openai:o1')
 

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/flight_booking.py

@@ -49,7 +49,7 @@ class Deps:
 
 # This agent is responsible for controlling the flow of the conversation.
 search_agent = Agent[Deps, FlightDetails | NoFlightFound](
-    'openai:gpt-4o',
+    'openai:gpt-5',
     output_type=FlightDetails | NoFlightFound,  # type: ignore
     retries=4,
     system_prompt=(
@@ -60,7 +60,7 @@ search_agent = Agent[Deps, FlightDetails | NoFlightFound](
 
 # This agent is responsible for extracting flight details from web page text.
 extraction_agent = Agent(
-    'openai:gpt-4o',
+    'openai:gpt-5',
     output_type=list[FlightDetails],
     system_prompt='Extract all the flight details from the given text.',
 )
@@ -112,7 +112,7 @@ class Failed(BaseModel):
 
 # This agent is responsible for extracting the user's seat selection
 seat_preference_agent = Agent[None, SeatPreference | Failed](
-    'openai:gpt-4o',
+    'openai:gpt-5',
     output_type=SeatPreference | Failed,
     system_prompt=(
         "Extract the user's seat preference. "

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/pydantic_model.py

@@ -22,7 +22,7 @@ class MyModel(BaseModel):
     country: str
 
 
-model = os.getenv('PYDANTIC_AI_MODEL', 'openai:gpt-4o')
+model = os.getenv('PYDANTIC_AI_MODEL', 'openai:gpt-5')
 print(f'Using model: {model}')
 agent = Agent(model, output_type=MyModel)
 

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/question_graph.py

@@ -26,14 +26,16 @@ from pydantic_graph.persistence.file import FileStatePersistence
 logfire.configure(send_to_logfire='if-token-present')
 logfire.instrument_pydantic_ai()
 
-ask_agent = Agent('openai:gpt-4o', output_type=str)
+ask_agent = Agent('openai:gpt-5', output_type=str)
 
 
 @dataclass
 class QuestionState:
     question: str | None = None
-    ask_agent_messages: list[ModelMessage] = field(default_factory=list)
-    evaluate_agent_messages: list[ModelMessage] = field(default_factory=list)
+    ask_agent_messages: list[ModelMessage] = field(default_factory=list[ModelMessage])
+    evaluate_agent_messages: list[ModelMessage] = field(
+        default_factory=list[ModelMessage]
+    )
 
 
 @dataclass
@@ -65,7 +67,7 @@ class EvaluationOutput(BaseModel, use_attribute_docstrings=True):
 
 
 evaluate_agent = Agent(
-    'openai:gpt-4o',
+    'openai:gpt-5',
     output_type=EvaluationOutput,
     system_prompt='Given a question and answer, evaluate if the answer is correct.',
 )

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/rag.py

@@ -30,6 +30,7 @@ import asyncpg
 import httpx
 import logfire
 import pydantic_core
+from anyio import create_task_group
 from openai import AsyncOpenAI
 from pydantic import TypeAdapter
 from typing_extensions import AsyncGenerator
@@ -48,7 +49,7 @@ class Deps:
     pool: asyncpg.Pool
 
 
-agent = Agent('openai:gpt-4o', deps_type=Deps)
+agent = Agent('openai:gpt-5', deps_type=Deps)
 
 
 @agent.tool
@@ -114,7 +115,7 @@ async def build_search_db():
     async with httpx.AsyncClient() as client:
         response = await client.get(DOCS_JSON)
         response.raise_for_status()
-    sections = sessions_ta.validate_json(response.content)
+    sections = sections_ta.validate_json(response.content)
 
     openai = AsyncOpenAI()
     logfire.instrument_openai(openai)
@@ -126,9 +127,9 @@ async def build_search_db():
                     await conn.execute(DB_SCHEMA)
 
         sem = asyncio.Semaphore(10)
-        async with asyncio.TaskGroup() as tg:
+        async with create_task_group() as tg:
             for section in sections:
-                tg.create_task(insert_doc_section(sem, openai, pool, section))
+                tg.start_soon(insert_doc_section, sem, openai, pool, section)
 
 
 async def insert_doc_section(
@@ -182,7 +183,7 @@ class DocsSection:
         return '\n\n'.join((f'path: {self.path}', f'title: {self.title}', self.content))
 
 
-sessions_ta = TypeAdapter(list[DocsSection])
+sections_ta = TypeAdapter(list[DocsSection])
 
 
 # pyright: reportUnknownMemberType=false

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/slack_lead_qualifier/agent.py

@@ -11,7 +11,7 @@ from .models import Analysis, Profile
 
 ### [agent]
 agent = Agent(
-    'openai:gpt-4o',
+    'openai:gpt-5',
     instructions=dedent(
         """
         When a new person joins our public Slack, please put together a brief snapshot so we can be most useful to them.

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/sql_gen.py

@@ -92,7 +92,7 @@ class InvalidRequest(BaseModel):
 
 Response: TypeAlias = Success | InvalidRequest
 agent = Agent[Deps, Response](
-    'google-gla:gemini-1.5-flash',
+    'google-gla:gemini-2.5-flash',
     # Type ignore while we wait for PEP-0747, nonetheless unions will work fine everywhere else
     output_type=Response,  # type: ignore
     deps_type=Deps,

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/stream_markdown.py

@@ -26,8 +26,8 @@ agent = Agent()
 
 # models to try, and the appropriate env var
 models: list[tuple[KnownModelName, str]] = [
-    ('google-gla:gemini-2.0-flash', 'GEMINI_API_KEY'),
-    ('openai:gpt-4o-mini', 'OPENAI_API_KEY'),
+    ('google-gla:gemini-2.5-flash', 'GEMINI_API_KEY'),
+    ('openai:gpt-5-mini', 'OPENAI_API_KEY'),
     ('groq:llama-3.3-70b-versatile', 'GROQ_API_KEY'),
 ]
 

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/weather_agent.py

@@ -32,7 +32,7 @@ class Deps:
 
 
 weather_agent = Agent(
-    'openai:gpt-4.1-mini',
+    'openai:gpt-5-mini',
     # 'Be concise, reply with one sentence.' is enough for some models (like openai) to use
     # the below tools appropriately, but others like anthropic and gemini require a bit more direction.
     instructions='Be concise, reply with one sentence.',

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/weather_agent_gradio.py

@@ -99,7 +99,6 @@ with gr.Blocks() as demo:
     past_messages = gr.State([])
     chatbot = gr.Chatbot(
         label='Packing Assistant',
-        type='messages',
         avatar_images=(None, 'https://ai.pydantic.dev/img/logo-white.svg'),
         examples=[
             {'text': 'What is the weather like in Miami?'},

{pydantic_ai_examples-1.0.13 → pydantic_ai_examples-1.49.0}/pyproject.toml

@@ -56,8 +56,8 @@ dependencies = [
     "rich>=13.9.2",
     "uvicorn>=0.32.0",
     "devtools>=0.12.2",
-    "gradio>=5.9.0",
-    "mcp[cli]>=1.4.1",
+    "gradio>=5.31.0",
+    "mcp[cli]>=1.25.0,<2.0",
     "modal>=1.0.4",
     "duckdb>=1.3.2",
     "datasets>=4.0.0",

pydantic_ai_examples-1.0.13/pydantic_ai_examples/bank_support.py

@@ -1,95 +0,0 @@
-"""Small but complete example of using Pydantic AI to build a support agent for a bank.
-
-Run with:
-
-    uv run -m pydantic_ai_examples.bank_support
-"""
-
-from dataclasses import dataclass
-
-from pydantic import BaseModel
-
-from pydantic_ai import Agent, RunContext
-
-
-class DatabaseConn:
-    """This is a fake database for example purposes.
-
-    In reality, you'd be connecting to an external database
-    (e.g. PostgreSQL) to get information about customers.
-    """
-
-    @classmethod
-    async def customer_name(cls, *, id: int) -> str | None:
-        if id == 123:
-            return 'John'
-
-    @classmethod
-    async def customer_balance(cls, *, id: int, include_pending: bool) -> float:
-        if id == 123:
-            if include_pending:
-                return 123.45
-            else:
-                return 100.00
-        else:
-            raise ValueError('Customer not found')
-
-
-@dataclass
-class SupportDependencies:
-    customer_id: int
-    db: DatabaseConn
-
-
-class SupportOutput(BaseModel):
-    support_advice: str
-    """Advice returned to the customer"""
-    block_card: bool
-    """Whether to block their card or not"""
-    risk: int
-    """Risk level of query"""
-
-
-support_agent = Agent(
-    'openai:gpt-4o',
-    deps_type=SupportDependencies,
-    output_type=SupportOutput,
-    instructions=(
-        'You are a support agent in our bank, give the '
-        'customer support and judge the risk level of their query. '
-        "Reply using the customer's name."
-    ),
-)
-
-
-@support_agent.instructions
-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
-) -> str:
-    """Returns the customer's current account balance."""
-    balance = await ctx.deps.db.customer_balance(
-        id=ctx.deps.customer_id,
-        include_pending=include_pending,
-    )
-    return f'${balance:.2f}'
-
-
-if __name__ == '__main__':
-    deps = SupportDependencies(customer_id=123, db=DatabaseConn())
-    result = support_agent.run_sync('What is my balance?', deps=deps)
-    print(result.output)
-    """
-    support_advice='Hello John, your current account balance, including pending transactions, is $123.45.' block_card=False risk=1
-    """
-
-    result = support_agent.run_sync('I just lost my card!', deps=deps)
-    print(result.output)
-    """
-    support_advice="I'm sorry to hear that, John. We are temporarily blocking your card to prevent unauthorized transactions." block_card=True risk=8
-    """