diff --git a/comfy_api_nodes/util/client.py b/comfy_api_nodes/util/client.py index 052301c33..b32075d43 100644 --- a/comfy_api_nodes/util/client.py +++ b/comfy_api_nodes/util/client.py @@ -51,7 +51,7 @@ class ApiEndpoint: @dataclass class _RequestConfig: - node_cls: type[IO.ComfyNode] + node_cls: type[IO.ComfyNode] | None endpoint: ApiEndpoint timeout: float content_type: str @@ -70,6 +70,17 @@ class _RequestConfig: price_extractor: Callable[[dict[str, Any]], float | None] | None = None is_rate_limited: Callable[[int, Any], bool] | None = None response_header_validator: Callable[[dict[str, str]], None] | None = None + base_url: str | None = None + auth_headers: dict[str, str] | None = None + allow_304: bool = False + error_parser: Callable[[int, Any], Exception | None] | None = None + # Optional callback to render a per-second progress label while + # waiting out a rate-limit / SERVER_BUSY / MAINTENANCE retry. Called + # with ``(status, body, retry_after_s)`` and should return the label + # string used by ``_display_time_progress`` (which renders it as + # ``Status: