Feat: allow custom prompt_id

Eleo-ai · Mar 6, 2024 · 067c9f3 · 067c9f3
1 parent 7bc9086
commit 067c9f3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/server.py b/server.py
@@ -519,6 +519,9 @@ async def post_prompt(request):
                     extra_data["client_id"] = json_data["client_id"]
                 if valid[0]:
                     prompt_id = str(uuid.uuid4())
+                    if "prompt_id" in json_data:
+                        prompt_id = json_data["prompt_id"]
+
                     outputs_to_execute = valid[2]
                     self.prompt_queue.put(
                         (number, prompt_id, prompt, extra_data, outputs_to_execute))