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

View File

@ -81,7 +81,7 @@ class PromptServer():
# Reusing existing session, remove old
self.sockets.pop(sid, None)
else:
sid = uuid.uuid4().hex
sid = uuid.uuid4().hex
self.sockets[sid] = ws
@ -313,7 +313,9 @@ class PromptServer():
if "client_id" in json_data:
extra_data["client_id"] = json_data["client_id"]
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:
resp_code = 400
out_string = valid[1]