298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-20 22:47:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor previous pr. (#8893)

Browse Source
This commit is contained in:
comfyanonymous 2025-07-13 01:59:17 -07:00 committed by GitHub
parent 480375f349
commit 4831e9c2c4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
execution.py
View File

@ -123,6 +123,8 @@ class CacheSet:
} }
return result return result
SENSITIVE_EXTRA_DATA_KEYS = ("auth_token_comfy_org", "api_key_comfy_org")
def get_input_data(inputs, class_def, unique_id, outputs=None, dynprompt=None, extra_data={}): def get_input_data(inputs, class_def, unique_id, outputs=None, dynprompt=None, extra_data={}):
valid_inputs = class_def.INPUT_TYPES() valid_inputs = class_def.INPUT_TYPES()
input_data_all = {} input_data_all = {}
@ -1045,11 +1047,10 @@ class PromptQueue:
if status is not None: if status is not None:
status_dict = copy.deepcopy(status._asdict()) status_dict = copy.deepcopy(status._asdict())
# Remove auth tokens from extra_data before storing in history # Remove sensitive data from extra_data before storing in history
if "auth_token_comfy_org" in prompt[3]: for sensitive_val in SENSITIVE_EXTRA_DATA_KEYS:
del prompt[3]["auth_token_comfy_org"] if sensitive_val in prompt[3]:
if "api_key_comfy_org" in prompt[3]: prompt[3].pop(sensitive_val)
del prompt[3]["api_key_comfy_org"]
self.history[prompt[1]] = { self.history[prompt[1]] = {
"prompt": prompt, "prompt": prompt,