scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(gemini): cover full-route custom api_base handling

Browse source
This commit is contained in:
StatPan 2026-05-01 21:29:49 +09:00
parent d8418fff08
commit 0d6d94c43c
1 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py
View file

@ -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