From 61c6db96ff87b58a608d6a150267cc9a52fd4cf5 Mon Sep 17 00:00:00 2001
From: David Montague <35119617+dmontagu@users.noreply.github.com>
Date: Wed, 9 Apr 2025 08:51:22 -0600
Subject: [PATCH 1/2] Fix instrumented model

---
 pydantic_ai_slim/pydantic_ai/models/instrumented.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pydantic_ai_slim/pydantic_ai/models/instrumented.py b/pydantic_ai_slim/pydantic_ai/models/instrumented.py
index bca315b281..52219eec26 100644
--- a/pydantic_ai_slim/pydantic_ai/models/instrumented.py
+++ b/pydantic_ai_slim/pydantic_ai/models/instrumented.py
@@ -146,6 +146,9 @@ async def request_stream(
                 if response_stream:
                     finish(response_stream.get(), response_stream.usage())
 
+    def customize_request_parameters(self, model_request_parameters: ModelRequestParameters) -> ModelRequestParameters:
+        return self.wrapped.customize_request_parameters(model_request_parameters)
+
     @contextmanager
     def _instrument(
         self,

From f27770b4a4cc3eb44f3959c1975bf097ef9a6b3c Mon Sep 17 00:00:00 2001
From: David Montague <35119617+dmontagu@users.noreply.github.com>
Date: Wed, 9 Apr 2025 08:52:50 -0600
Subject: [PATCH 2/2] Fix the fix

---
 pydantic_ai_slim/pydantic_ai/models/instrumented.py | 3 ---
 pydantic_ai_slim/pydantic_ai/models/wrapper.py      | 3 +++
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pydantic_ai_slim/pydantic_ai/models/instrumented.py b/pydantic_ai_slim/pydantic_ai/models/instrumented.py
index 52219eec26..bca315b281 100644
--- a/pydantic_ai_slim/pydantic_ai/models/instrumented.py
+++ b/pydantic_ai_slim/pydantic_ai/models/instrumented.py
@@ -146,9 +146,6 @@ async def request_stream(
                 if response_stream:
                     finish(response_stream.get(), response_stream.usage())
 
-    def customize_request_parameters(self, model_request_parameters: ModelRequestParameters) -> ModelRequestParameters:
-        return self.wrapped.customize_request_parameters(model_request_parameters)
-
     @contextmanager
     def _instrument(
         self,
diff --git a/pydantic_ai_slim/pydantic_ai/models/wrapper.py b/pydantic_ai_slim/pydantic_ai/models/wrapper.py
index 547ada8b66..30f18674f0 100644
--- a/pydantic_ai_slim/pydantic_ai/models/wrapper.py
+++ b/pydantic_ai_slim/pydantic_ai/models/wrapper.py
@@ -37,6 +37,9 @@ async def request_stream(
         async with self.wrapped.request_stream(messages, model_settings, model_request_parameters) as response_stream:
             yield response_stream
 
+    def customize_request_parameters(self, model_request_parameters: ModelRequestParameters) -> ModelRequestParameters:
+        return self.wrapped.customize_request_parameters(model_request_parameters)
+
     @property
     def model_name(self) -> str:
         return self.wrapped.model_name