pydantic-ai 0.0.25__tar.gz → 0.0.26__tar.gz

{pydantic_ai-0.0.25 → pydantic_ai-0.0.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.0.25
+Version: 0.0.26
 Summary: Agent Framework / shim to use Pydantic with LLMs
 Project-URL: Homepage, https://ai.pydantic.dev
 Project-URL: Source, https://github.com/pydantic/pydantic-ai
@@ -28,9 +28,9 @@ Classifier: Topic :: Internet
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.9
-Requires-Dist: pydantic-ai-slim[anthropic,cohere,groq,mistral,openai,vertexai]==0.0.25
+Requires-Dist: pydantic-ai-slim[anthropic,cohere,groq,mistral,openai,vertexai]==0.0.26
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.0.25; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.0.26; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=2.3; extra == 'logfire'
 Description-Content-Type: text/markdown

{pydantic_ai-0.0.25 → pydantic_ai-0.0.26}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pydantic-ai"
-version = "0.0.25"
+version = "0.0.26"
 description = "Agent Framework / shim to use Pydantic with LLMs"
 authors = [
     { name = "Samuel Colvin", email = "samuel@pydantic.dev" },
@@ -33,7 +33,7 @@ classifiers = [
 ]
 requires-python = ">=3.9"
 
-dependencies = ["pydantic-ai-slim[openai,vertexai,groq,anthropic,mistral,cohere]==0.0.25"]
+dependencies = ["pydantic-ai-slim[openai,vertexai,groq,anthropic,mistral,cohere]==0.0.26"]
 
 [project.urls]
 Homepage = "https://ai.pydantic.dev"
@@ -42,7 +42,7 @@ Documentation = "https://ai.pydantic.dev"
 Changelog = "https://github.com/pydantic/pydantic-ai/releases"
 
 [project.optional-dependencies]
-examples = ["pydantic-ai-examples==0.0.25"]
+examples = ["pydantic-ai-examples==0.0.26"]
 logfire = ["logfire>=2.3"]
 
 [tool.uv.sources]

{pydantic_ai-0.0.25 → pydantic_ai-0.0.26}/tests/conftest.py

@@ -20,6 +20,8 @@ from typing_extensions import TypeAlias
 from vcr import VCR
 
 import pydantic_ai.models
+from pydantic_ai.messages import BinaryContent
+from pydantic_ai.models import cached_async_http_client
 
 __all__ = 'IsDatetime', 'IsFloat', 'IsNow', 'IsStr', 'TestEnv', 'ClientWithHandler', 'try_import'
 
@@ -197,3 +199,46 @@ def vcr_config():
         'filter_headers': ['authorization', 'x-api-key'],
         'decode_compressed_response': True,
     }
+
+
+@pytest.fixture(autouse=True)
+async def close_cached_httpx_client() -> AsyncIterator[None]:
+    yield
+    await cached_async_http_client().aclose()
+
+
+@pytest.fixture(scope='session')
+def assets_path() -> Path:
+    return Path(__file__).parent / 'assets'
+
+
+@pytest.fixture(scope='session')
+def audio_content(assets_path: Path) -> BinaryContent:
+    audio_bytes = assets_path.joinpath('marcelo.mp3').read_bytes()
+    return BinaryContent(data=audio_bytes, media_type='audio/mpeg')
+
+
+@pytest.fixture(scope='session')
+def image_content(assets_path: Path) -> BinaryContent:
+    image_bytes = assets_path.joinpath('kiwi.png').read_bytes()
+    return BinaryContent(data=image_bytes, media_type='image/png')
+
+
+@pytest.fixture(scope='session')
+def openai_api_key() -> str:
+    return os.getenv('OPENAI_API_KEY', 'mock-api-key')
+
+
+@pytest.fixture(scope='session')
+def gemini_api_key() -> str:
+    return os.getenv('GEMINI_API_KEY', 'mock-api-key')
+
+
+@pytest.fixture(scope='session')
+def groq_api_key() -> str:
+    return os.getenv('GROQ_API_KEY', 'mock-api-key')
+
+
+@pytest.fixture(scope='session')
+def anthropic_api_key() -> str:
+    return os.getenv('ANTHROPIC_API_KEY', 'mock-api-key')

{pydantic_ai-0.0.25 → pydantic_ai-0.0.26}/tests/graph/test_history.py

@@ -53,7 +53,7 @@ async def test_dump_load_history(graph: Graph[MyState, None, int]):
         [
             NodeStep(state=MyState(x=2, y=''), node=Foo(), start_ts=IsNow(tz=timezone.utc), duration=IsFloat()),
             NodeStep(state=MyState(x=2, y='y'), node=Bar(), start_ts=IsNow(tz=timezone.utc), duration=IsFloat()),
-            EndStep(result=End(4), ts=IsNow(tz=timezone.utc)),
+            EndStep(result=End(data=4), ts=IsNow(tz=timezone.utc)),
         ]
     )
     history_json = graph.dump_history(result.history)

{pydantic_ai-0.0.25 → pydantic_ai-0.0.26}/tests/graph/test_state.py

@@ -52,7 +52,7 @@ async def test_run_graph():
                 start_ts=IsNow(tz=timezone.utc),
                 duration=IsFloat(),
             ),
-            EndStep(result=End('x=2 y=y'), ts=IsNow(tz=timezone.utc)),
+            EndStep(result=End(data='x=2 y=y'), ts=IsNow(tz=timezone.utc)),
         ]
     )
     assert state == MyState(x=2, y='y')

{pydantic_ai-0.0.25 → pydantic_ai-0.0.26}/tests/json_body_serializer.py

@@ -13,7 +13,7 @@ else:
         from yaml import Dumper, Loader
 
 FILTERED_HEADER_PREFIXES = ['anthropic-', 'cf-', 'x-']
-FILTERED_HEADERS = {'authorization', 'date', 'request-id', 'server', 'user-agent', 'via'}
+FILTERED_HEADERS = {'authorization', 'date', 'request-id', 'server', 'user-agent', 'via', 'set-cookie'}
 
 
 class LiteralDumper(Dumper):