Skip to content

Commit 73f3a81

Browse files
committed
chore: test model coverage
Address TestModel coverage issue due to logic structure in _get_output. This is not introduced by the current PR but was preventing it from passing coverage checks.
1 parent f1fd950 commit 73f3a81

File tree

1 file changed

+4
-4
lines changed
  • pydantic_ai_slim/pydantic_ai/models

1 file changed

+4
-4
lines changed

pydantic_ai_slim/pydantic_ai/models/test.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -184,14 +184,14 @@ def _get_output(self, model_request_parameters: ModelRequestParameters) -> _Wrap
184184

185185
if k := output_tool.outer_typed_dict_key:
186186
return _WrappedToolOutput({k: self.custom_output_args})
187-
else:
188-
return _WrappedToolOutput(self.custom_output_args)
187+
188+
return _WrappedToolOutput(self.custom_output_args)
189189
elif model_request_parameters.allow_text_output:
190190
return _WrappedTextOutput(None)
191-
elif model_request_parameters.output_tools:
191+
elif model_request_parameters.output_tools: # pragma: no branch
192192
return _WrappedToolOutput(None)
193193
else:
194-
return _WrappedTextOutput(None)
194+
return _WrappedTextOutput(None) # pragma: no cover
195195

196196
def _node_response(
197197
self,

0 commit comments

Comments
 (0)