From e1090f5f06bd253129fb22e2d5311eda3489e5ac Mon Sep 17 00:00:00 2001
From: Kyrylo Hrymailo <kirill.grim@gmail.com>
Date: Mon, 26 May 2025 13:19:31 +0200
Subject: [PATCH] Added passing of context to the prompt

---
 chat_server/sio/handlers/prompt.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/chat_server/sio/handlers/prompt.py b/chat_server/sio/handlers/prompt.py
index 4f54140..14d6815 100644
--- a/chat_server/sio/handlers/prompt.py
+++ b/chat_server/sio/handlers/prompt.py
@@ -53,12 +53,14 @@ async def new_prompt(sid, data):
     cid = data["cid"]
     prompt_text = data["prompt_text"]
     created_on = int(data.get("created_on") or time())
+    context = data.get("context", {})
     try:
         formatted_data = {
             "_id": prompt_id,
             "cid": cid,
             "is_completed": "0",
             "data": {"prompt_text": prompt_text},
+            "context": context,
             "created_on": created_on,
         }
         MongoDocumentsAPI.PROMPTS.add_item(data=formatted_data)