formatting

Author: dbschmigelski

src/strands/models/litellm.py

@@ -216,7 +216,7 @@ async def structured_output(
 
         if len(response.choices) > 1:
             raise ValueError("Multiple choices found in the response.")
-        
+
         # Find the first choice with tool_calls
         for choice in response.choices:
             if choice.finish_reason == "tool_calls":

tests/strands/models/test_litellm.py

@@ -289,6 +289,18 @@ async def test_structured_output(litellm_acompletion, model, test_output_model_c
     assert tru_result == exp_result
 
 
+@pytest.mark.asyncio
+async def test_structured_output_unsupported_model(litellm_acompletion, model, test_output_model_cls):
+    messages = [{"role": "user", "content": [{"text": "Generate a person"}]}]
+
+    with unittest.mock.patch.object(strands.models.litellm, "supports_response_schema", return_value=False):
+        with pytest.raises(ValueError, match="Model does not support response_format"):
+            stream = model.structured_output(test_output_model_cls, messages)
+            await stream.__anext__()
+
+    litellm_acompletion.assert_not_called()
+
+
 def test_config_validation_warns_on_unknown_keys(litellm_acompletion, captured_warnings):
     """Test that unknown config keys emit a warning."""
     LiteLLMModel(client_args={"api_key": "test"}, model_id="test-model", invalid_param="test")

tests_integ/models/test_model_litellm.py

@@ -44,13 +44,13 @@ class Weather(pydantic.BaseModel):
 def yellow_color():
     class Color(pydantic.BaseModel):
         """Describes a color with its basic name.
-        
+
         Used to extract and normalize color names from text or images.
         The color name should be a simple, common color like 'red', 'blue', 'yellow', etc.
         """
 
         simple_color_name: str = pydantic.Field(
-            description="The basic color name (e.g., 'red', 'blue', 'yellow', 'green', 'orange', 'purple', 'black', 'white')"
+            description="The basic color name (e.g., 'red', 'blue', 'yellow', 'green', 'orange', 'purple')"
         )
 
         @pydantic.field_validator("simple_color_name", mode="after")