pydantic-ai-examples 0.8.1__tar.gz → 1.0.0b1__tar.gz

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai-examples
-Version: 0.8.1
+Version: 1.0.0b1
 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
@@ -17,26 +17,25 @@ Classifier: Operating System :: Unix
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Internet
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Requires-Dist: asyncpg>=0.30.0
 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; python_version > '3.9'
+Requires-Dist: gradio>=5.9.0
 Requires-Dist: logfire[asyncpg,fastapi,httpx,sqlite3]>=3.14.1
-Requires-Dist: mcp[cli]>=1.4.1; python_version >= '3.10'
+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]==0.8.1
-Requires-Dist: pydantic-evals==0.8.1
+Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,groq,openai,vertexai]==1.0.0b1
+Requires-Dist: pydantic-evals==1.0.0b1
 Requires-Dist: python-multipart>=0.0.17
 Requires-Dist: rich>=13.9.2
 Requires-Dist: uvicorn>=0.32.0

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pydantic_ai_examples/chat_app.py

@@ -10,14 +10,14 @@ from __future__ import annotations as _annotations
 import asyncio
 import json
 import sqlite3
-from collections.abc import AsyncIterator
+from collections.abc import AsyncIterator, Callable
 from concurrent.futures.thread import ThreadPoolExecutor
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from datetime import datetime, timezone
 from functools import partial
 from pathlib import Path
-from typing import Annotated, Any, Callable, Literal, TypeVar
+from typing import Annotated, Any, Literal, TypeVar
 
 import fastapi
 import logfire
@@ -25,8 +25,7 @@ from fastapi import Depends, Request
 from fastapi.responses import FileResponse, Response, StreamingResponse
 from typing_extensions import LiteralString, ParamSpec, TypedDict
 
