cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-11-04 16:56:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: update gemini model-s

Browse Source
This commit is contained in:
yihong0618
2024-12-13 19:30:00 +08:00
parent 5cb3df00b8
commit 3b65e61377
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
handlers/gemini.py
View File

@ -16,7 +16,7 @@ from . import *
markdown_symbol.head_level_1 = "📌" # If you want, Customizing the head level 1 symbol markdown_symbol.head_level_1 = "📌" # If you want, Customizing the head level 1 symbol
markdown_symbol.link = "🔗" # If you want, Customizing the link symbol markdown_symbol.link = "🔗" # If you want, Customizing the link symbol
GOOGLE_GEMINI_KEY = environ.get("GOOGLE_GEMINI_KEY") GOOGLE_GEMINI_KEY = environ.get("GEMIMI_PRO_KEY")
genai.configure(api_key=GOOGLE_GEMINI_KEY) genai.configure(api_key=GOOGLE_GEMINI_KEY)
generation_config = { generation_config = {
@ -40,9 +40,9 @@ gemini_file_player_dict = ExpiringDict(max_len=100, max_age_seconds=600)
def make_new_gemini_convo(is_pro=False) -> ChatSession: def make_new_gemini_convo(is_pro=False) -> ChatSession:
model_name = "gemini-1.5-flash-latest" model_name = "gemini-1.5-flash-002"
if is_pro: if is_pro:
model_name = "gemini-1.5-pro-exp-0801" model_name = "gemini-2.0-flash-exp"
model = genai.GenerativeModel( model = genai.GenerativeModel(
model_name=model_name, model_name=model_name,
@ -188,7 +188,7 @@ def gemini_photo_handler(message: Message, bot: TeleBot) -> None:
with open("gemini_temp.jpg", "wb") as temp_file: with open("gemini_temp.jpg", "wb") as temp_file:
temp_file.write(downloaded_file) temp_file.write(downloaded_file)
model = genai.GenerativeModel("gemini-pro-vision") model = genai.GenerativeModel("gemini-2.0-flash-exp")
with open("gemini_temp.jpg", "rb") as image_file: with open("gemini_temp.jpg", "rb") as image_file:
image_data = image_file.read() image_data = image_file.read()
contents = { contents = {