xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 23:37:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove workflow json from history_v2 response - client will need to use /history_v2/:prompt_id

Browse Source
This commit is contained in:
Richard Yu 2025-07-10 15:49:27 -07:00
parent a2b8688e20
commit fdc2a53653
2 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
execution.py
View File

@ -1135,6 +1135,9 @@ class PromptQueue:
# Extract and filter prompt data # Extract and filter prompt data
if "prompt" in history_entry: if "prompt" in history_entry:
priority, prompt_id, _, extra_data, _ = history_entry["prompt"] priority, prompt_id, _, extra_data, _ = history_entry["prompt"]
# Remove workflow from extra_pnginfo
if "extra_pnginfo" in extra_data:
extra_data["extra_pnginfo"].pop("workflow", None)
filtered_prompt = [priority, prompt_id, extra_data] filtered_prompt = [priority, prompt_id, extra_data]
else: else:
filtered_prompt = None filtered_prompt = None

16
tests/inference/test_execution.py
View File

@ -759,7 +759,13 @@ class TestExecution:
12345, # priority/timestamp 12345, # priority/timestamp
'test-prompt-123', # prompt_id 'test-prompt-123', # prompt_id
{'nodes': {'1': {'class_type': 'SaveImage'}}}, # workflow (should be filtered) {'nodes': {'1': {'class_type': 'SaveImage'}}}, # workflow (should be filtered)
{'client_id': 'test-client'}, # extra_data { # extra_data
'client_id': 'test-client',
'extra_pnginfo': {
'workflow': {'nodes': {'1': {'class_type': 'SaveImage'}}}, # Should be filtered out
'version': '1.0' # Should be preserved
}
},
['1'] # execute_outputs (should be filtered) ['1'] # execute_outputs (should be filtered)
) )
@ -798,7 +804,13 @@ class TestExecution:
# Verify correct elements are preserved # Verify correct elements are preserved
assert filtered_prompt[0] == 12345, "Priority should be preserved" assert filtered_prompt[0] == 12345, "Priority should be preserved"
assert filtered_prompt[1] == 'test-prompt-123', "Prompt ID should be preserved" assert filtered_prompt[1] == 'test-prompt-123', "Prompt ID should be preserved"
assert filtered_prompt[2] == {'client_id': 'test-client'}, "Extra data should be preserved"
# Verify extra_data filtering
extra_data = filtered_prompt[2]
assert extra_data['client_id'] == 'test-client', "Client ID should be preserved"
assert 'extra_pnginfo' in extra_data, "extra_pnginfo should be present"
assert 'workflow' not in extra_data['extra_pnginfo'], "Workflow should be filtered out"
assert extra_data['extra_pnginfo']['version'] == '1.0', "Other extra_pnginfo data should be preserved"
# Verify other fields are unchanged # Verify other fields are unchanged
assert history_item["outputs"] == {'1': {'images': []}}, "Outputs should be unchanged" assert history_item["outputs"] == {'1': {'images': []}}, "Outputs should be unchanged"