Skip to content

Commit 8df053c

Browse files
committed
fix more tests
1 parent 831aead commit 8df053c

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

pydantic_ai_slim/pydantic_ai/models/fallback.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,9 @@ async def request_stream(
125125
def profile(self) -> ModelProfile:
126126
raise NotImplementedError('FallbackModel does not have its own model profile.')
127127

128+
def customize_request_parameters(self, model_request_parameters: ModelRequestParameters) -> ModelRequestParameters:
129+
return model_request_parameters
130+
128131
def _set_span_attributes(self, model: Model):
129132
with suppress(Exception):
130133
span = get_current_span()

tests/models/test_fallback.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,7 @@ def test_first_failed_instrumented(capfire: CaptureLogfire) -> None:
138138
'output_mode': 'text',
139139
'output_object': None,
140140
'output_tools': [],
141+
'prompted_output_template': None,
141142
'allow_text_output': True,
142143
'allow_image_output': False,
143144
},
@@ -245,6 +246,7 @@ async def test_first_failed_instrumented_stream(capfire: CaptureLogfire) -> None
245246
'output_mode': 'text',
246247
'output_object': None,
247248
'output_tools': [],
249+
'prompted_output_template': None,
248250
'allow_text_output': True,
249251
'allow_image_output': False,
250252
},
@@ -354,6 +356,7 @@ def test_all_failed_instrumented(capfire: CaptureLogfire) -> None:
354356
'output_mode': 'text',
355357
'output_object': None,
356358
'output_tools': [],
359+
'prompted_output_template': None,
357360
'allow_text_output': True,
358361
'allow_image_output': False,
359362
},

0 commit comments

Comments
 (0)