update; model response

Author: mahiro72

pydantic_ai_slim/pydantic_ai/durable_exec/dbos/_model.py

@@ -46,6 +46,10 @@ def model_name(self) -> str:
     def provider_name(self) -> str:
         return self.response.provider_name or ''  # pragma: no cover
 
+    @property
+    def provider_url(self) -> str | None:
+        return self.response.provider_url  # pragma: no cover
+
     @property
     def timestamp(self) -> datetime:
         return self.response.timestamp  # pragma: no cover

pydantic_ai_slim/pydantic_ai/durable_exec/prefect/_model.py

@@ -55,6 +55,10 @@ def model_name(self) -> str:
     def provider_name(self) -> str:
         return self.response.provider_name or ''  # pragma: no cover
 
+    @property
+    def provider_url(self) -> str | None:
+        return self.response.provider_url  # pragma: no cover
+
     @property
     def timestamp(self) -> datetime:
         return self.response.timestamp  # pragma: no cover

pydantic_ai_slim/pydantic_ai/messages.py

@@ -1249,6 +1249,9 @@ class ModelResponse:
     provider_name: str | None = None
     """The name of the LLM provider that generated the response."""
 
+    provider_url: str | None = None
+    """The base URL of the LLM provider that generated the response."""
+
     provider_details: Annotated[
         dict[str, Any] | None,
         # `vendor_details` is deprecated, but we still want to support deserializing model responses stored in a DB before the name was changed
@@ -1341,6 +1344,7 @@ def cost(self) -> genai_types.PriceCalculation:
             self.usage,
             self.model_name,
             provider_id=self.provider_name,
+            provider_api_url=self.provider_url,
             genai_request_timestamp=self.timestamp,
         )