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

return prompt as dict rather than array

Browse Source
This commit is contained in:
Richard Yu 2025-07-11 15:27:30 -07:00
parent fdc2a53653
commit 36e73633af
2 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
execution.py
View File

@ -1138,7 +1138,11 @@ class PromptQueue:
# Remove workflow from extra_pnginfo # Remove workflow from extra_pnginfo
if "extra_pnginfo" in extra_data: if "extra_pnginfo" in extra_data:
extra_data["extra_pnginfo"].pop("workflow", None) extra_data["extra_pnginfo"].pop("workflow", None)
filtered_prompt = [priority, prompt_id, extra_data] filtered_prompt = {
"priority": priority,
"prompt_id": prompt_id,
"extra_data": extra_data
}
else: else:
filtered_prompt = None filtered_prompt = None

12
tests/inference/test_execution.py
View File

@ -798,15 +798,17 @@ class TestExecution:
# Verify prompt field is filtered # Verify prompt field is filtered
filtered_prompt = history_item["prompt"] filtered_prompt = history_item["prompt"]
assert isinstance(filtered_prompt, list), "Filtered prompt should be a list" assert isinstance(filtered_prompt, dict), "Filtered prompt should be a dictionary"
assert len(filtered_prompt) == 3, f"Filtered prompt should have 3 elements, got {len(filtered_prompt)}" assert "priority" in filtered_prompt, "Filtered prompt should have priority"
assert "prompt_id" in filtered_prompt, "Filtered prompt should have prompt_id"
assert "extra_data" in filtered_prompt, "Filtered prompt should have extra_data"
# Verify correct elements are preserved # Verify correct elements are preserved
assert filtered_prompt[0] == 12345, "Priority should be preserved" assert filtered_prompt["priority"] == 12345, "Priority should be preserved"
assert filtered_prompt[1] == 'test-prompt-123', "Prompt ID should be preserved" assert filtered_prompt["prompt_id"] == 'test-prompt-123', "Prompt ID should be preserved"
# Verify extra_data filtering # Verify extra_data filtering
extra_data = filtered_prompt[2] extra_data = filtered_prompt["extra_data"]
assert extra_data['client_id'] == 'test-client', "Client ID should be preserved" assert extra_data['client_id'] == 'test-client', "Client ID should be preserved"
assert 'extra_pnginfo' in extra_data, "extra_pnginfo should be present" 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 'workflow' not in extra_data['extra_pnginfo'], "Workflow should be filtered out"