Upload chat.py

chatbot/plugins/chat.py

@@ -37,15 +37,23 @@ from chatbot.plugins.languages import LANGUAGES
 
 from . import force_sub
 
-async def process_stream(message):
+SYSTEM_PROMPT = f"""
+Your name is Randy Dev. A kind and friendly AI assistant that answers in a short and concise answer.
+Give short step-by-step reasoning if required.
+
+python language powered by @xtdevs on telegram support and language models Meta AI
+
+{datetime.datetime.now()}
+"""
+
+async def process_stream(message, prompt):
     client_hf = InferenceClient(api_key=HF_KEY)
-    messages=[
-        {"role": "system", "content": f"Your name is Randy Dev. A kind and friendly AI assistant that answers in a short and concise answer.\nGive short step-by-step reasoning if required.\n\n{datetime.datetime.now()}"},
-        {"role": "user", "content": message}
-    ]
+    backup_chat = await db._get_openai_chat_from_db(message.from_user.id)
+    backup_chat.append({"role": "system", "content": BASE_PROMPT})
+    backup_chat.append({"role": "user", "content": prompt})
     stream = client_hf.chat.completions.create(
         model="mistralai/Mixtral-8x7B-Instruct-v0.1",
-        messages=messages,
+        messages=backup_chat,
         max_tokens=500,
         stream=True
     )
@@ -53,6 +61,8 @@ async def process_stream(message):
     for chunk in stream:
         new_content = chunk.choices[0].delta.content
         accumulated_text += new_content
+    backup_chat.append({"role": "assistant", "content": accumulated_text})
+    await db._update_openai_chat_in_db(message.from_user.id, backup_chat)
     return accumulated_text
 
 @Client.on_message(
@@ -79,11 +89,13 @@ async def startbot(client: Client, message: Message):
         reply_markup=InlineKeyboardMarkup(buttons)
     )
 
+
 @Client.on_message(
-    (filters.private | filters.group)
+    filters.private
     & filters.command(["ask"])
     & ~filters.forwarded
 )
+@force_sub
 async def askcmd(client: Client, message: Message):
     if len(message.command) > 1:
         prompt = message.text.split(maxsplit=1)[1]
@@ -94,7 +106,7 @@ async def askcmd(client: Client, message: Message):
     await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
     await asyncio.sleep(1.5)
     try:
-        output = await process_stream(prompt)
+        output = await process_stream(message, prompt)
         if len(output) > 4096:
             with open("chat.txt", "w+", encoding="utf8") as out_file:
                 out_file.write(output)
@@ -102,8 +114,11 @@ async def askcmd(client: Client, message: Message):
                 document="chat.txt",
                 disable_notification=True
             )
+            await pro.delete()
             os.remove("chat.txt")
         else:
             await message.reply_text(output, disable_web_page_preview=True)
+        await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
+        return
     except Exception as e:
         return await message.reply_text(f"Error: {e}")