From effebdab3876c09804f6ac2c1603349f092be9d8 Mon Sep 17 00:00:00 2001
From: Sameer Kankute <sameer@berri.ai>
Date: Wed, 26 Nov 2025 15:29:05 +0530
Subject: [PATCH] Add custom llm provider in vertex ai embeddings

---
 .../vertex_passthrough_logging_handler.py                      | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/litellm/proxy/pass_through_endpoints/llm_provider_handlers/vertex_passthrough_logging_handler.py b/litellm/proxy/pass_through_endpoints/llm_provider_handlers/vertex_passthrough_logging_handler.py
index dae2a7f081..407cbd2f3d 100644
--- a/litellm/proxy/pass_through_endpoints/llm_provider_handlers/vertex_passthrough_logging_handler.py
+++ b/litellm/proxy/pass_through_endpoints/llm_provider_handlers/vertex_passthrough_logging_handler.py
@@ -148,6 +148,8 @@ class VertexPassthroughLoggingHandler:
 
             logging_obj.model = model
             logging_obj.model_call_details["model"] = logging_obj.model
+            logging_obj.model_call_details["custom_llm_provider"] = "vertex_ai"
+            logging_obj.custom_llm_provider = "vertex_ai"
             response_cost = litellm.completion_cost(
                 completion_response=litellm_prediction_response,
                 model=model,
@@ -156,6 +158,7 @@ class VertexPassthroughLoggingHandler:
 
             kwargs["response_cost"] = response_cost
             kwargs["model"] = model
+            kwargs["custom_llm_provider"] = "vertex_ai"
             logging_obj.model_call_details["response_cost"] = response_cost
 
             return {