_

database.py

@@ -28,7 +28,8 @@ class Database:
         self.user_premium = self.db["user_premium"]
         self.user_blacklists = self.db["user_blacklist"]
         self.backup_chatbot = self.db["google_genai"]
-
+        self.cohere = self.db["cohere"]
+        
     async def connect(self):
         try:
             await self.client.admin.command("ping")
@@ -49,6 +50,17 @@ class Database:
     def get_datetime(self) -> str:
         return datetime.datetime.now().strftime("%d/%m/%Y - %H:%M")
 
+    async def _get_cohere_chat_from_db(self, user_id):
+        user_data = await self.cohere.find_one({"user_id": user_id})
+        return user_data.get("cohere_chat", []) if user_data else []
+
+    async def _update_cohere_chat_in_db(self, user_id, cohere_chat):
+        await self.cohere.update_one(
+            {"user_id": user_id},
+            {"$set": {"cohere_chat": cohere_chat}},
+            upsert=True
+        )
+
     async def _update_chatbot_chat_in_db(self, user_id, chatbot_chat):
         await self.backup_chatbot.update_one(
             {"user_id": user_id},