diff --git a/tests/pass_through_unit_tests/test_unit_test_vertex_pass_through.py b/tests/pass_through_unit_tests/test_unit_test_vertex_pass_through.py deleted file mode 100644 index 066e434c262..00000000000 --- a/tests/pass_through_unit_tests/test_unit_test_vertex_pass_through.py +++ /dev/null @@ -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, -) diff --git a/tests/router_unit_tests/test_router_adding_deployments.py b/tests/router_unit_tests/test_router_adding_deployments.py index fca3f147e5b..aabaf37f1d8 100644 --- a/tests/router_unit_tests/test_router_adding_deployments.py +++ b/tests/router_unit_tests/test_router_adding_deployments.py @@ -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" )