From d8d215dd3af99e6ff7b807ec8a14c21370af11a9 Mon Sep 17 00:00:00 2001 From: Christian Byrne Date: Wed, 30 Apr 2025 10:39:46 -0700 Subject: [PATCH] Use common download function in kling nodes module (#67) --- comfy_api_nodes/nodes_kling.py | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/comfy_api_nodes/nodes_kling.py b/comfy_api_nodes/nodes_kling.py index 7d0f2f48a..035c4f382 100644 --- a/comfy_api_nodes/nodes_kling.py +++ b/comfy_api_nodes/nodes_kling.py @@ -27,7 +27,7 @@ from comfy_api_nodes.apis.client import ( ) from comfy_api_nodes.apinode_utils import ( tensor_to_base64_string, - download_url_to_bytesio, + download_url_to_video_output, ) from comfy.comfy_types.node_typing import IO, InputTypeOptions, ComfyNodeABC from comfy_api.input_impl import VideoFromFile @@ -128,16 +128,6 @@ def _get_camera_control_inputs() -> dict[str, tuple[IO, InputTypeOptions]]: } -def download_url_to_video_output(video_url: str) -> tuple[VideoFromFile]: - """Downloads a video from a URL and returns a VIDEO output.""" - video_io = download_url_to_bytesio(video_url) - if video_io is None: - error_msg = f"Failed to download video from {video_url}" - logging.error(error_msg) - raise KlingApiError(error_msg) - return (VideoFromFile(video_io),) - - class KlingNodeABC(ComfyNodeABC): """Base class for Kling nodes."""