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