fix args=None case

2024-02-14 14:49:42 +09:00 · 2024-02-14 14:49:42 +09:00 · 917525ee43
commit 917525ee43
parent a4e33c8c67
1 changed files with 3 additions and 2 deletions
--- a/litellm/llms/vertex_ai.py
+++ b/litellm/llms/vertex_ai.py
@ -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,