scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-22 12:50:58 -07:00
parent 3ce3689282
commit 6f719d0461
2 changed files with 4 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
tests/pass_through_unit_tests/test_unit_test_vertex_pass_through.py
View file

@ -1,28 +0,0 @@
import json
import os
import sys
from datetime import datetime
from unittest.mock import AsyncMock, Mock, patch
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system-path
import httpx
import pytest
import litellm
from litellm.litellm_core_utils.litellm_logging import Logging as LiteLLMLoggingObj
from litellm.proxy.vertex_ai_endpoints.vertex_endpoints import (
get_litellm_virtual_key,
vertex_proxy_route,
_get_vertex_env_vars,
set_default_vertex_config,
VertexPassThroughCredentials,
default_vertex_config,
)
from litellm.proxy.vertex_ai_endpoints.vertex_passthrough_router import (
VertexPassThroughRouter,
)

8
tests/router_unit_tests/test_router_adding_deployments.py
View file

@ -123,21 +123,21 @@ def test_add_vertex_pass_through_deployment():
router.add_deployment(deployment)
# Get the vertex credentials from the router
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,
)
# current state of pass-through vertex router
print("\n vertex_pass_through_router.deployment_key_to_vertex_credentials\n\n")
print(
json.dumps(
vertex_pass_through_router.deployment_key_to_vertex_credentials,
passthrough_endpoint_router.deployment_key_to_vertex_credentials,
indent=4,
default=str,
)
)
vertex_creds = vertex_pass_through_router.get_vertex_credentials(
vertex_creds = passthrough_endpoint_router.get_vertex_credentials(
project_id="test-project", location="us-central1"
)