pydantic-ai 0.0.54__tar.gz → 0.0.55__tar.gz

{pydantic_ai-0.0.54 → pydantic_ai-0.0.55}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.0.54
+Version: 0.0.55
 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,bedrock,cli,cohere,evals,groq,mcp,mistral,openai,vertexai]==0.0.54
+Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cli,cohere,evals,groq,mcp,mistral,openai,vertexai]==0.0.55
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.0.54; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.0.55; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.11.0; extra == 'logfire'
 Description-Content-Type: text/markdown

{pydantic_ai-0.0.54 → pydantic_ai-0.0.55}/tests/models/test_model.py

@@ -11,6 +11,7 @@ TEST_CASES = [
     ('OPENAI_API_KEY', 'openai:gpt-3.5-turbo', 'gpt-3.5-turbo', 'openai', 'openai', 'OpenAIModel'),
     ('OPENAI_API_KEY', 'gpt-3.5-turbo', 'gpt-3.5-turbo', 'openai', 'openai', 'OpenAIModel'),
     ('OPENAI_API_KEY', 'o1', 'o1', 'openai', 'openai', 'OpenAIModel'),
+    ('AZURE_OPENAI_API_KEY', 'azure:gpt-3.5-turbo', 'gpt-3.5-turbo', 'azure', 'azure', 'OpenAIModel'),
     ('GEMINI_API_KEY', 'google-gla:gemini-1.5-flash', 'gemini-1.5-flash', 'google-gla', 'gemini', 'GeminiModel'),
     ('GEMINI_API_KEY', 'gemini-1.5-flash', 'gemini-1.5-flash', 'google-gla', 'gemini', 'GeminiModel'),
     (

{pydantic_ai-0.0.54 → pydantic_ai-0.0.55}/tests/models/test_openai.py

@@ -170,7 +170,12 @@ async def test_request_simple_success(allow_model_requests: None):
         ]
     )
     assert get_mock_chat_completion_kwargs(mock_client) == [
-        {'messages': [{'content': 'hello', 'role': 'user'}], 'model': 'gpt-4o', 'n': 1},
+        {
+            'messages': [{'content': 'hello', 'role': 'user'}],
+            'model': 'gpt-4o',
+            'n': 1,
+            'extra_headers': {'User-Agent': IsStr(regex=r'pydantic-ai\/.*')},
+        },
         {
             'messages': [
                 {'content': 'hello', 'role': 'user'},
@@ -179,6 +184,7 @@ async def test_request_simple_success(allow_model_requests: None):
             ],
             'model': 'gpt-4o',
             'n': 1,
+            'extra_headers': {'User-Agent': IsStr(regex=r'pydantic-ai\/.*')},
         },
     ]
 
@@ -551,6 +557,7 @@ async def test_system_prompt_role(
             ],
             'model': 'gpt-4o',
             'n': 1,
+            'extra_headers': {'User-Agent': IsStr(regex=r'pydantic-ai\/.*')},
         }
     ]
 
@@ -626,6 +633,7 @@ async def test_image_url_input(allow_model_requests: None):
                     }
                 ],
                 'n': 1,
+                'extra_headers': {'User-Agent': IsStr(regex=r'pydantic-ai\/.*')},
             }
         ]
     )

{pydantic_ai-0.0.54 → pydantic_ai-0.0.55}/tests/providers/test_provider_names.py

@@ -15,6 +15,7 @@ with try_import() as imports_successful:
     from openai import OpenAIError
 
     from pydantic_ai.providers.anthropic import AnthropicProvider
+    from pydantic_ai.providers.azure import AzureProvider
     from pydantic_ai.providers.cohere import CohereProvider
     from pydantic_ai.providers.deepseek import DeepSeekProvider
     from pydantic_ai.providers.google_gla import GoogleGLAProvider
@@ -28,6 +29,7 @@ with try_import() as imports_successful:
         ('cohere', CohereProvider, 'CO_API_KEY'),
         ('deepseek', DeepSeekProvider, 'DEEPSEEK_API_KEY'),
         ('openai', OpenAIProvider, 'OPENAI_API_KEY'),
+        ('azure', AzureProvider, 'AZURE_OPENAI'),
         ('google-vertex', GoogleVertexProvider, None),
         ('google-gla', GoogleGLAProvider, 'GEMINI_API_KEY'),
         ('groq', GroqProvider, 'GROQ_API_KEY'),