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

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

@@ -21,3 +21,5 @@ node_modules/
 /test_tmp/
 .mcp.json
 .claude/
+/.cursor/
+/.devcontainer/

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai-examples
-Version: 1.9.0
+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.9.0
-Requires-Dist: pydantic-evals==1.9.0
+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.9.0 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/ag_ui/api/agentic_generative_ui.py

@@ -27,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):

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

@@ -62,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'
     )
 
 

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.9.0 → 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."""
@@ -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.9.0 → pydantic_ai_examples-1.49.0}/pydantic_ai_examples/question_graph.py

@@ -32,8 +32,10 @@ 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

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

@@ -115,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)
@@ -183,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.9.0 → 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.9.0 → 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.9.0/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-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], 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
-    """