add tests

Author: GDaamn

pydantic_ai_slim/pydantic_ai/models/test.py

@@ -83,6 +83,7 @@ class TestModel(Model):
     """
     _model_name: str = field(default='test', repr=False)
     _system: str = field(default='test', repr=False)
+    __provider_response_id: str = field(default='resp_test', repr=False)
 
     def __init__(
         self,
@@ -132,6 +133,7 @@ async def request_stream(
             _structured_response=model_response,
             _messages=messages,
             _provider_name=self._system,
+            _provider_response_id=self.__provider_response_id,
         )
 
     @property
@@ -285,6 +287,7 @@ class TestStreamedResponse(StreamedResponse):
     _structured_response: ModelResponse
     _messages: InitVar[Iterable[ModelMessage]]
     _provider_name: str
+    _provider_response_id: str
     _timestamp: datetime = field(default_factory=_utils.now_utc, init=False)
 
     def __post_init__(self, _messages: Iterable[ModelMessage]):
@@ -327,6 +330,11 @@ def model_name(self) -> str:
         """Get the model name of the response."""
         return self._model_name
 
+    @property
+    def provider_response_id(self) -> str:
+        """Get the provider name."""
+        return self._provider_response_id
+
     @property
     def provider_name(self) -> str:
         """Get the provider name."""

tests/test_streaming.py

@@ -68,6 +68,7 @@ async def ret_a(x: str) -> str:
                     model_name='test',
                     timestamp=IsNow(tz=timezone.utc),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 ),
                 ModelRequest(
                     parts=[
@@ -98,6 +99,7 @@ async def ret_a(x: str) -> str:
                     model_name='test',
                     timestamp=IsNow(tz=timezone.utc),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 ),
                 ModelRequest(
                     parts=[
@@ -112,6 +114,7 @@ async def ret_a(x: str) -> str:
                     model_name='test',
                     timestamp=IsNow(tz=timezone.utc),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 ),
             ]
         )
@@ -230,48 +233,55 @@ def upcase(text: str) -> str:
                     model_name='test',
                     timestamp=IsNow(tz=timezone.utc),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 ),
                 ModelResponse(
                     parts=[TextPart(content='The cat ')],
                     usage=RequestUsage(input_tokens=51, output_tokens=2),
                     model_name='test',
                     timestamp=IsNow(tz=timezone.utc),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 ),
                 ModelResponse(
                     parts=[TextPart(content='The cat sat ')],
                     usage=RequestUsage(input_tokens=51, output_tokens=3),
                     model_name='test',
                     timestamp=IsNow(tz=timezone.utc),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 ),
                 ModelResponse(
                     parts=[TextPart(content='The cat sat on ')],
                     usage=RequestUsage(input_tokens=51, output_tokens=4),
                     model_name='test',
                     timestamp=IsNow(tz=timezone.utc),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 ),
                 ModelResponse(
                     parts=[TextPart(content='The cat sat on the ')],
                     usage=RequestUsage(input_tokens=51, output_tokens=5),
                     model_name='test',
                     timestamp=IsNow(tz=timezone.utc),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 ),
                 ModelResponse(
                     parts=[TextPart(content='The cat sat on the mat.')],
                     usage=RequestUsage(input_tokens=51, output_tokens=7),
                     model_name='test',
                     timestamp=IsNow(tz=timezone.utc),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 ),
                 ModelResponse(
                     parts=[TextPart(content='The cat sat on the mat.')],
                     usage=RequestUsage(input_tokens=51, output_tokens=7),
                     model_name='test',
                     timestamp=IsNow(tz=timezone.utc),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 ),
             ]
         )
@@ -796,6 +806,7 @@ def regular_tool(x: int) -> int:
                 model_name='test',
                 timestamp=IsNow(tz=timezone.utc),
                 provider_name='test',
+                provider_response_id='resp_test',
             ),
             ModelRequest(
                 parts=[
@@ -810,6 +821,7 @@ def regular_tool(x: int) -> int:
                 model_name='test',
                 timestamp=IsNow(tz=timezone.utc),
                 provider_name='test',
+                provider_response_id='resp_test',
             ),
             ModelRequest(
                 parts=[
@@ -914,6 +926,7 @@ def output_validator_simple(data: str) -> str:
             timestamp=IsNow(tz=timezone.utc),
             kind='response',
             provider_name='test',
+            provider_response_id='resp_test',
         )
         for text in [
             '',
@@ -1197,6 +1210,7 @@ def my_tool(x: int) -> int:
                     model_name='test',
                     timestamp=IsDatetime(),
                     provider_name='test',
+                    provider_response_id='resp_test',
                 )
             ]
         )

tests/test_usage_limits.py

@@ -97,6 +97,7 @@ async def ret_a(x: str) -> str:
                         model_name='test',
                         timestamp=IsNow(tz=timezone.utc),
                         provider_name='test',
+                        provider_response_id='resp_test',
                     ),
                     ModelRequest(
                         parts=[