pydantic-ai-examples 1.27.0__py3-none-any.whl → 1.28.0__py3-none-any.whl

pydantic_ai_examples/bank_support.py

@@ -5,6 +5,7 @@ Run with:
     uv run -m pydantic_ai_examples.bank_support
 """
 
+import sqlite3
 from dataclasses import dataclass
 
 from pydantic import BaseModel
@@ -12,25 +13,24 @@ from pydantic import BaseModel
 from pydantic_ai import Agent, RunContext
 
 
+@dataclass
 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
+    """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')
 
@@ -69,27 +69,33 @@ async def add_customer_name(ctx: RunContext[SupportDependencies]) -> str:
 
 
 @support_agent.tool
-async def customer_balance(
-    ctx: RunContext[SupportDependencies], include_pending: bool
-) -> str:
+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,
-        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
-    """
+    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.27.0.dist-info → pydantic_ai_examples-1.28.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai-examples
-Version: 1.27.0
+Version: 1.28.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
@@ -34,8 +34,8 @@ Requires-Dist: logfire[asyncpg,fastapi,httpx,sqlite3]>=3.14.1
 Requires-Dist: mcp[cli]>=1.4.1
 Requires-Dist: modal>=1.0.4
 Requires-Dist: pandas>=2.2.3
-Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,groq,openai,vertexai]==1.27.0
-Requires-Dist: pydantic-evals==1.27.0
+Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,groq,openai,vertexai]==1.28.0
+Requires-Dist: pydantic-evals==1.28.0
 Requires-Dist: python-multipart>=0.0.17
 Requires-Dist: rich>=13.9.2
 Requires-Dist: uvicorn>=0.32.0

{pydantic_ai_examples-1.27.0.dist-info → pydantic_ai_examples-1.28.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 pydantic_ai_examples/__main__.py,sha256=i0LEo2JBOZ-gnHED0ou5Bya43gi7KmOyQ_jKN7M5Ces,1647
-pydantic_ai_examples/bank_support.py,sha256=O9r6yQraDRPibWLOC1wyZu0n_Nr7Exg4u7wEHh25czk,2664
+pydantic_ai_examples/bank_support.py,sha256=5y53eLwfRUu1Ys_qQxd8hzbSMbnkCnd5Cx27fpK8LEo,2937
 pydantic_ai_examples/chat_app.html,sha256=90XhxrpDAT09mPVTn9edEn8PqAD-tHxWkeeMz9r_okQ,2580
 pydantic_ai_examples/chat_app.py,sha256=AX28rI0TR9lTPLDOd0uEkmirvfIZCRyCGr_VdGiOkHM,7053
 pydantic_ai_examples/chat_app.ts,sha256=2KfZ2rJU2o0iCPjelyqEi5sH6vfemzWaa5Evx_VcAE4,3307
@@ -44,7 +44,7 @@ pydantic_ai_examples/slack_lead_qualifier/modal.py,sha256=f464AaeyP-n3UIfvEVVc4D
 pydantic_ai_examples/slack_lead_qualifier/models.py,sha256=WTp6D2WCASXqrjPVT3vGgTSYATLPBM3_cjq9wvXMRao,1586
 pydantic_ai_examples/slack_lead_qualifier/slack.py,sha256=VJVfMeUXYClWUJBLHNuaW8PB2sxjNzpTC-O_AJwcxQ4,833
 pydantic_ai_examples/slack_lead_qualifier/store.py,sha256=04vB4eZWKk_Tx0b9K4QuVI1U24JEyJyBS4X76cui7OI,896
-pydantic_ai_examples-1.27.0.dist-info/METADATA,sha256=taHUTObMyBLvyV4-8LV7XxNTXT-Wz2ymhY1l0MAvFNs,2763
-pydantic_ai_examples-1.27.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-pydantic_ai_examples-1.27.0.dist-info/licenses/LICENSE,sha256=vA6Jc482lEyBBuGUfD1pYx-cM7jxvLYOxPidZ30t_PQ,1100
-pydantic_ai_examples-1.27.0.dist-info/RECORD,,
+pydantic_ai_examples-1.28.0.dist-info/METADATA,sha256=9zRFoZNSLisTvAHLdSywPKojZZ2nqO_OTlmteLAUEr8,2763
+pydantic_ai_examples-1.28.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+pydantic_ai_examples-1.28.0.dist-info/licenses/LICENSE,sha256=vA6Jc482lEyBBuGUfD1pYx-cM7jxvLYOxPidZ30t_PQ,1100
+pydantic_ai_examples-1.28.0.dist-info/RECORD,,