add response_id to StreamedResponse

Author: GDaamn

pydantic_ai_slim/pydantic_ai/models/__init__.py

@@ -609,6 +609,7 @@ def get(self) -> ModelResponse:
             timestamp=self.timestamp,
             usage=self.usage(),
             provider_name=self.provider_name,
+            provider_response_id=self.provider_response_id,
         )
 
     def usage(self) -> RequestUsage:
@@ -627,6 +628,11 @@ def provider_name(self) -> str | None:
         """Get the provider name."""
         raise NotImplementedError()
 
+    @property
+    def provider_response_id(self) -> str | None:
+        """Get the provider response id."""
+        return None
+
     @property
     @abstractmethod
     def timestamp(self) -> datetime:

pydantic_ai_slim/pydantic_ai/models/openai.py

@@ -532,6 +532,7 @@ async def _process_streamed_response(
             _response=peekable_response,
             _timestamp=number_to_datetime(first_chunk.created),
             _provider_name=self._provider.name,
+            _provider_response_id=first_chunk.id,
         )
 
     def _get_tools(self, model_request_parameters: ModelRequestParameters) -> list[chat.ChatCompletionToolParam]:
@@ -847,6 +848,7 @@ async def _process_streamed_response(
             _response=peekable_response,
             _timestamp=number_to_datetime(first_chunk.response.created_at),
             _provider_name=self._provider.name,
+            _provider_response_id=first_chunk.response.id,
         )
 
     @overload
@@ -1161,6 +1163,7 @@ class OpenAIStreamedResponse(StreamedResponse):
     _response: AsyncIterable[ChatCompletionChunk]
     _timestamp: datetime
     _provider_name: str
+    _provider_response_id: str
 
     async def _get_event_iterator(self) -> AsyncIterator[ModelResponseStreamEvent]:
         async for chunk in self._response:
@@ -1209,6 +1212,11 @@ def provider_name(self) -> str:
         """Get the provider name."""
         return self._provider_name
 
+    @property
+    def provider_response_id(self) -> str:
+        """Get the provider response id."""
+        return self._provider_response_id
+
     @property
     def timestamp(self) -> datetime:
         """Get the timestamp of the response."""
@@ -1223,6 +1231,7 @@ class OpenAIResponsesStreamedResponse(StreamedResponse):
     _response: AsyncIterable[responses.ResponseStreamEvent]
     _timestamp: datetime
     _provider_name: str
+    _provider_response_id: str
 
     async def _get_event_iterator(self) -> AsyncIterator[ModelResponseStreamEvent]:  # noqa: C901
         async for chunk in self._response:
@@ -1348,6 +1357,11 @@ def provider_name(self) -> str:
         """Get the provider name."""
         return self._provider_name
 
+    @property
+    def provider_response_id(self) -> str:
+        """Get the provider response id."""
+        return self._provider_response_id
+
     @property
     def timestamp(self) -> datetime:
         """Get the timestamp of the response."""