diff --git a/comfy/cli_args.py b/comfy/cli_args.py index a4cabcc65..81432ccd2 100644 --- a/comfy/cli_args.py +++ b/comfy/cli_args.py @@ -231,6 +231,13 @@ parser.add_argument( help="Set the base URL for the ComfyUI API. (default: https://api.comfy.org)", ) +parser.add_argument( + "--comfy-platform-base", + type=str, + default="https://platform.comfy.org", + help="Set the base URL for the ComfyUI Platform. (default: https://platform.comfy.org)", +) + database_default_path = os.path.abspath( os.path.join(os.path.dirname(__file__), "..", "user", "comfyui.db") ) diff --git a/comfy_api/feature_flags.py b/comfy_api/feature_flags.py index adb5a3144..6d9dc75d0 100644 --- a/comfy_api/feature_flags.py +++ b/comfy_api/feature_flags.py @@ -99,6 +99,8 @@ _CORE_FEATURE_FLAGS: dict[str, Any] = { "extension": {"manager": {"supports_v4": True}}, "node_replacements": True, "assets": args.enable_assets, + "comfy_api_base_url": args.comfy_api_base, + "comfy_platform_base_url": args.comfy_platform_base, } # CLI-provided flags cannot overwrite core flags