Skip to main content

RunningHub ComfyUI SDK - 轻松调用RunningHub的ComfyUI API

Project description

RunningHub ComfyUI SDK

runninghub-sdk 是一个面向 RunningHub ComfyUI OpenAPI 的 Python SDK,支持任务创建、状态轮询、结果查询、文件上传,以及用链式 NodeModifier 修改工作流节点参数。

当前版本也支持 RunningHub AI App 接口,包括获取 AI App 可调用节点示例和直接发起 AI App 任务。

这个子目录可以直接作为 Python 包发布根目录使用,也可以单独进入目录后执行构建和上传命令发布到 PyPI。

特性

  • 同时支持同步和异步调用
  • 基于 httpx,接口简单,依赖精简
  • 提供完整类型注解,适合 IDE 自动补全
  • 支持 NodeModifier 链式修改节点输入
  • 支持图片、视频、音频、LoRA 等文件上传
  • 支持自动轮询等待任务完成

安装

pip install runninghub-sdk

如果你是在本仓库中本地开发,也可以进入 runninghub_sdk 目录后安装开发依赖:

pip install -e .[dev]

快速开始

同步调用

from runninghub_sdk import RunningHubClient, modify_nodes

with RunningHubClient(api_key="your-api-key") as client:
    modifier = (
        modify_nodes()
        .text("6", "a cinematic sunset over the sea")
        .seed("3", 12345)
        .steps("3", 25)
    )

    task = client.run_with_modifier("workflow-id", modifier)
    outputs = client.wait_for_completion(task.task_id)

    for output in outputs:
        print(output.file_url)

异步调用

import asyncio

from runninghub_sdk import RunningHubClient, modify_nodes


async def main() -> None:
    async with RunningHubClient(api_key="your-api-key") as client:
        modifier = modify_nodes().text("6", "a beautiful landscape")
        task = await client.async_run_with_modifier("workflow-id", modifier)
        outputs = await client.async_wait_for_completion(task.task_id)

        for output in outputs:
            print(output.file_url)


asyncio.run(main())

核心能力

AI App 接口

同步方法 异步方法 说明
get_ai_app_api_demo() async_get_ai_app_api_demo() 获取 AI App 调用示例、节点信息、封面和标签
run_ai_app() async_run_ai_app() 发起 AI App 任务
run_ai_app_with_modifier() async_run_ai_app_with_modifier() 使用修改器发起 AI App 任务
list_public_models() async_list_public_models() 获取公共模型列表,支持类型、名称、基础模型、标签分页筛选
run_model_api() async_run_model_api() 通用标准模型 API 调用,适用于图像、视频、音频、3D 等标准模型端点
preview_model_price() async_preview_model_price() 按实际请求参数预估标准模型调用价格
wait_for_query_v2_completion() async_wait_for_query_v2_completion() 基于 /openapi/v2/query 轮询标准模型任务完成

账户与调试接口

同步方法 异步方法 说明
get_account_status() async_get_account_status() 获取账户信息,包括余额、当前任务数、API 类型
list_api_keys() async_list_api_keys() 查询 API Key 列表
get_queue_status() async_get_queue_status() 查询当前 API Key 的队列状态
validate_api_key() async_validate_api_key() 通过队列状态接口验证当前 API Key 是否有效
get_webhook_detail() async_get_webhook_detail() 根据任务 ID 查询 webhook 事件详情
retry_webhook() async_retry_webhook() 重新发送指定 webhook 事件

任务接口

同步方法 异步方法 说明
run() async_run() 发起任务
run_with_modifier() async_run_with_modifier() 使用修改器发起任务
get_status() async_get_status() 查询任务状态
get_outputs() async_get_outputs() 查询任务输出
wait_for_completion() async_wait_for_completion() 轮询直到任务完成
query_v2() async_query_v2() 调用 V2 查询接口

文件上传

同步方法 异步方法 说明
upload_file() async_upload_file() 上传通用文件
upload_image() async_upload_image() 上传图片
upload_lora() async_upload_lora() 上传 LoRA

工作流读取

同步方法 异步方法 说明
get_workflow_json() async_get_workflow_json() 获取工作流 JSON 字符串
get_workflow_json_parsed() async_get_workflow_json_parsed() 获取解析后的工作流对象

NodeModifier

NodeModifier 用于用链式 API 构造 node_info_list,让工作流参数修改更直观。

from runninghub_sdk import modify_nodes

modifier = (
    modify_nodes()
    .text("6", "a portrait in film style")
    .negative_text("7", "blurry, low quality")
    .seed("3", 12345)
    .steps("3", 25)
    .cfg("3", 7.5)
    .size("5", 1024, 768)
    .sampler("3", "dpmpp_2m")
    .scheduler("3", "karras")
    .image("10", "uploaded-file.png")
)

常用方法如下:

方法 说明
set(node_id, field_name, value) 通用设置
text(node_id, text) 设置提示词
negative_text(node_id, text) 设置负面提示词
seed(node_id, seed) 设置随机种子
steps(node_id, steps) 设置采样步数
cfg(node_id, cfg) 设置 CFG
size(node_id, width, height) 设置图像尺寸
sampler(node_id, name) 设置采样器
scheduler(node_id, name) 设置调度器
denoise(node_id, value) 设置去噪强度
image(node_id, file_name) 设置图片文件
video(node_id, file_name) 设置视频文件
audio(node_id, file_name) 设置音频文件
lora(node_id, file_name) 设置 LoRA 文件
checkpoint(node_id, name) 设置模型名

AI App 使用

AI App 接口适合直接调用 RunningHub 页面上的应用。webappId 可以从 AI App 详情页链接中获取,例如 https://www.runninghub.cn/ai-detail/1937084629516193794 最后的数字就是 webappId

推荐先读取 AI App 的可调用示例,再按节点修改参数并运行:

from runninghub_sdk import RunningHubClient, modify_nodes

with RunningHubClient(api_key="your-api-key") as client:
    demo = client.get_ai_app_api_demo("1937084629516193794")

    for node in demo.node_info_list:
        print(node.node_id, node.field_name, node.field_type, node.description)

    modifier = (
        modify_nodes()
        .set("52", "prompt", "把人物发型改成齐耳短发")
        .set("37", "aspect_ratio", "1:1")
    )

    task = client.run_ai_app_with_modifier(
        "1937084629516193794",
        modifier,
    )
    outputs = client.wait_for_completion(task.task_id)

    for output in outputs:
        print(output.file_url)

如果 AI App 包含 IMAGEAUDIOVIDEO 一类输入,通常先上传文件,再把返回的 fileName 设置回对应节点的 fieldValue

from runninghub_sdk import RunningHubClient, modify_nodes

with RunningHubClient(api_key="your-api-key") as client:
    uploaded = client.upload_image("input.png")

    modifier = (
        modify_nodes()
        .set("39", "image", uploaded["fileName"])
        .set("52", "prompt", "保留人物姿态,改成胶片质感")
    )

    task = client.run_ai_app_with_modifier("1937084629516193794", modifier)
    outputs = client.wait_for_completion(task.task_id)

    for output in outputs:
        print(output.file_url)

AI App 开启加密访问时,可以在运行时传入 access_password

task = client.run_ai_app(
    webapp_id="1937084629516193794",
    node_info_list=[
        {"nodeId": "52", "fieldName": "prompt", "fieldValue": "一张电影感人像"}
    ],
    access_password="your-password",
)

获取公共模型列表

可以通过公共模型列表接口拉取 RunningHub 提供的可用模型,并按类型、名称、基础模型、标签做筛选。

from runninghub_sdk import RunningHubClient

with RunningHubClient(api_key="your-api-key") as client:
    models = client.list_public_models(
        resource_type="UNET",
        resource_name="realDream",
        base_models=["Flux2-Klein-9B"],
        current=1,
        size=10,
    )

    print(models.total)
    for record in models.records:
        print(record.resource_name, record.resource_type)
        if record.versions:
            print(record.versions[0].version_resource_name)

resource_type 当前支持文档中的 UNETCHECKPOINTLORAGGUF

