randydev commited on
Commit
9678ff1
1 Parent(s): 33068c5

Update database.py

Browse files
Files changed (1) hide show
  1. 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 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") 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}