diff --git a/litellm/llms/bedrock/chat/invoke_transformations/amazon_mistral_transformation.py b/litellm/llms/bedrock/chat/invoke_transformations/amazon_mistral_transformation.py index 8ad93ccabf8..61546b659a8 100644 --- a/litellm/llms/bedrock/chat/invoke_transformations/amazon_mistral_transformation.py +++ b/litellm/llms/bedrock/chat/invoke_transformations/amazon_mistral_transformation.py @@ -102,7 +102,7 @@ class AmazonMistralConfig(AmazonInvokeConfig, BaseConfig): if "choices" in completion_response and len(completion_response["choices"]) > 0: outputText = completion_response["choices"][0]["message"]["content"] model_response.choices[0].finish_reason = completion_response["choices"][0]["finish_reason"] - elif "outputs" in completion_response: + elif "outputs" in completion_response and len(completion_response["outputs"]) > 0: outputText = completion_response["outputs"][0]["text"] model_response.choices[0].finish_reason = completion_response["outputs"][0]["stop_reason"] else: diff --git a/tests/test_litellm/llms/bedrock/chat/test_mistral_config.py b/tests/test_litellm/llms/bedrock/chat/test_mistral_config.py index b5cb5e92ff5..9deb3a7a5ed 100644 --- a/tests/test_litellm/llms/bedrock/chat/test_mistral_config.py +++ b/tests/test_litellm/llms/bedrock/chat/test_mistral_config.py @@ -42,3 +42,12 @@ def test_mistral_get_outputText_empty_choices(): AmazonMistralConfig.get_outputText( completion_response={"choices": []}, model_response=model_response ) + + +def test_mistral_get_outputText_empty_outputs(): + model_response = ModelResponse() + + with pytest.raises(BedrockError): + AmazonMistralConfig.get_outputText( + completion_response={"outputs": []}, model_response=model_response + )