From c5afe9bf85ced33da45c012e5fff1f1a53896d28 Mon Sep 17 00:00:00 2001
From: owl <lonelyowl@hitler.rocks>
Date: Fri, 14 Mar 2025 16:36:52 +0700
Subject: [PATCH] fix prompting
---
tgbot/shit/prompting.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/tgbot/shit/prompting.py b/tgbot/shit/prompting.py
index c906f00..4dcea1b 100644
--- a/tgbot/shit/prompting.py
+++ b/tgbot/shit/prompting.py
@@ -39,8 +39,10 @@ async def ws_task():
message = await ws.recv()
message_dict = json.loads(message)
- prompt_id = message_dict["id"]
- image_url = filename2cdn(message_dict["image_filename"])
+ prompt = message_dict["data"]
+
+ prompt_id = prompt["id"]
+ image_url = filename2cdn(prompt["image_filename"])
chat_id = scheduled_prompts.get(prompt_id)
print(f"Received message: {message}")