标准模型 API 使用

标准模型 API 的端点很多,不适合为每个模型单独维护一套方法。SDK 提供了通用调用入口 run_model_api(),你只需要传模型端点和对应请求体即可。

例如调用 f-2-dev/text-to-image

from runninghub_sdk import RunningHubClient

with RunningHubClient(api_key="your-api-key") as client:
    task = client.run_model_api(
        "/openapi/v2/rhart-image/f-2-dev/text-to-image",
        {
            "12##text": "在一片非洲大草原上,一只真实非洲狮的摄影照片",
            "41##select": "9:16",
            "30##value": 1024,
            "29##value": 1024,
            "43##file_type": "png",
        },
    )

    result = client.wait_for_query_v2_completion(task.task_id)
    print(result.results)

也可以只传相对路径,SDK 会自动补成 /openapi/v2/...

task = client.run_model_api(
    "rhart-audio/text-to-audio/speech-2.8-hd",
    {
        "text": "Bonjour! How are you today?",
        "voice_id": "Wise_Woman",
        "enable_base64_output": False,
        "english_normalization": False,
    },
)

调用前可以先做价格预估。把原始模型路径换给 preview_model_price() 即可,SDK 会自动转换成 /openapi/v2/price-preview/...

price = client.preview_model_price(
    "rhart-image/f-2-dev/text-to-image",
    {
        "12##text": "一张电影感狮子海报",
        "41##select": "9:16",
        "43##file_type": "png",
    },
)

print(price.estimated_price, price.currency)

账户、队列与 webhook 调试

from runninghub_sdk import RunningHubClient

with RunningHubClient(api_key="your-api-key") as client:
    if not client.validate_api_key():
        print("API Key 无效")
        raise SystemExit(1)

    account = client.get_account_status()
    print(account.remain_coins, account.current_task_counts, account.api_type)

    keys = client.list_api_keys()
    for key in keys:
        print(key.key, key.status, key.visible)

    queue = client.get_queue_status()
    print(queue.api_key_type, queue.running_count, queue.queued_count)

Webhook 调试示例:

from runninghub_sdk import RunningHubClient

with RunningHubClient(api_key="your-api-key") as client:
    detail = client.get_webhook_detail("1904154698679771137")
    print(detail.id, detail.callback_status, detail.retry_count)

    client.retry_webhook(detail.id, detail.webhook_url)

错误处理

from runninghub_sdk import ErrorCode, RunningHubError, TaskError, TimeoutError

try:
    outputs = client.wait_for_completion(task.task_id)
except TimeoutError as error:
    print(f"任务超时: {error.task_id}")
except TaskError as error:
    print(f"任务失败: {error.failed_reason}")
except RunningHubError as error:
    if error.code == ErrorCode.API_KEY_INVALID:
        print("API Key 无效")

类型定义

SDK 暴露了常用类型,便于静态检查和 IDE 补全:

from runninghub_sdk import (
    CreateTaskResponse,
    NodeInput,
    TaskOutput,
    TaskStatus,
    UploadResponseData,
)

更多示例

更多可运行示例见 EXAMPLES.md

详细使用案例

下面给出几类更贴近实际业务的调用方式。完整脚本可继续参考 EXAMPLES.md

案例 1:先读取工作流结构,再按节点动态改参

适合你拿到一个现成 workflow,但还不确定提示词节点、采样节点、尺寸节点编号时使用。

from runninghub_sdk import RunningHubClient, modify_nodes

with RunningHubClient(api_key="your-api-key") as client:
    workflow = client.get_workflow_json_parsed("your-workflow-id")

    print("可用节点:")
    for node_id, node_data in workflow.items():
        class_type = node_data.get("class_type", "unknown")
        inputs = list(node_data.get("inputs", {}).keys())
        print(node_id, class_type, inputs)

    modifier = (
        modify_nodes()
        .text("6", "a cinematic portrait, 85mm lens, natural light")
        .negative_text("7", "low quality, blurry, deformed")
        .seed("3", 20260510)
        .steps("3", 30)
        .cfg("3", 7.0)
        .size("5", 1024, 1536)
    )

    task = client.run_with_modifier("your-workflow-id", modifier)
    outputs = client.wait_for_completion(task.task_id)

    for output in outputs:
        print(output.file_url)

