qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make the prompt_id a uuid and return it when queueing the prompt.

This commit is contained in:
comfyanonymous 2023-05-10 16:41:43 -04:00
parent 3a7c3acc72
commit 974958ff81
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server.py
View File

@ -81,7 +81,7 @@ class PromptServer():
# Reusing existing session, remove old # Reusing existing session, remove old
self.sockets.pop(sid, None) self.sockets.pop(sid, None)
else: else:
sid = uuid.uuid4().hex sid = uuid.uuid4().hex
self.sockets[sid] = ws self.sockets[sid] = ws
@ -313,7 +313,9 @@ class PromptServer():
if "client_id" in json_data: if "client_id" in json_data:
extra_data["client_id"] = json_data["client_id"] extra_data["client_id"] = json_data["client_id"]
if valid[0]: if valid[0]:
self.prompt_queue.put((number, id(prompt), prompt, extra_data, valid[2])) prompt_id = str(uuid.uuid4())
self.prompt_queue.put((number, prompt_id, prompt, extra_data, valid[2]))
return web.json_response({"prompt_id": prompt_id})
else: else:
resp_code = 400 resp_code = 400
out_string = valid[1] out_string = valid[1]