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

974958ff · comfyanonymous · 3a7c3acc · 974958ff
Commit 974958ff authored May 10, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

server.py server.py +4 -2

No files found.
--- a/server.py
+++ b/server.py
@@ -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]