From f62437f13cfd88454a014a79e5e7218c69e43243 Mon Sep 17 00:00:00 2001 From: shrey kharbanda Date: Thu, 24 Sep 2026 00:23:13 +0000 Subject: [PATCH] revert(videos): keep direct catalog read for resolution inference --- litellm/llms/vertex_ai/videos/transformation.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/llms/vertex_ai/videos/transformation.py b/litellm/llms/vertex_ai/videos/transformation.py index 4ee67cb2618..6c29059f0a9 100644 --- a/litellm/llms/vertex_ai/videos/transformation.py +++ b/litellm/llms/vertex_ai/videos/transformation.py @@ -257,9 +257,9 @@ class VertexAIVideoConfig(BaseVideoConfig, VertexBase): @staticmethod def _supports_resolution_inference(model: str) -> bool: - model_name: Final = model.removeprefix("vertex_ai/") - model_info: Final = litellm.get_model_info(model=model_name, custom_llm_provider="vertex_ai") - return model_info.get("output_cost_per_second_1080p") is not None + model_key: Final = model if model.startswith("vertex_ai/") else f"vertex_ai/{model}" + model_info: Final = litellm.model_cost.get(model_key) + return model_info is not None and model_info.get("output_cost_per_second_1080p") is not None def validate_environment( self,