From 67e9395017c0a5c56772f8b23184c03d246617b4 Mon Sep 17 00:00:00 2001 From: Christian Byrne Date: Thu, 1 May 2025 17:40:05 -0700 Subject: [PATCH] Fix: `download_url_to_video_output` return type (#103) --- comfy_api_nodes/apinode_utils.py | 6 ++---- comfy_api_nodes/nodes_kling.py | 4 ++-- comfy_api_nodes/nodes_pika.py | 2 +- 3 files changed, 5 insertions(+), 7 deletions(-) diff --git a/comfy_api_nodes/apinode_utils.py b/comfy_api_nodes/apinode_utils.py index e5d0e29a9..3ff04c2a3 100644 --- a/comfy_api_nodes/apinode_utils.py +++ b/comfy_api_nodes/apinode_utils.py @@ -23,9 +23,7 @@ import uuid from io import BytesIO -def download_url_to_video_output( - video_url: str, timeout: int = None -) -> tuple[VideoFromFile]: +def download_url_to_video_output(video_url: str, timeout: int = None) -> VideoFromFile: """Downloads a video from a URL and returns a `VIDEO` output. Args: @@ -39,7 +37,7 @@ def download_url_to_video_output( error_msg = f"Failed to download video from {video_url}" logging.error(error_msg) raise ValueError(error_msg) - return (VideoFromFile(video_io),) + return VideoFromFile(video_io) def downscale_image_tensor(image, total_pixels=1536 * 1024) -> torch.Tensor: diff --git a/comfy_api_nodes/nodes_kling.py b/comfy_api_nodes/nodes_kling.py index ce29aa78b..ca64f8a8e 100644 --- a/comfy_api_nodes/nodes_kling.py +++ b/comfy_api_nodes/nodes_kling.py @@ -314,7 +314,7 @@ class KlingTextToVideoNode(KlingNodeBase): video_url = str(final_response.data.task_result.videos[0].url) logging.debug("Kling task %s succeeded. Video URL: %s", task_id, video_url) - return download_url_to_video_output(video_url) + return (download_url_to_video_output(video_url),) class KlingImage2VideoNode(KlingNodeBase): @@ -452,7 +452,7 @@ class KlingImage2VideoNode(KlingNodeBase): video_url = str(final_response.data.task_result.videos[0].url) logging.info("Attempting to download video from URL: %s", video_url) - return download_url_to_video_output(video_url) + return (download_url_to_video_output(video_url),) NODE_CLASS_MAPPINGS = { diff --git a/comfy_api_nodes/nodes_pika.py b/comfy_api_nodes/nodes_pika.py index ad44a6e60..825d570d7 100644 --- a/comfy_api_nodes/nodes_pika.py +++ b/comfy_api_nodes/nodes_pika.py @@ -158,7 +158,7 @@ class PikaNodeBase(ComfyNodeABC): video_url = str(final_response.url) logging.debug("Pika task %s succeeded. Video URL: %s", task_id, video_url) - return download_url_to_video_output(video_url) + return (download_url_to_video_output(video_url),) class PikaImageToVideoV2_2(PikaNodeBase):