coverage

Author: DouweM

pydantic_ai_slim/pydantic_ai/models/__init__.py

@@ -446,7 +446,7 @@ def prepare_request(
         if params.output_object and params.output_mode not in ('native', 'prompted'):
             params = replace(params, output_object=None)
         if params.prompted_output_template and params.output_mode != 'prompted':
-            params = replace(params, prompted_output_template=None)
+            params = replace(params, prompted_output_template=None)  # pragma: no cover
 
         # Set default prompted output template
         if params.output_mode == 'prompted' and not params.prompted_output_template:

pydantic_ai_slim/pydantic_ai/models/anthropic.py

@@ -252,7 +252,9 @@ def prepare_request(
         ):
             if model_request_parameters.output_mode == 'auto':
                 model_request_parameters = replace(model_request_parameters, output_mode='prompted')
-            elif model_request_parameters.output_mode == 'tool' and not model_request_parameters.allow_text_output:
+            elif (
+                model_request_parameters.output_mode == 'tool' and not model_request_parameters.allow_text_output
+            ):  # pragma: no branch
                 # This would result in `tool_choice=required`, which Anthropic does not support with thinking.
                 raise UserError(
                     'Anthropic does not support thinking and output tools at the same time. Use `output_type=PromptedOutput(...)` instead.'

pydantic_ai_slim/pydantic_ai/models/fallback.py

@@ -128,7 +128,7 @@ def profile(self) -> ModelProfile:
         raise NotImplementedError('FallbackModel does not have its own model profile.')
 
     def customize_request_parameters(self, model_request_parameters: ModelRequestParameters) -> ModelRequestParameters:
-        return model_request_parameters
+        return model_request_parameters  # pragma: no cover
 
     def prepare_request(
         self, model_settings: ModelSettings | None, model_request_parameters: ModelRequestParameters

tests/models/test_fallback.py

@@ -684,7 +684,7 @@ def native_output_func(_: list[ModelMessage], info: AgentInfo) -> ModelResponse:
     def prompted_output_func(_: list[ModelMessage], info: AgentInfo) -> ModelResponse:
         nonlocal enabled_model
         if enabled_model != 'prompted':
-            raise ModelHTTPError(status_code=500, model_name='prompted-model', body=None)
+            raise ModelHTTPError(status_code=500, model_name='prompted-model', body=None)  # pragma: no cover
 
         assert info.model_request_parameters == snapshot(
             ModelRequestParameters(
@@ -848,9 +848,7 @@ def prompted_output_func(_: list[ModelMessage], info: AgentInfo) -> ModelRespons
                     'gen_ai.output.messages': [
                         {'role': 'assistant', 'parts': [{'type': 'text', 'content': '{"bar":"baz"}'}]}
                     ],
-                    'gen_ai.system_instructions': [
-                        {'type': 'text', 'content': 'Be kind'}
-                    ],  # TODO (DouweM): We'd expect the prompted output instructions here as well
+                    'gen_ai.system_instructions': [{'type': 'text', 'content': 'Be kind'}],
                     'gen_ai.usage.input_tokens': 51,
                     'gen_ai.usage.output_tokens': 4,
                     'gen_ai.response.model': 'function:prompted_output_func:',