-from pydantic_ai import Agent
-from pydantic_ai.exceptions import UnexpectedModelBehavior
+from pydantic_ai import Agent, UnexpectedModelBehavior
 from pydantic_ai.messages import (
     ModelMessage,
     ModelMessagesTypeAdapter,

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pydantic_ai_examples/evals/custom_evaluators.py

@@ -1,6 +1,11 @@
 from dataclasses import dataclass
 from datetime import timedelta
 
+from pydantic_ai_examples.evals.models import (
+    TimeRangeBuilderSuccess,
+    TimeRangeInputs,
+    TimeRangeResponse,
+)
 from pydantic_evals.evaluators import (
     Evaluator,
     EvaluatorContext,
@@ -8,12 +13,6 @@ from pydantic_evals.evaluators import (
 )
 from pydantic_evals.otel import SpanQuery
 
-from pydantic_ai_examples.evals.models import (
-    TimeRangeBuilderSuccess,
-    TimeRangeInputs,
-    TimeRangeResponse,
-)
-
 
 @dataclass
 class ValidateTimeRange(Evaluator[TimeRangeInputs, TimeRangeResponse]):

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pydantic_ai_examples/evals/example_01_generate_dataset.py

@@ -2,11 +2,10 @@ import asyncio
 from pathlib import Path
 from types import NoneType
 
+from pydantic_ai_examples.evals.models import TimeRangeInputs, TimeRangeResponse
 from pydantic_evals import Dataset
 from pydantic_evals.generation import generate_dataset
 
-from pydantic_ai_examples.evals.models import TimeRangeInputs, TimeRangeResponse
-
 
 async def main():
     dataset = await generate_dataset(

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pydantic_ai_examples/evals/example_02_add_custom_evaluators.py

@@ -1,8 +1,6 @@
 from pathlib import Path
 from types import NoneType
 
-from pydantic_evals import Dataset
-
 from pydantic_ai_examples.evals.custom_evaluators import (
     CUSTOM_EVALUATOR_TYPES,
     AgentCalledTool,
@@ -13,6 +11,7 @@ from pydantic_ai_examples.evals.models import (
     TimeRangeInputs,
     TimeRangeResponse,
 )
+from pydantic_evals import Dataset
 
 
 def main():

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pydantic_ai_examples/evals/example_03_unit_testing.py

@@ -2,7 +2,6 @@ from pathlib import Path
 from types import NoneType
 
 import logfire
-from pydantic_evals import Dataset
 
 from pydantic_ai_examples.evals import infer_time_range
 from pydantic_ai_examples.evals.custom_evaluators import (
@@ -12,6 +11,7 @@ from pydantic_ai_examples.evals.models import (
     TimeRangeInputs,
     TimeRangeResponse,
 )
+from pydantic_evals import Dataset
 
 logfire.configure(
     send_to_logfire='if-token-present',
@@ -29,7 +29,9 @@ def evaluate_dataset():
     report = dataset.evaluate_sync(infer_time_range)
     print(report)
 
-    assertion_pass_rate = report.averages().assertions
+    averages = report.averages()
+    assert averages is not None
+    assertion_pass_rate = averages.assertions
     assert assertion_pass_rate is not None, 'There should be at least one assertion'
     assert assertion_pass_rate > 0.9, (
         f'The assertion pass rate was {assertion_pass_rate:.1%}; it should be above 90%.'

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pydantic_ai_examples/flight_booking.py

@@ -11,9 +11,8 @@ import logfire
 from pydantic import BaseModel, Field
 from rich.prompt import Prompt
 
-from pydantic_ai import Agent, ModelRetry, RunContext
+from pydantic_ai import Agent, ModelRetry, RunContext, RunUsage, UsageLimits
 from pydantic_ai.messages import ModelMessage
-from pydantic_ai.usage import RunUsage, UsageLimits
 
 # 'if-token-present' means nothing will be sent (and the example will work) if you don't have logfire configured
 logfire.configure(send_to_logfire='if-token-present')

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pydantic_ai_examples/rag.py

@@ -34,8 +34,7 @@ from openai import AsyncOpenAI
 from pydantic import TypeAdapter
 from typing_extensions import AsyncGenerator
 
-from pydantic_ai import RunContext
-from pydantic_ai.agent import Agent
+from pydantic_ai import Agent, RunContext
 
 # 'if-token-present' means nothing will be sent (and the example will work) if you don't have logfire configured
 logfire.configure(send_to_logfire='if-token-present')

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pydantic_ai_examples/sql_gen.py

@@ -16,14 +16,13 @@ from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from datetime import date
-from typing import Annotated, Any, Union
+from typing import Annotated, Any, TypeAlias
 
 import asyncpg
 import logfire
 from annotated_types import MinLen
 from devtools import debug
 from pydantic import BaseModel, Field
-from typing_extensions import TypeAlias
 
 from pydantic_ai import Agent, ModelRetry, RunContext, format_as_xml
 
@@ -91,7 +90,7 @@ class InvalidRequest(BaseModel):
     error_message: str
 
 
-Response: TypeAlias = Union[Success, InvalidRequest]
+Response: TypeAlias = Success | InvalidRequest
 agent = Agent[Deps, Response](
     'google-gla:gemini-1.5-flash',
     # Type ignore while we wait for PEP-0747, nonetheless unions will work fine everywhere else

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pyproject.toml

@@ -28,7 +28,6 @@ classifiers = [
     "Programming Language :: Python",
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3 :: Only",
-    "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
@@ -44,7 +43,7 @@ classifiers = [
     "Topic :: Software Development :: Libraries :: Python Modules",
     "Topic :: Internet",
 ]
-requires-python = ">=3.9"
+requires-python = ">=3.10"
 
 [tool.hatch.metadata.hooks.uv-dynamic-versioning]
 dependencies = [
@@ -57,8 +56,8 @@ dependencies = [
     "rich>=13.9.2",
     "uvicorn>=0.32.0",
     "devtools>=0.12.2",
-    "gradio>=5.9.0; python_version>'3.9'",
-    "mcp[cli]>=1.4.1; python_version >= '3.10'",
+    "gradio>=5.9.0",
+    "mcp[cli]>=1.4.1",
     "modal>=1.0.4",
     "duckdb>=1.3.2",
     "datasets>=4.0.0",

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pydantic_ai_examples/evals/example_04_compare_models.py

@@ -2,7 +2,6 @@ from pathlib import Path
 from types import NoneType
 
 import logfire
-from pydantic_evals import Dataset
 
 from pydantic_ai_examples.evals import infer_time_range
 from pydantic_ai_examples.evals.agent import time_range_agent
@@ -13,6 +12,7 @@ from pydantic_ai_examples.evals.models import (
     TimeRangeInputs,
     TimeRangeResponse,
 )
+from pydantic_evals import Dataset
 
 logfire.configure(
     send_to_logfire='if-token-present',

{pydantic_ai_examples-0.8.1 → pydantic_ai_examples-1.0.0b1}/pydantic_ai_examples/question_graph.py

@@ -12,6 +12,9 @@ from pathlib import Path
 
 import logfire
 from groq import BaseModel
+
+from pydantic_ai import Agent, format_as_xml
+from pydantic_ai.messages import ModelMessage
 from pydantic_graph import (
     BaseNode,
     End,
@@ -20,9 +23,6 @@ from pydantic_graph import (
 )
 from pydantic_graph.persistence.file import FileStatePersistence
 
-from pydantic_ai import Agent, format_as_xml
-from pydantic_ai.messages import ModelMessage
-
 # 'if-token-present' means nothing will be sent (and the example will work) if you don't have logfire configured
 logfire.configure(send_to_logfire='if-token-present')
 logfire.instrument_pydantic_ai()