From d78764dda967d03af841e19e50861d59b370a938 Mon Sep 17 00:00:00 2001 From: Robin Huang Date: Mon, 5 May 2025 12:36:24 -0700 Subject: [PATCH] Fixed Kling: Check attributes of pydantic types. (#144) --- comfy_api_nodes/nodes_kling.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/comfy_api_nodes/nodes_kling.py b/comfy_api_nodes/nodes_kling.py index b36c1c544..1bd834fd3 100644 --- a/comfy_api_nodes/nodes_kling.py +++ b/comfy_api_nodes/nodes_kling.py @@ -151,8 +151,9 @@ def is_valid_task_creation_response(response: KlingText2VideoResponse) -> bool: def is_valid_video_response(response: KlingText2VideoResponse) -> bool: """Verifies that the response contains a task result with at least one video.""" return ( - "task_result" in response.data - and "videos" in response.data.task_result + response.data is not None + and response.data.task_result is not None + and response.data.task_result.videos is not None and len(response.data.task_result.videos) > 0 )