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

remove unused "completed" and cached node info from history_v2 response

Browse Source
This commit is contained in:
Richard Yu 2025-07-10 14:50:19 -07:00
parent 565b537620
commit a2b8688e20
2 changed files with 54 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
execution.py
View File

@ -1130,13 +1130,28 @@ class PromptQueue:
# Build history items with prompt_id field
history_items = []
for key in selected_keys:
item = copy.deepcopy(self.history[key])
item["prompt_id"] = key
history_entry = self.history[key]
# Remove prompt[2] (workflow) and prompt[4] (execute_outputs) to reduce response size
if "prompt" in item:
priority, prompt_id, _, extra_data, _ = item["prompt"]
item["prompt"] = [priority, prompt_id, extra_data]
# Extract and filter prompt data
if "prompt" in history_entry:
priority, prompt_id, _, extra_data, _ = history_entry["prompt"]
filtered_prompt = [priority, prompt_id, extra_data]
else:
filtered_prompt = None
# Create lightweight history response
item = {
"prompt_id": key,
"outputs": history_entry.get("outputs", {}),
"meta": history_entry.get("meta", {}),
"prompt": filtered_prompt,
"status": {
"status_str": history_entry["status"]["status_str"],
"messages": [(e, {k: v for k, v in d.items() if k != "nodes"})
if e == "execution_cached" else (e, d)
for e, d in history_entry["status"]["messages"]]
} if history_entry.get("status") else None
}
history_items.append(item)

35
tests/inference/test_execution.py
View File

@ -766,7 +766,14 @@ class TestExecution:
queue.history['test-prompt-123'] = {
'prompt': mock_prompt_tuple,
'outputs': {'1': {'images': []}},
'status': {'completed': True, 'messages': []},
'status': {
'status_str': 'success',
'completed': True, # Should be filtered out
'messages': [
('execution_cached', {'nodes': ['node1', 'node2'], 'timestamp': 1234567890}), # 'nodes' should be filtered
('execution_start', {'timestamp': 1234567800}) # Should remain unchanged
]
},
'meta': {'1': {'node_id': '1'}}
}
@ -795,5 +802,29 @@ class TestExecution:
# Verify other fields are unchanged
assert history_item["outputs"] == {'1': {'images': []}}, "Outputs should be unchanged"
assert history_item["status"] == {'completed': True, 'messages': []}, "Status should be unchanged"
assert history_item["meta"] == {'1': {'node_id': '1'}}, "Meta should be unchanged"
# Verify status field filtering
status = history_item["status"]
assert "status_str" in status, "Status should have status_str"
assert status["status_str"] == 'success', "Status string should be preserved"
assert "completed" not in status, "Completed field should be filtered out"
assert "messages" in status, "Status should have messages"
# Verify message filtering
messages = status["messages"]
assert len(messages) == 2, "Should have 2 messages"
# Check execution_cached message has nodes filtered out
execution_cached_msg = messages[0]
assert execution_cached_msg[0] == 'execution_cached', "First message should be execution_cached"
cached_data = execution_cached_msg[1]
assert "nodes" not in cached_data, "Nodes field should be filtered from execution_cached messages"
assert "timestamp" in cached_data, "Timestamp should be preserved in execution_cached messages"
assert cached_data["timestamp"] == 1234567890, "Timestamp value should be correct"
# Check execution_start message remains unchanged
execution_start_msg = messages[1]
assert execution_start_msg[0] == 'execution_start', "Second message should be execution_start"
start_data = execution_start_msg[1]
assert start_data == {'timestamp': 1234567800}, "execution_start message should be unchanged"