pydantic-ai-examples 1.3.0__py3-none-any.whl → 1.5.0__py3-none-any.whl

pydantic_ai_examples/ag_ui/api/shared_state.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from enum import StrEnum
+from enum import Enum
 from textwrap import dedent
 
 from pydantic import BaseModel, Field
@@ -12,7 +12,7 @@ from pydantic_ai import Agent, RunContext
 from pydantic_ai.ag_ui import StateDeps
 
 
-class SkillLevel(StrEnum):
+class SkillLevel(str, Enum):
     """The level of skill required for the recipe."""
 
     BEGINNER = 'Beginner'
@@ -20,7 +20,7 @@ class SkillLevel(StrEnum):
     ADVANCED = 'Advanced'
 
 
-class SpecialPreferences(StrEnum):
+class SpecialPreferences(str, Enum):
     """Special preferences for the recipe."""
 
     HIGH_PROTEIN = 'High Protein'
@@ -32,7 +32,7 @@ class SpecialPreferences(StrEnum):
     VEGAN = 'Vegan'
 
 
-class CookingTime(StrEnum):
+class CookingTime(str, Enum):
     """The cooking time of the recipe."""
 
     FIVE_MIN = '5 min'

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
@@ -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(

{pydantic_ai_examples-1.3.0.dist-info → pydantic_ai_examples-1.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai-examples
-Version: 1.3.0
+Version: 1.5.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.3.0
-Requires-Dist: pydantic-evals==1.3.0
+Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,groq,openai,vertexai]==1.5.0
+Requires-Dist: pydantic-evals==1.5.0
 Requires-Dist: python-multipart>=0.0.17
 Requires-Dist: rich>=13.9.2
 Requires-Dist: uvicorn>=0.32.0

{pydantic_ai_examples-1.3.0.dist-info → pydantic_ai_examples-1.5.0.dist-info}/RECORD

@@ -8,7 +8,7 @@ pydantic_ai_examples/flight_booking.py,sha256=dPzbP386m85IzEpIPu_oh7kBi-zfdsgFzz
 pydantic_ai_examples/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pydantic_ai_examples/pydantic_model.py,sha256=EQsHBig2bvb0PG_2XDgP9Le5xJ4n8eZJzQDGQYhDykg,775
 pydantic_ai_examples/question_graph.py,sha256=lVh48-vu0zRqx6Tw1Z5nh-L0fXsUqfOs5Ir_zG8128g,5075
-pydantic_ai_examples/rag.py,sha256=lWIusQc1_CQ4GrXRJgteXfDpTfl2AZrd4qZcOXPBC5o,7976
+pydantic_ai_examples/rag.py,sha256=Jbd1fo8LIXp34qCz9tPXPNqYjQUGtUmfK86iWMThR0Q,8011
 pydantic_ai_examples/roulette_wheel.py,sha256=2YHKbGzYOkLsd98hO3ntjM6pChR1UpmsRrLD36Qh5f0,1654
 pydantic_ai_examples/sql_gen.py,sha256=rRp_C0ZvfLu6rkR5wrVvgah6wd92nh1vT4UYurx5pmU,5159
 pydantic_ai_examples/stream_markdown.py,sha256=rdhrR6XkqoeOpfv_xFbRpKDV4drEAd8ZrsFBXVrdvHs,2454
@@ -22,7 +22,7 @@ pydantic_ai_examples/ag_ui/api/agentic_chat.py,sha256=Vrz7TVLHRs28gEdVXQ44R8jJtR
 pydantic_ai_examples/ag_ui/api/agentic_generative_ui.py,sha256=z2MVonvUKw-9vA4RC8Z8Jz0qYYcmzkxY-1CLotXzDDo,3489
 pydantic_ai_examples/ag_ui/api/human_in_the_loop.py,sha256=130SDox5HoYjusPr2ubliI8udhxtNxUS3kD9RPb0zYc,765
 pydantic_ai_examples/ag_ui/api/predictive_state_updates.py,sha256=w98JrwWJfu0fIMdgWbcFcmLz8wZT4bMdfWsCbV9k8Co,2124
-pydantic_ai_examples/ag_ui/api/shared_state.py,sha256=1ok84aE4H0J5pWRhcoftf26Y9BAKqfV20ppisRfH52c,3799
+pydantic_ai_examples/ag_ui/api/shared_state.py,sha256=W3Q5EOUgNuhgJ6ncb_JlvQXf9JjMy-SeMBDTKnofr54,3802
 pydantic_ai_examples/ag_ui/api/tool_based_generative_ui.py,sha256=eT--lWjTzL0S3aIu9C14yeoixLjFXPWqwcdiuIlUAJk,219
 pydantic_ai_examples/evals/__init__.py,sha256=4f1v2o4F-gnUVtlkZU-dpwwwbLhqRxMcZv676atjNLg,115
 pydantic_ai_examples/evals/agent.py,sha256=KjCsUiL28RCNT6NwoQnQCwJ0xRw3EUGdIrYhlIjmVqI,2042
@@ -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.3.0.dist-info/METADATA,sha256=9rh2NL4w-tyrMPZ2Gc71zRSdzq91_YMsubccGjYnEcg,2760
-pydantic_ai_examples-1.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pydantic_ai_examples-1.3.0.dist-info/licenses/LICENSE,sha256=vA6Jc482lEyBBuGUfD1pYx-cM7jxvLYOxPidZ30t_PQ,1100
-pydantic_ai_examples-1.3.0.dist-info/RECORD,,
+pydantic_ai_examples-1.5.0.dist-info/METADATA,sha256=VXu1zKnx4HIj4sBYV_AOBdfUycrW1y8IinDc6e2atwA,2760
+pydantic_ai_examples-1.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pydantic_ai_examples-1.5.0.dist-info/licenses/LICENSE,sha256=vA6Jc482lEyBBuGUfD1pYx-cM7jxvLYOxPidZ30t_PQ,1100
+pydantic_ai_examples-1.5.0.dist-info/RECORD,,