fix: linting fixes

2024-07-03 21:55:00 -07:00 · 2024-07-03 21:55:00 -07:00 · 19c982d0f9
commit 19c982d0f9
parent 17869fc5e9
5 changed files with 62 additions and 16 deletions
--- a/litellm/main.py
+++ b/litellm/main.py
@ -2008,6 +2008,8 @@ def completion(
                    vertex_credentials=vertex_credentials,
                    logging_obj=logging,
                    acompletion=acompletion,
+                    headers=headers,
+                    custom_prompt_dict=custom_prompt_dict,
                )
            else:
                model_response = vertex_ai.completion(