Merge pull request #1 from Andrewmatilde/main

Make execute() async function

.github/workflows/main.yml

@@ -0,0 +1,20 @@
+name: Publish to Comfy registry
+on:
+  workflow_dispatch:
+  push:
+    branches:
+      - main
+    paths:
+      - "pyproject.toml"
+
+jobs:
+  publish-node:
+    name: Publish Custom Node to registry
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out code
+        uses: actions/checkout@v4
+      - name: Publish Custom Node
+        uses: Comfy-Org/publish-node-action@main
+        with:
+          personal_access_token: ${{ secrets.REGISTRY_ACCESS_TOKEN }} ## Add your own personal access token to your Github Repository secrets and reference it here.

py/flux_dev.py

@@ -97,7 +97,7 @@ class FluxDevNode:
     CATEGORY = "UCLOUD_MODELVERSE"
     FUNCTION = "execute"
 
-    def execute(self,
+    async def execute(self,
                 client,
                 prompt,
                 width=864,
@@ -129,7 +129,7 @@ class FluxDevNode:
         ))
             for i in range(num_requests)]
 
-        image_urls = asyncio.run(client.run_tasks(tasks))
+        image_urls = await client.run_tasks(tasks)
 
         output_images_list = []
         for image_url in image_urls:

py/flux_kontext_max.py

@@ -59,7 +59,7 @@ class FluxKontextMaxNode:
     CATEGORY = "UCLOUD_MODELVERSE"
     FUNCTION = "execute"
 
-    def execute(self,
+    async def execute(self,
                 client,
                 prompt,
                 images,
@@ -100,7 +100,7 @@ class FluxKontextMaxNode:
             ))
                 for i in range(num_requests)]
 
-        image_urls = asyncio.run(client.run_tasks(tasks))
+        image_urls = await client.run_tasks(tasks)
 
         output_images_list = []
         for image_url in image_urls:

py/flux_kontext_max_t2i.py

@@ -63,7 +63,7 @@ class FluxKontextMaxT2INode:
     CATEGORY = "UCLOUD_MODELVERSE"
     FUNCTION = "execute"
 
-    def execute(self,
+    async def execute(self,
                 client,
                 prompt,
                 num_images=1,
@@ -88,7 +88,7 @@ class FluxKontextMaxT2INode:
         ))
             for i in range(num_requests)]
 
-        image_urls = asyncio.run(client.run_tasks(tasks))
+        image_urls = await client.run_tasks(tasks)
 
         output_images_list = []
         for image_url in image_urls:

py/flux_kontext_pro.py

@@ -59,7 +59,7 @@ class FluxKontextProNode:
     CATEGORY = "UCLOUD_MODELVERSE"
     FUNCTION = "execute"
 
-    def execute(self,
+    async def execute(self,
                 client,
                 prompt,
                 images,
@@ -100,7 +100,7 @@ class FluxKontextProNode:
             ))
                 for i in range(num_requests)]
 
-        image_urls = asyncio.run(client.run_tasks(tasks))
+        image_urls = await client.run_tasks(tasks)
 
         output_images_list = []
         for image_url in image_urls:

py/flux_kontext_pro_t2i.py

@@ -63,7 +63,7 @@ class FluxKontextProT2INode:
     CATEGORY = "UCLOUD_MODELVERSE"
     FUNCTION = "execute"
 
-    def execute(self,
+    async def execute(self,
                 client,
                 prompt,
                 num_images=1,
@@ -88,7 +88,7 @@ class FluxKontextProT2INode:
         ))
             for i in range(num_requests)]
 
-        image_urls = asyncio.run(client.run_tasks(tasks))
+        image_urls = await client.run_tasks(tasks)
 
         output_images_list = []
         for image_url in image_urls:

py/step1x_edit.py

@@ -60,7 +60,7 @@ class Step1xEditNode:
     CATEGORY = "UCLOUD_MODELVERSE"
     FUNCTION = "execute"
 
-    def execute(self,
+    async def execute(self,
                 client,
                 prompt,
                 image,
@@ -89,7 +89,7 @@ class Step1xEditNode:
         ))
             for i in range(num_requests)]
 
-        image_urls = asyncio.run(client.run_tasks(tasks))
+        image_urls = await client.run_tasks(tasks)
 
         output_images_list = []
         for image_url in image_urls:

pyproject.toml

@@ -1,15 +1,15 @@
 [project]
 name = "ucloud-comfyui"
 description = "This is a set of ComfyUI nodes for UCloud Moderlverse Image generation model API."
-version = "1.0.0"
+version = "1.0.2"
 license = {file = "LICENSE"}
 dependencies = ["requests", "pillow"]
 
 [project.urls]
-Repository = "https://github.com/ucloud/ucloud-comfyui"
+Repository = "https://github.com/ucloud/comfyui-plugin"
 
 [tool.comfy]
-PublisherId = "ucloud-modelverse"
-DisplayName = "ucloud-comfyui"
-Icon = ""
+PublisherId = "khalil-song"
+DisplayName = "ComfyUl-UCloud"
+Icon = "https://www-s.ucloud.cn/2025/07/4a29a785049191245cea58d2176157d2_1753185180124.jpg"
 includes = []

requirements.txt

@@ -1,2 +1,2 @@
-requests >= 3.32.4
+requests >= 2.32.4
 pillow >= 11