From c6d7234c97e1fbe7a2e68f37510b3110216553ba Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 4 Mar 2024 21:46:00 -0800 Subject: [PATCH] fix(bedrock.py): fix conditional --- litellm/llms/bedrock.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/llms/bedrock.py b/litellm/llms/bedrock.py index 37084e62b36..79ba04f42ef 100644 --- a/litellm/llms/bedrock.py +++ b/litellm/llms/bedrock.py @@ -954,13 +954,14 @@ def completion( if ( len(outputText) > 0 and hasattr(model_response.choices[0], "message") - and model_response.choices[0].message.tool_calls is None + and getattr(model_response.choices[0].message, "tool_calls", None) + is None ): model_response["choices"][0]["message"]["content"] = outputText elif ( hasattr(model_response.choices[0], "message") - and hasattr(model_response.choices[0].message, "tool_calls") - and model_response.choices[0].message.tool_calls is not None + and getattr(model_response.choices[0].message, "tool_calls", None) + is not None ): pass else: