randydev commited on
Commit
27501d9
·
verified ·
1 Parent(s): 08a91d1

Update chatbot/plugins/keyboards_gemini.py

Browse files
chatbot/plugins/keyboards_gemini.py CHANGED
@@ -5,21 +5,21 @@ from database import users_collection
5
  def get_gemini_keyboard():
6
  keyboard = []
7
  keyboard.append([
8
- InlineKeyboardButton("Gemini Flash EXP", callback_data="model_gemini-2.0-flash-exp"),
9
- InlineKeyboardButton("Gemini EXP", callback_data="model_gemini-exp-1206")
10
  ])
11
  keyboard.append([
12
- InlineKeyboardButton("Gemini Flash 8B", callback_data="model_gemini-1.5-flash-8b"),
13
- InlineKeyboardButton("Gemini 2.0 Flash ", callback_data="model_gemini-2.0-flash-001")
14
  ])
15
  keyboard.append([
16
- InlineKeyboardButton("Gemini PRO", callback_data="model_gemini-1.5-pro"),
17
- InlineKeyboardButton("Gemini Flash", callback_data="model_gemini-1.5-flash"),
18
  ])
19
  return InlineKeyboardMarkup(keyboard)
20
 
21
- @Client.on_callback_query(filters.regex(r"^model_"))
22
- async def set_model(client: Client, callback_query: CallbackQuery):
23
  model = callback_query.data.split("_")[1]
24
  user_id = callback_query.from_user.id
25
  await users_collection.update_one(
 
5
  def get_gemini_keyboard():
6
  keyboard = []
7
  keyboard.append([
8
+ InlineKeyboardButton("Gemini Flash EXP", callback_data="models_gemini-2.0-flash-exp"),
9
+ InlineKeyboardButton("Gemini EXP", callback_data="models_gemini-exp-1206")
10
  ])
11
  keyboard.append([
12
+ InlineKeyboardButton("Gemini Flash 8B", callback_data="models_gemini-1.5-flash-8b"),
13
+ InlineKeyboardButton("Gemini 2.0 Flash ", callback_data="models_gemini-2.0-flash-001")
14
  ])
15
  keyboard.append([
16
+ InlineKeyboardButton("Gemini PRO", callback_data="models_gemini-1.5-pro"),
17
+ InlineKeyboardButton("Gemini Flash", callback_data="models_gemini-1.5-flash"),
18
  ])
19
  return InlineKeyboardMarkup(keyboard)
20
 
21
+ @Client.on_callback_query(filters.regex(r"^models_$"))
22
+ async def set_modelok(client: Client, callback_query: CallbackQuery):
23
  model = callback_query.data.split("_")[1]
24
  user_id = callback_query.from_user.id
25
  await users_collection.update_one(