fix args=None case

This commit is contained in:
kan-bayashi 2024-02-14 14:49:42 +09:00
parent a4e33c8c67
commit 917525ee43

View file

@ -437,8 +437,9 @@ def completion(
):
function_call = response.candidates[0].content.parts[0].function_call
args_dict = {}
for k, v in function_call.args.items():
args_dict[k] = v
if function_call.args is not None:
for k, v in function_call.args.items():
args_dict[k] = v
args_str = json.dumps(args_dict)
message = litellm.Message(
content=None,