Skip to content

Add timestamp to execution messages #4076

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
comfyanonymous merged 3 commits into from
Jul 21, 2024
Merged

Add timestamp to execution messages #4076

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
b071fff
Add timestamp to execution messages
huchenlei Jul 21, 2024
535cb42
Add execution_end message
huchenlei Jul 21, 2024
1b041cc
Rename to execution_success
huchenlei Jul 21, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 9 additions & 1 deletion execution.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import logging
import threading
import heapq
import time
import traceback
import inspect
from typing import List, Literal, NamedTuple, Optional
Expand Down Expand Up @@ -283,7 +284,11 @@ def reset(self):
self.success = True
self.old_prompt = {}

def add_message(self, event, data, broadcast: bool):
def add_message(self, event, data: dict, broadcast: bool):
data = {
**data,
"timestamp": int(time.time() * 1000),
}
self.status_messages.append((event, data))
if self.server.client_id is not None or broadcast:
self.server.send_sync(event, data, self.server.client_id)
Expand Down Expand Up @@ -394,6 +399,9 @@ def execute(self, prompt, prompt_id, extra_data={}, execute_outputs=[]):
if self.success is not True:
self.handle_execution_error(prompt_id, prompt, current_outputs, executed, error, ex)
break
else:
# Only execute when the while-loop ends without break
self.add_message("execution_success", { "prompt_id": prompt_id }, broadcast=False)

for x in executed:
self.old_prompt[x] = copy.deepcopy(prompt[x])
Expand Down