wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-19 21:08:19 +08:00
Code Issues Actions 24 Packages Projects Releases Wiki Activity

Add guardrails

Browse Source
This commit is contained in:
Alexis Rolland 2026-07-14 17:14:51 +08:00
parent 718d3a6e2b
commit 9ea052a045
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy_api_nodes/util/_helpers.py
View File

@ -58,13 +58,16 @@ def get_comfy_api_headers(node_cls: type[IO.ComfyNode]) -> dict[str, str]:
relative/cloud URLs resolved against ``default_base_url()``; because the result relative/cloud URLs resolved against ``default_base_url()``; because the result
includes auth, callers must not attach it to arbitrary absolute/presigned URLs. includes auth, callers must not attach it to arbitrary absolute/presigned URLs.
""" """
return { headers = {
**get_auth_header(node_cls), **get_auth_header(node_cls),
"Comfy-Env": get_deploy_environment(), "Comfy-Env": get_deploy_environment(),
"Comfy-Usage-Source": get_usage_source(node_cls), "Comfy-Usage-Source": get_usage_source(node_cls),
"Comfy-Core-Version": comfyui_version, "Comfy-Core-Version": comfyui_version,
"Comfy-Job-Id": get_executing_context().prompt_id,
} }
ctx = get_executing_context()
if ctx is not None:
headers["Comfy-Job-Id"] = ctx.prompt_id
return headers
def default_base_url() -> str: def default_base_url() -> str: