pydantic-ai 0.0.38__tar.gz → 0.0.39__tar.gz

{pydantic_ai-0.0.38 → pydantic_ai-0.0.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.0.38
+Version: 0.0.39
 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,groq,mistral,openai,vertexai]==0.0.38
+Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cli,cohere,groq,mistral,openai,vertexai]==0.0.39
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.0.38; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.0.39; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=2.3; extra == 'logfire'
 Description-Content-Type: text/markdown
@@ -98,11 +98,6 @@ Provides the ability to [stream](https://ai.pydantic.dev/results/#streamed-resul
 * __Graph Support__
 [Pydantic Graph](https://ai.pydantic.dev/graph) provides a powerful way to define graphs using typing hints, this is useful in complex applications where standard control flow can degrade to spaghetti code.
 
-## In Beta!
-
-PydanticAI is in early beta, the API is still subject to change and there's a lot more to do.
-[Feedback](https://github.com/pydantic/pydantic-ai/issues) is very welcome!
-
 ## Hello World Example
 
 Here's a minimal example of PydanticAI:

{pydantic_ai-0.0.38 → pydantic_ai-0.0.39}/README.md

@@ -61,11 +61,6 @@ Provides the ability to [stream](https://ai.pydantic.dev/results/#streamed-resul
 * __Graph Support__
 [Pydantic Graph](https://ai.pydantic.dev/graph) provides a powerful way to define graphs using typing hints, this is useful in complex applications where standard control flow can degrade to spaghetti code.
 
-## In Beta!
-
-PydanticAI is in early beta, the API is still subject to change and there's a lot more to do.
-[Feedback](https://github.com/pydantic/pydantic-ai/issues) is very welcome!
-
 ## Hello World Example
 
 Here's a minimal example of PydanticAI:

{pydantic_ai-0.0.38 → pydantic_ai-0.0.39}/pyproject.toml

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

{pydantic_ai-0.0.38 → pydantic_ai-0.0.39}/tests/models/test_groq.py

@@ -1,11 +1,13 @@
 from __future__ import annotations as _annotations
 
 import json
+import os
 from collections.abc import Sequence
 from dataclasses import dataclass
 from datetime import datetime, timezone
 from functools import cached_property
 from typing import Any, Literal, Union, cast
+from unittest.mock import patch
 
 import httpx
 import pytest
@@ -25,7 +27,7 @@ from pydantic_ai.messages import (
     ToolReturnPart,
     UserPromptPart,
 )
-from pydantic_ai.result import Usage
+from pydantic_ai.usage import Usage
 
 from ..conftest import IsNow, raise_if_exception, try_import
 from .mock_async_stream import MockAsyncStream
@@ -45,6 +47,7 @@ with try_import() as imports_successful:
     from groq.types.completion_usage import CompletionUsage
 
     from pydantic_ai.models.groq import GroqModel
+    from pydantic_ai.providers.groq import GroqProvider
 
     # note: we use Union here so that casting works with Python 3.9
     MockChatCompletion = Union[chat.ChatCompletion, Exception]
@@ -57,7 +60,7 @@ pytestmark = [
 
 
 def test_init():
-    m = GroqModel('llama-3.3-70b-versatile', api_key='foobar')
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(api_key='foobar'))
     assert m.client.api_key == 'foobar'
     assert m.model_name == 'llama-3.3-70b-versatile'
     assert m.system == 'groq'
@@ -121,7 +124,7 @@ def completion_message(message: ChatCompletionMessage, *, usage: CompletionUsage
 async def test_request_simple_success(allow_model_requests: None):
     c = completion_message(ChatCompletionMessage(content='world', role='assistant'))
     mock_client = MockGroq.create_mock(c)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m)
 
     result = await agent.run('hello')
@@ -158,7 +161,7 @@ async def test_request_simple_usage(allow_model_requests: None):
         usage=CompletionUsage(completion_tokens=1, prompt_tokens=2, total_tokens=3),
     )
     mock_client = MockGroq.create_mock(c)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m)
 
     result = await agent.run('Hello')
@@ -180,7 +183,7 @@ async def test_request_structured_response(allow_model_requests: None):
         )
     )
     mock_client = MockGroq.create_mock(c)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m, result_type=list[int])
 
     result = await agent.run('Hello')
@@ -254,7 +257,7 @@ async def test_request_tool_call(allow_model_requests: None):
         completion_message(ChatCompletionMessage(content='final response', role='assistant')),
     ]
     mock_client = MockGroq.create_mock(responses)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m, system_prompt='this is the system prompt')
 
     @agent.tool_plain
@@ -349,7 +352,7 @@ def text_chunk(text: str, finish_reason: FinishReason | None = None) -> chat.Cha
 async def test_stream_text(allow_model_requests: None):
     stream = text_chunk('hello '), text_chunk('world'), chunk([])
     mock_client = MockGroq.create_mock_stream(stream)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m)
 
     async with agent.run_stream('') as result:
@@ -361,7 +364,7 @@ async def test_stream_text(allow_model_requests: None):
 async def test_stream_text_finish_reason(allow_model_requests: None):
     stream = text_chunk('hello '), text_chunk('world'), text_chunk('.', finish_reason='stop')
     mock_client = MockGroq.create_mock_stream(stream)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m)
 
     async with agent.run_stream('') as result:
@@ -408,7 +411,7 @@ async def test_stream_structured(allow_model_requests: None):
         chunk([]),
     )
     mock_client = MockGroq.create_mock_stream(stream)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m, result_type=MyTypedDict)
 
     async with agent.run_stream('') as result:
@@ -459,7 +462,7 @@ async def test_stream_structured_finish_reason(allow_model_requests: None):
         struc_chunk(None, None, finish_reason='stop'),
     )
     mock_client = MockGroq.create_mock_stream(stream)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m, result_type=MyTypedDict)
 
     async with agent.run_stream('') as result:
@@ -479,7 +482,7 @@ async def test_stream_structured_finish_reason(allow_model_requests: None):
 async def test_no_content(allow_model_requests: None):
     stream = chunk([ChoiceDelta()]), chunk([ChoiceDelta()])
     mock_client = MockGroq.create_mock_stream(stream)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m, result_type=MyTypedDict)
 
     with pytest.raises(UnexpectedModelBehavior, match='Received empty model response'):
@@ -490,7 +493,7 @@ async def test_no_content(allow_model_requests: None):
 async def test_no_delta(allow_model_requests: None):
     stream = chunk([]), text_chunk('hello '), text_chunk('world')
     mock_client = MockGroq.create_mock_stream(stream)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m)
 
     async with agent.run_stream('') as result:
@@ -501,7 +504,7 @@ async def test_no_delta(allow_model_requests: None):
 
 @pytest.mark.vcr()
 async def test_image_url_input(allow_model_requests: None, groq_api_key: str):
-    m = GroqModel('llama-3.2-11b-vision-preview', api_key=groq_api_key)
+    m = GroqModel('llama-3.2-11b-vision-preview', provider=GroqProvider(api_key=groq_api_key))
     agent = Agent(m)
 
     result = await agent.run(
@@ -530,7 +533,7 @@ Potatoes are a versatile food that can be prepared in many different ways, such
 async def test_audio_as_binary_content_input(allow_model_requests: None, media_type: str):
     c = completion_message(ChatCompletionMessage(content='world', role='assistant'))
     mock_client = MockGroq.create_mock(c)
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m)
 
     base64_content = b'//uQZ'
@@ -543,7 +546,7 @@ async def test_audio_as_binary_content_input(allow_model_requests: None, media_t
 async def test_image_as_binary_content_input(
     allow_model_requests: None, groq_api_key: str, image_content: BinaryContent
 ) -> None:
-    m = GroqModel('llama-3.2-11b-vision-preview', api_key=groq_api_key)
+    m = GroqModel('llama-3.2-11b-vision-preview', provider=GroqProvider(api_key=groq_api_key))
     agent = Agent(m)
 
     result = await agent.run(['What is the name of this fruit?', image_content])
@@ -560,10 +563,24 @@ def test_model_status_error(allow_model_requests: None) -> None:
             body={'error': 'test error'},
         )
     )
-    m = GroqModel('llama-3.3-70b-versatile', groq_client=mock_client)
+    m = GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(groq_client=mock_client))
     agent = Agent(m)
     with pytest.raises(ModelHTTPError) as exc_info:
         agent.run_sync('hello')
     assert str(exc_info.value) == snapshot(
         "status_code: 500, model_name: llama-3.3-70b-versatile, body: {'error': 'test error'}"
     )
