Upload chat.py

chatbot/plugins/chat.py

@@ -38,8 +38,7 @@ from pyrogram import Client, filters
 from pyrogram.types import *
 from pyrogram.errors import *
 from config import *
-from database import db
-from database import users_collection
+from database import users_collection, db
 from logger import LOGS
 
 from akenoai import *
@@ -179,11 +178,13 @@ gen = genai.Client(api_key=GOOGLE_API_KEY)
 async def search_auto_by_cohere(user_id, prompt):
     COHERE_API_KEY = os.environ.get("COHERE_API_KEY")
     co = cohere.Client(api_key=COHERE_API_KEY)
+    chat_history = await db._get_cohere_chat_from_db(user_id)
+    chat_history.append({"role": "User", "message": prompt})
     stream = co.chat_stream( 
         model='command-r-08-2024',
         message=prompt,
         temperature=0.3,
-        chat_history=[],
+        chat_history=chat_history,
         prompt_truncation='AUTO',
         connectors=[{"id":"web-search"}]
     )
@@ -218,6 +219,8 @@ async def search_auto_by_cohere(user_id, prompt):
         output = "I couldn't generate a response."
 
     all_answer = f"{output}\n\n{references}"
+    chat_history.append({"role": "Chatbot", "message": all_answer})
+    await db._update_cohere_chat_in_db(user_id, chat_history)
     return all_answer