From 0d6d94c43c83632bd6bac29afdade5ab53f7c774 Mon Sep 17 00:00:00 2001 From: StatPan Date: Fri, 1 May 2026 21:29:49 +0900 Subject: [PATCH] test(gemini): cover full-route custom api_base handling --- tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py b/tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py index 88212ded130..818a82a8a18 100644 --- a/tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py +++ b/tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py @@ -1,17 +1,14 @@ import json import os import sys -from unittest.mock import MagicMock, call, patch +from unittest.mock import MagicMock, patch import pytest -from litellm.constants import DEFAULT_MAX_RECURSE_DEPTH - sys.path.insert( 0, os.path.abspath("../../..") ) # Adds the parent directory to the system path -import litellm from litellm.llms.vertex_ai.vertex_ai_aws_wif import VertexAIAwsWifAuth from litellm.llms.vertex_ai.vertex_llm_base import VertexBase @@ -71,7 +68,7 @@ class TestVertexBase: project_id="different-project", custom_llm_provider="vertex_ai", ) - print(f"result: {result}") + assert result == ("fake-token-1", "project-1") @pytest.mark.parametrize("is_async", [True, False], ids=["async", "sync"]) @pytest.mark.asyncio