phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: update tests

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-22 12:56:42 -07:00
parent fe004bc45c
commit 6b2f385ddf
3 changed files with 4 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/router_unit_tests/test_router_adding_deployments.py
View file

@ -36,11 +36,11 @@ def test_initialize_deployment_for_pass_through_success():
)
# Verify the credentials were properly set
from litellm.proxy.vertex_ai_endpoints.vertex_endpoints import (
vertex_pass_through_router,
from litellm.proxy.pass_through_endpoints.llm_passthrough_endpoints import (
passthrough_endpoint_router,
)
vertex_creds = vertex_pass_through_router.get_vertex_credentials(
vertex_creds = passthrough_endpoint_router.get_vertex_credentials(
project_id="test-project", location="us-central1"
)
assert vertex_creds.vertex_project == "test-project"