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 4a2e45ca2e7..8dc9e16eb85 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 @@ -130,13 +130,6 @@ class VertexPassthroughLoggingHandler: "kwargs": kwargs, } - elif "predict" in url_route: - return VertexPassthroughLoggingHandler._handle_predict_response( - httpx_response=httpx_response, - logging_obj=logging_obj, - url_route=url_route, - kwargs=kwargs, - ) elif "rawPredict" in url_route or "streamRawPredict" in url_route: from litellm.llms.vertex_ai.vertex_ai_partner_models import ( get_vertex_ai_partner_model_config, @@ -191,6 +184,13 @@ class VertexPassthroughLoggingHandler: "result": litellm_prediction_response, "kwargs": kwargs, } + elif "predict" in url_route: + return VertexPassthroughLoggingHandler._handle_predict_response( + httpx_response=httpx_response, + logging_obj=logging_obj, + url_route=url_route, + kwargs=kwargs, + ) elif "search" in url_route: litellm_vs_response = (