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="
|
9 |
-
InlineKeyboardButton("Gemini EXP", callback_data="
|
10 |
])
|
11 |
keyboard.append([
|
12 |
-
InlineKeyboardButton("Gemini Flash 8B", callback_data="
|
13 |
-
InlineKeyboardButton("Gemini 2.0 Flash ", callback_data="
|
14 |
])
|
15 |
keyboard.append([
|
16 |
-
InlineKeyboardButton("Gemini PRO", callback_data="
|
17 |
-
InlineKeyboardButton("Gemini Flash", callback_data="
|
18 |
])
|
19 |
return InlineKeyboardMarkup(keyboard)
|
20 |
|
21 |
-
@Client.on_callback_query(filters.regex(r"^
|
22 |
-
async def
|
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(
|