remove kling support

video-db · Jan 3, 2025 · ce83b42 · ce83b42
1 parent dbba8c6
commit ce83b42
Showing 1 changed file with 1 addition and 17 deletions.
diff --git a/backend/director/agents/video_generation.py b/backend/director/agents/video_generation.py
@@ -12,7 +12,6 @@
     StabilityAITool,
     PARAMS_CONFIG as STABILITYAI_PARAMS_CONFIG,
 )
-from director.tools.kling import KlingAITool, PARAMS_CONFIG as KLING_PARAMS_CONFIG
 from director.tools.fal_video import (
     FalVideoGenerationTool,
     PARAMS_CONFIG as FAL_VIDEO_GEN_PARAMS_CONFIG,
@@ -21,7 +20,7 @@
 
 logger = logging.getLogger(__name__)
 
-SUPPORTED_ENGINES = ["stabilityai", "kling", "fal"]
+SUPPORTED_ENGINES = ["stabilityai", "fal"]
 
 VIDEO_GENERATION_AGENT_PARAMETERS = {
     "type": "object",
@@ -66,11 +65,6 @@
                     "properties": STABILITYAI_PARAMS_CONFIG["text_to_video"],
                     "description": "Config to use when stabilityai engine is used",
                 },
-                "kling_config": {
-                    "type": "object",
-                    "properties": KLING_PARAMS_CONFIG["text_to_video"],
-                    "description": "Config to use when kling engine is used",
-                },
                 "fal_config": {
                     "type": "object",
                     "properties": FAL_VIDEO_GEN_PARAMS_CONFIG["text_to_video"],
@@ -131,16 +125,6 @@ async def run_async(
                     raise Exception("Stability AI API key not found")
                 video_gen_tool = StabilityAITool(api_key=STABILITYAI_API_KEY)
                 config_key = "stabilityai_config"
-            elif engine == "kling":
-                KLING_AI_ACCESS_API_KEY = os.getenv("KLING_AI_ACCESS_API_KEY")
-                KLING_AI_SECRET_API_KEY = os.getenv("KLING_AI_SECRET_API_KEY")
-                if not KLING_AI_ACCESS_API_KEY or not KLING_AI_SECRET_API_KEY:
-                    raise Exception("Kling AI API key not found")
-                video_gen_tool = KlingAITool(
-                    access_key=KLING_AI_ACCESS_API_KEY,
-                    secret_key=KLING_AI_SECRET_API_KEY,
-                )
-                config_key = "kling_config"
             elif engine == "fal":
                 FAL_KEY = os.getenv("FAL_KEY")
                 if not FAL_KEY: