phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(bedrock.py): fix conditional

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-04 21:46:00 -08:00
parent b83c1bf09e
commit ee5bedf541
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/bedrock.py
View file

@ -954,13 +954,14 @@ def completion(
if ( if (
len(outputText) > 0 len(outputText) > 0
and hasattr(model_response.choices[0], "message") 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 model_response["choices"][0]["message"]["content"] = outputText
elif ( elif (
hasattr(model_response.choices[0], "message") hasattr(model_response.choices[0], "message")
and hasattr(model_response.choices[0].message, "tool_calls") and getattr(model_response.choices[0].message, "tool_calls", None)
and model_response.choices[0].message.tool_calls is not None is not None
): ):
pass pass
else: else: