Upload database.py

akn/utils/database.py

@@ -16,6 +16,7 @@ class Database:
         # tiktok bot
         self.db_tiktok = self.client["tiktokbot"]
         self.magic_bot = self.db_tiktok["magicbot"]
+        self.gemini_bot = self.db_tiktok["geminibot"]
 
         # akeno
         self.afk = self.db["afk"]
@@ -139,6 +140,9 @@ class Database:
     async def get_all_magic_bot(self) -> list:
         return [i async for i in self.magic_bot.find({})]
 
+    async def get_all_gemini_bot(self) -> list:
+        return [i async for i in self.gemini_bot.find({})]
+
     async def is_gbanned(self, user_id: int) -> bool:
         if await self.gban.find_one({"user_id": user_id}):
             return True