From 0578040c5562991d1f3ccc46a451841b888e6ab5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 20 May 2025 11:52:10 -0700 Subject: [PATCH] fix: vertex getting creds for pass through --- .../passthrough_endpoint_router.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/litellm/proxy/pass_through_endpoints/passthrough_endpoint_router.py b/litellm/proxy/pass_through_endpoints/passthrough_endpoint_router.py index a02cacc3cc0..520bda177b9 100644 --- a/litellm/proxy/pass_through_endpoints/passthrough_endpoint_router.py +++ b/litellm/proxy/pass_through_endpoints/passthrough_endpoint_router.py @@ -130,6 +130,9 @@ class PassthroughEndpointRouter: vertex_location=location, vertex_credentials=vertex_credentials, ) + verbose_router_logger.debug( + f"Adding vertex credentials for deployment_key: {deployment_key}, vertex_pass_through_credentials: {vertex_pass_through_credentials}" + ) self.deployment_key_to_vertex_credentials[ deployment_key ] = vertex_pass_through_credentials @@ -154,12 +157,24 @@ class PassthroughEndpointRouter: project_id=project_id, location=location, ) + verbose_router_logger.debug( + f"Getting vertex credentials for deployment_key: {deployment_key}" + ) if deployment_key is None: + verbose_router_logger.debug( + "No deployment key found for project-id, location" + ) return self.default_vertex_config if deployment_key in self.deployment_key_to_vertex_credentials: + verbose_router_logger.debug( + f"Found vertex credentials for deployment_key: {deployment_key}" + ) return self.deployment_key_to_vertex_credentials[deployment_key] else: + verbose_router_logger.debug( + "No vertex credentials found for deployment_key, returning default vertex config" + ) return self.default_vertex_config def _get_credential_name_for_provider(