phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +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, tools=tools,
) )
if tools is not None and hasattr( if tools is not None and bool(
response.candidates[0].content.parts[0], "function_call" getattr(response.candidates[0].content.parts[0], "function_call", None)
): ):
function_call = response.candidates[0].content.parts[0].function_call function_call = response.candidates[0].content.parts[0].function_call
args_dict = {} args_dict = {}
if function_call.args is not None: for k, v in function_call.args.items():
for k, v in function_call.args.items(): args_dict[k] = v
args_dict[k] = v
args_str = json.dumps(args_dict) args_str = json.dumps(args_dict)
message = litellm.Message( message = litellm.Message(
content=None, content=None,