diff --git a/litellm/llms/gemini/videos/transformation.py b/litellm/llms/gemini/videos/transformation.py index d1ae47af26..ce2519e917 100644 --- a/litellm/llms/gemini/videos/transformation.py +++ b/litellm/llms/gemini/videos/transformation.py @@ -15,17 +15,16 @@ from litellm.images.utils import ImageEditRequestUtils import litellm from litellm.types.llms.gemini import GeminiLongRunningOperationResponse, GeminiVideoGenerationInstance, GeminiVideoGenerationParameters, GeminiVideoGenerationRequest from litellm.constants import DEFAULT_GOOGLE_VIDEO_DURATION_SECONDS +from litellm.llms.base_llm.videos.transformation import BaseVideoConfig + if TYPE_CHECKING: from litellm.litellm_core_utils.litellm_logging import Logging as _LiteLLMLoggingObj - from ...base_llm.videos.transformation import BaseVideoConfig as _BaseVideoConfig from ...base_llm.chat.transformation import BaseLLMException as _BaseLLMException LiteLLMLoggingObj = _LiteLLMLoggingObj - BaseVideoConfig = _BaseVideoConfig BaseLLMException = _BaseLLMException else: LiteLLMLoggingObj = Any - BaseVideoConfig = Any BaseLLMException = Any diff --git a/tests/test_litellm/test_video_generation.py b/tests/test_litellm/test_video_generation.py index be7400d9b7..1a66cd29c7 100644 --- a/tests/test_litellm/test_video_generation.py +++ b/tests/test_litellm/test_video_generation.py @@ -14,6 +14,7 @@ import litellm from litellm.types.videos.main import VideoObject, VideoResponse from litellm.videos.main import video_generation, avideo_generation, video_status, avideo_status from litellm.llms.openai.videos.transformation import OpenAIVideoConfig +from litellm.llms.gemini.videos.transformation import GeminiVideoConfig from litellm.llms.custom_httpx.llm_http_handler import BaseLLMHTTPHandler from litellm.cost_calculator import default_video_cost_calculator from litellm.litellm_core_utils.litellm_logging import Logging as LitellmLogging @@ -813,5 +814,12 @@ def test_openai_video_config_has_async_transform(): cfg = OpenAIVideoConfig() assert callable(getattr(cfg, "async_transform_video_content_response", None)) + +def test_gemini_video_config_has_async_transform(): + """Ensure GeminiVideoConfig exposes async_transform_video_content_response at runtime.""" + cfg = GeminiVideoConfig() + assert callable(getattr(cfg, "async_transform_video_content_response", None)) + + if __name__ == "__main__": pytest.main([__file__])