+
+
+async def test_init_with_provider():
+    provider = GroqProvider(api_key='api-key')
+    model = GroqModel('llama3-8b-8192', provider=provider)
+    assert model.model_name == 'llama3-8b-8192'
+    assert model.client == provider.client
+
+
+async def test_init_with_provider_string():
+    with patch.dict(os.environ, {'GROQ_API_KEY': 'env-api-key'}, clear=False):
+        model = GroqModel('llama3-8b-8192', provider='groq')
+        assert model.model_name == 'llama3-8b-8192'
+        assert model.client is not None

pydantic_ai-0.0.39/tests/providers/test_groq.py

@@ -0,0 +1,66 @@
+from __future__ import annotations as _annotations
+
+import os
+import re
+from unittest.mock import patch
+
+import httpx
+import pytest
+
+from ..conftest import try_import
+
+with try_import() as imports_successful:
+    from groq import AsyncGroq
+
+    from pydantic_ai.providers.groq import GroqProvider
+
+
+pytestmark = pytest.mark.skipif(not imports_successful(), reason='groq not installed')
+
+
+def test_groq_provider():
+    provider = GroqProvider(api_key='api-key')
+    assert provider.name == 'groq'
+    assert provider.base_url == 'https://api.groq.com'
+    assert isinstance(provider.client, AsyncGroq)
+    assert provider.client.api_key == 'api-key'
+
+
+def test_groq_provider_need_api_key() -> None:
+    with patch.dict(os.environ, {}, clear=True):
+        with pytest.raises(
+            ValueError,
+            match=re.escape(
+                'Set the `GROQ_API_KEY` environment variable or pass it via `GroqProvider(api_key=...)`'
+                'to use the Groq provider.'
+            ),
+        ):
+            GroqProvider()
+
+
+def test_groq_provider_pass_http_client() -> None:
+    http_client = httpx.AsyncClient()
+    provider = GroqProvider(http_client=http_client, api_key='api-key')
+    assert provider.client._client == http_client  # type: ignore[reportPrivateUsage]
+
+
+def test_groq_provider_pass_groq_client() -> None:
+    groq_client = AsyncGroq(api_key='api-key')
+    provider = GroqProvider(groq_client=groq_client)
+    assert provider.client == groq_client
+
+
+def test_groq_provider_with_env_base_url(monkeypatch: pytest.MonkeyPatch) -> None:
+    # Test with environment variable for base_url
+    monkeypatch.setenv('GROQ_BASE_URL', 'https://custom.groq.com/v1')
+    provider = GroqProvider(api_key='api-key')
+    assert provider.base_url == 'https://custom.groq.com/v1'
+
+
+def test_infer_groq_provider():
+    with patch.dict(os.environ, {'GROQ_API_KEY': 'test-api-key'}, clear=False):
+        from pydantic_ai.providers import infer_provider
+
+        provider = infer_provider('groq')
+        assert provider.name == 'groq'
+        assert isinstance(provider, GroqProvider)

{pydantic_ai-0.0.38 → pydantic_ai-0.0.39}/tests/providers/test_provider_names.py

@@ -14,6 +14,7 @@ with try_import() as imports_successful:
     from pydantic_ai.providers.deepseek import DeepSeekProvider
     from pydantic_ai.providers.google_gla import GoogleGLAProvider
     from pydantic_ai.providers.google_vertex import GoogleVertexProvider
+    from pydantic_ai.providers.groq import GroqProvider
     from pydantic_ai.providers.openai import OpenAIProvider
 
     test_infer_provider_params = [
@@ -21,6 +22,7 @@ with try_import() as imports_successful:
         ('openai', OpenAIProvider, None),
         ('google-vertex', GoogleVertexProvider, None),
         ('google-gla', GoogleGLAProvider, 'GEMINI_API_KEY'),
+        ('groq', GroqProvider, 'GROQ_API_KEY'),
     ]
 
 if not imports_successful():

{pydantic_ai-0.0.38 → pydantic_ai-0.0.39}/tests/test_live.py

@@ -50,8 +50,9 @@ def vertexai(http_client: httpx.AsyncClient, tmp_path: Path) -> Model:
 
 def groq(http_client: httpx.AsyncClient, _tmp_path: Path) -> Model:
     from pydantic_ai.models.groq import GroqModel
+    from pydantic_ai.providers.groq import GroqProvider
 
-    return GroqModel('llama-3.3-70b-versatile', http_client=http_client)
+    return GroqModel('llama-3.3-70b-versatile', provider=GroqProvider(http_client=http_client))
 
 
 def anthropic(http_client: httpx.AsyncClient, _tmp_path: Path) -> Model: