Spaces:
Running
Running
Update database.py
Browse files- database.py +2 -2
database.py
CHANGED
@@ -640,9 +640,9 @@ class Database:
|
|
640 |
user_data = await self.chatbot.find_one({"chat_id": chat_id})
|
641 |
return user_data.get("user_id") if user_data else None
|
642 |
|
643 |
-
async def
|
644 |
user_data = await self.gemini_plan.find_one({"user_id": user_id})
|
645 |
-
return user_data.get("is_plan") if user_data else None
|
646 |
|
647 |
async def remove_chatbot(self, chat_id):
|
648 |
unset_data = {"user_id": None}
|
|
|
640 |
user_data = await self.chatbot.find_one({"chat_id": chat_id})
|
641 |
return user_data.get("user_id") if user_data else None
|
642 |
|
643 |
+
async def is_gemini_plan(self, user_id):
|
644 |
user_data = await self.gemini_plan.find_one({"user_id": user_id})
|
645 |
+
return user_data.get("is_plan", False) if user_data else None
|
646 |
|
647 |
async def remove_chatbot(self, chat_id):
|
648 |
unset_data = {"user_id": None}
|