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

fix if-else condition when function_call exits but empty

Browse source
This commit is contained in:
kan-bayashi 2024-02-14 17:38:00 +09:00
parent 917525ee43
commit 90775910e2
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/vertex_ai.py
View file

@ -432,14 +432,13 @@ def completion(
tools=tools,
)
if tools is not None and hasattr(
response.candidates[0].content.parts[0], "function_call"
if tools is not None and bool(
getattr(response.candidates[0].content.parts[0], "function_call", None)
):
function_call = response.candidates[0].content.parts[0].function_call
args_dict = {}
if function_call.args is not None:
for k, v in function_call.args.items():
args_dict[k] = v
for k, v in function_call.args.items():
args_dict[k] = v
args_str = json.dumps(args_dict)
message = litellm.Message(
content=None,