70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
from comfy_api.latest import ComfyExtension, io
|
||
|
||
|
||
def build_config_options(base_url, api_key, api_protocol, timeout):
|
||
"""Return only values that explicitly override a target API config."""
|
||
base_url = base_url.strip() if isinstance(base_url, str) else ""
|
||
api_key = api_key.strip() if isinstance(api_key, str) else ""
|
||
options = {}
|
||
if base_url:
|
||
options["base_url"] = base_url
|
||
if api_key:
|
||
options["api_key"] = api_key
|
||
if api_protocol and api_protocol != "inherit":
|
||
options["api_protocol"] = api_protocol
|
||
if isinstance(timeout, int) and not isinstance(timeout, bool) and timeout > 0:
|
||
options["timeout"] = timeout
|
||
return options
|
||
|
||
|
||
class ConfigOptions(io.ComfyNode):
|
||
"""
|
||
这个节点用于覆盖 API 的配置参数（base_url、api_key、api_protocol、timeout）
|
||
"""
|
||
|
||
@classmethod
|
||
def define_schema(cls) -> io.Schema:
|
||
return io.Schema(
|
||
node_id="YCYY_API_Config_Options",
|
||
display_name="API Config Options",
|
||
category="YCYY/API/utils",
|
||
inputs=[
|
||
io.String.Input(
|
||
id="base_url",
|
||
multiline=True,
|
||
tooltip="Override the API base URL"
|
||
),
|
||
io.String.Input(
|
||
id="api_key",
|
||
default="",
|
||
multiline=True,
|
||
tooltip="Override the API key"
|
||
),
|
||
io.Combo.Input(
|
||
id="api_protocol",
|
||
options=["inherit", "openai-completions", "openai-responses", "anthropic-messages"],
|
||
default="inherit",
|
||
tooltip="Override the API protocol, or inherit the selected API configuration"
|
||
),
|
||
io.Int.Input(
|
||
id="timeout",
|
||
default=0,
|
||
min=0,
|
||
max=600,
|
||
tooltip="Override the request timeout in seconds; 0 inherits the selected API configuration"
|
||
)
|
||
],
|
||
outputs=[
|
||
io.Custom("YCYY_API_CONFIG_OPTIONS").Output(
|
||
id="config_options",
|
||
display_name="config_options",
|
||
tooltip="Override configuration options"
|
||
)
|
||
],
|
||
description="This node provides override API config options."
|
||
)
|
||
|
||
@classmethod
|
||
def execute(cls, base_url, api_key, api_protocol, timeout) -> io.NodeOutput:
|
||
return io.NodeOutput(build_config_options(base_url, api_key, api_protocol, timeout))
|