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

feat: gemini new model

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong0618
2024-08-02 14:13:24 +08:00
parent 7ecafca139
commit 34571633a2
6 changed files with 109 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
handlers/gemini.py
View File

@ -42,12 +42,13 @@ gemini_file_player_dict = ExpiringDict(max_len=100, max_age_seconds=600)
def make_new_gemini_convo(is_pro=False) -> ChatSession:
model_name = "gemini-1.5-flash-latest"
if is_pro:
model_name = "models/gemini-1.5-pro-latest"
model_name = "gemini-1.5-pro-exp-0801"
model = genai.GenerativeModel(
model_name=model_name,
generation_config=generation_config,
safety_settings=safety_settings,
tools="code_execution",
)
convo = model.start_chat()
return convo