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

feat: use expire dict

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong0618
2024-06-14 18:55:20 +08:00
parent bc3ecdb9f8
commit 785b75e994
13 changed files with 127 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
handlers/gemini.py
View File

@ -7,8 +7,8 @@ from google.generativeai import ChatSession
from google.generativeai.types.generation_types import StopCandidateException
from telebot import TeleBot
from telebot.types import Message
from expiringdict import ExpiringDict
from telegramify_markdown import convert
from telegramify_markdown.customize import markdown_symbol
from . import *
@ -34,9 +34,9 @@ safety_settings = [
]
# Global history cache
gemini_player_dict = {}
gemini_pro_player_dict = {}
gemini_file_player_dict = {}
gemini_player_dict = ExpiringDict(max_len=1000, max_age_seconds=300)
gemini_pro_player_dict = ExpiringDict(max_len=1000, max_age_seconds=300)
gemini_file_player_dict = ExpiringDict(max_len=100, max_age_seconds=300)
def make_new_gemini_convo(is_pro=False) -> ChatSession: