From 26aa20df73d653784b13f8b9a81f7e61e16d9bd1 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 10 Sep 2024 17:38:03 -0700 Subject: [PATCH] fix test get token url --- litellm/tests/test_amazing_vertex_completion.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 9b9eee211..4c065b8d3 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -2183,6 +2183,7 @@ def test_get_token_url(): assert should_use_v1beta1_features is True _, url = vertex_llm._get_token_and_url( + auth_header=None, vertex_project=vertex_ai_project, vertex_location=vertex_ai_location, vertex_credentials=vertex_credentials, @@ -2203,6 +2204,7 @@ def test_get_token_url(): ) _, url = vertex_llm._get_token_and_url( + auth_header=None, vertex_project=vertex_ai_project, vertex_location=vertex_ai_location, vertex_credentials=vertex_credentials,