From 0cab40b28ffc006bed07511af0ca32cd77bc27c3 Mon Sep 17 00:00:00 2001
From: Christian Byrne <cbyrne@comfy.org>
Date: Mon, 5 May 2025 22:33:21 -0700
Subject: [PATCH] [Bug] [Kling] Fix Kling camera control (#161)

---
 comfy_api_nodes/nodes_kling.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/comfy_api_nodes/nodes_kling.py b/comfy_api_nodes/nodes_kling.py
index e60c57b0e..df4e07fdd 100644
--- a/comfy_api_nodes/nodes_kling.py
+++ b/comfy_api_nodes/nodes_kling.py
@@ -412,10 +412,13 @@ class KlingTextToVideoNode(KlingNodeBase):
         mode: str,
         aspect_ratio: str,
         camera_control: Optional[KlingCameraControl] = None,
+        model_name: Optional[str] = None,
+        duration: Optional[str] = None,
         auth_token: Optional[str] = None,
     ) -> tuple[VideoFromFile, str, str]:
         validate_prompts(prompt, negative_prompt, MAX_PROMPT_LENGTH_T2V)
-        mode, duration, model_name = self.get_mode_string_mapping()[mode]
+        if model_name is None:
+            mode, duration, model_name = self.get_mode_string_mapping()[mode]
         initial_operation = SynchronousOperation(
             endpoint=ApiEndpoint(
                 path=PATH_TEXT_TO_VIDEO,
@@ -502,9 +505,9 @@ class KlingCameraControlT2VNode(KlingTextToVideoNode):
         auth_token: Optional[str] = None,
     ):
         return super().api_call(
-            model_name=KlingVideoGenModelName.kling_v1_5,
+            model_name=KlingVideoGenModelName.kling_v1,
             cfg_scale=cfg_scale,
-            mode=KlingVideoGenMode.pro,
+            mode=KlingVideoGenMode.std,
             aspect_ratio=KlingVideoGenAspectRatio(aspect_ratio),
             duration=KlingVideoGenDuration.field_5,
             prompt=prompt,