Skip to content

Commit a5a078e

Browse files
authored
Fix bug with JSON schema generation when using InstrumentedModel (#1424)
1 parent 7f1e03d commit a5a078e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pydantic_ai_slim/pydantic_ai/models/wrapper.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,9 @@ async def request_stream(
3737
async with self.wrapped.request_stream(messages, model_settings, model_request_parameters) as response_stream:
3838
yield response_stream
3939

40+
def customize_request_parameters(self, model_request_parameters: ModelRequestParameters) -> ModelRequestParameters:
41+
return self.wrapped.customize_request_parameters(model_request_parameters)
42+
4043
@property
4144
def model_name(self) -> str:
4245
return self.wrapped.model_name

0 commit comments

Comments
 (0)