fix prompting

This commit is contained in:
Owl 2025-03-14 16:36:52 +07:00
parent 7b3fe6460b
commit c5afe9bf85

View File

@ -39,8 +39,10 @@ async def ws_task():
message = await ws.recv() message = await ws.recv()
message_dict = json.loads(message) message_dict = json.loads(message)
prompt_id = message_dict["id"] prompt = message_dict["data"]
image_url = filename2cdn(message_dict["image_filename"])
prompt_id = prompt["id"]
image_url = filename2cdn(prompt["image_filename"])
chat_id = scheduled_prompts.get(prompt_id) chat_id = scheduled_prompts.get(prompt_id)
print(f"Received message: {message}") print(f"Received message: {message}")