From ad1d86e8e691ad124bc45f30ba3a0f20975f4167 Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaan@berri.ai>
Date: Tue, 21 Apr 2026 08:35:16 -0700
Subject: [PATCH] fix(vertex passthrough): add embedContent and
 batchEmbedContents to TRACKED_VERTEX_ROUTES

---
 .../pass_through_endpoints/success_handler.py | 20 ++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/litellm/proxy/pass_through_endpoints/success_handler.py b/litellm/proxy/pass_through_endpoints/success_handler.py
index 33819b888d0..0bc0183aa7c 100644
--- a/litellm/proxy/pass_through_endpoints/success_handler.py
+++ b/litellm/proxy/pass_through_endpoints/success_handler.py
@@ -46,6 +46,8 @@ class PassThroughEndpointLogging:
             "search",
             "batchPredictionJobs",
             "predictLongRunning",
+            "embedContent",
+            "batchEmbedContents",
         ]
 
         # Anthropic
@@ -283,9 +285,9 @@ class PassThroughEndpointLogging:
 
             standard_logging_response_object = vertex_ai_live_handler_result["result"]
             kwargs = vertex_ai_live_handler_result["kwargs"]
-        return_dict[
-            "standard_logging_response_object"
-        ] = standard_logging_response_object
+        return_dict["standard_logging_response_object"] = (
+            standard_logging_response_object
+        )
 
         return_dict["kwargs"] = kwargs
         return return_dict
@@ -308,9 +310,9 @@ class PassThroughEndpointLogging:
         standard_logging_response_object: Optional[
             PassThroughEndpointLoggingResultValues
         ] = None
-        logging_obj.model_call_details[
-            "passthrough_logging_payload"
-        ] = passthrough_logging_payload
+        logging_obj.model_call_details["passthrough_logging_payload"] = (
+            passthrough_logging_payload
+        )
         if self.is_assemblyai_route(url_route):
             if (
                 AssemblyAIPassthroughLoggingHandler._should_log_request(
@@ -487,8 +489,8 @@ class PassThroughEndpointLogging:
             kwargs["response_cost"] = passthrough_logging_payload.get(
                 "cost_per_request"
             )
-            logging_obj.model_call_details[
-                "response_cost"
-            ] = passthrough_logging_payload.get("cost_per_request")
+            logging_obj.model_call_details["response_cost"] = (
+                passthrough_logging_payload.get("cost_per_request")
+            )
 
         return kwargs