from comfy_api.latest import ComfyExtension, io

class ConfigOptions(io.ComfyNode):
    """
    这个节点用于覆盖 API 的配置参数（api_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="api_url",
                    multiline=True,
                    tooltip="Override the API URL"
                ),
                io.String.Input(
                    id="api_key",
                    default="",
                    multiline=True,
                    tooltip="Override the API key"
                ),
                io.Combo.Input(
                    id="api_protocol",
                    options=["openai-completions", "openai-responses", "anthropic-messages"],
                    default="openai-completions",
                    tooltip="Override the API protocol"
                ),
                io.Int.Input(
                    id="timeout",
                    default=120,
                    min=1,
                    max=600,
                    tooltip="Override the request timeout in seconds"
                )
            ],
            outputs=[
                io.AnyType.Output(
                    id="config_options",
                    display_name="config_options",
                    tooltip="Override configuration options"
                )
            ],
            description="This node provides override API config options."
        )

    @classmethod
    def execute(cls, api_url, api_key, api_protocol, timeout) -> io.NodeOutput:
        # 验证和清理输入
        api_url = api_url.strip() if api_url else ""
        api_key = api_key.strip() if api_key else ""

        config_options = {
            "api_url": api_url,
            "api_key": api_key,
            "api_protocol": api_protocol,
            "timeout": timeout
        }

        return io.NodeOutput(config_options)