案例 2:AI App 场景下先上传素材,再运行应用

适合图生图、音频驱动、视频输入这类 AI App。流程一般是:获取节点示例、上传文件、把上传结果回填到节点、发起任务。

from runninghub_sdk import RunningHubClient, modify_nodes

with RunningHubClient(api_key="your-api-key") as client:
    demo = client.get_ai_app_api_demo("1937084629516193794")
    print(demo.webapp_name)

    uploaded = client.upload_image("./assets/reference.png")

    modifier = (
        modify_nodes()
        .set("39", "image", uploaded["fileName"])
        .set("52", "prompt", "保持人物身份一致,改成电影海报风格")
        .set("37", "aspect_ratio", "3:4")
    )

    task = client.run_ai_app_with_modifier(
        "1937084629516193794",
        modifier,
    )
    outputs = client.wait_for_completion(task.task_id)

    for output in outputs:
        print(output.file_type, output.file_url)

案例 3:调用标准模型 API 前先做价格预估

适合标准模型 API 接口较多、计费需要前置校验的情况。推荐顺序:先 preview_model_price(),再 run_model_api(),最后走 V2 查询。

from runninghub_sdk import RunningHubClient

endpoint = "rhart-image/f-2-dev/text-to-image"
payload = {
    "12##text": "a product poster of a premium coffee grinder on a marble table",
    "41##select": "4:3",
    "30##value": 1280,
    "29##value": 960,
    "43##file_type": "png",
}

with RunningHubClient(api_key="your-api-key") as client:
    price = client.preview_model_price(endpoint, payload)
    print("预估价格:", price.estimated_price, price.currency)

    task = client.run_model_api(endpoint, payload)
    result = client.wait_for_query_v2_completion(task.task_id)

    print("任务状态:", result.status)
    print("输出结果:", result.results)

案例 4:上线前做账户、队列和 webhook 自检

适合接入生产环境前检查账户余额、当前队列占用,以及排查 webhook 回调失败问题。

from runninghub_sdk import RunningHubClient

with RunningHubClient(api_key="your-api-key") as client:
    account = client.get_account_status()
    print(account.remain_coins, account.api_type, account.api_type_enum)

    queue = client.get_queue_status()
    print(queue.api_key_type, queue.api_key_type_enum)
    print(queue.running_count, queue.queued_count)

    detail = client.get_webhook_detail("1904154698679771137")
    print(detail.callback_status, detail.callback_status_enum)
    print(detail.callback_response)

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

runninghub_sdk-1.1.2.tar.gz (34.0 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

runninghub_sdk-1.1.2-py3-none-any.whl (30.1 kB view details)

Uploaded Python 3

File details

Details for the file runninghub_sdk-1.1.2.tar.gz.

File metadata

  • Download URL: runninghub_sdk-1.1.2.tar.gz
  • Upload date:
  • Size: 34.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for runninghub_sdk-1.1.2.tar.gz
Algorithm Hash digest
SHA256 4c46a1dc3ab9735d6b197d2b3d02ce661b4732bdf1a1c53e379de97a2781536a
MD5 44de692340b9d932393e1b7dcf9e684e
BLAKE2b-256 495c0bcb77a9c4cb16898568862e197a5f4fb9798d1f1dfeffba8d252bd3c051

See more details on using hashes here.

File details

Details for the file runninghub_sdk-1.1.2-py3-none-any.whl.

File metadata

  • Download URL: runninghub_sdk-1.1.2-py3-none-any.whl
  • Upload date:
  • Size: 30.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for runninghub_sdk-1.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 7ac17e11cdc74a0cc8731783e6b0a9514a320ebdf5683a73d4c5a75f6473e67e
MD5 e9c2d64a8951703c55afd1c45ab85266
BLAKE2b-256 190e093e10f5c61edf8e986feccc6566b915d7b7d32b7c53e2f3a44568ad8173

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page