cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-07-06 14: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

10
handlers/yi.py
View File

@ -6,6 +6,7 @@ import requests
from telebot import TeleBot
from telebot.types import Message
from telegramify_markdown import convert
from expiringdict import ExpiringDict
from . import *
@ -21,7 +22,8 @@ client = OpenAI(
)
# Global history cache
yi_player_dict = {}
yi_player_dict = ExpiringDict(max_len=1000, max_age_seconds=300)
yi_pro_player_dict = ExpiringDict(max_len=1000, max_age_seconds=300)
def yi_handler(message: Message, bot: TeleBot) -> None:
@ -95,12 +97,12 @@ def yi_pro_handler(message: Message, bot: TeleBot) -> None:
player_message = []
# restart will lose all TODO
if str(message.from_user.id) not in yi_player_dict:
yi_player_dict[str(message.from_user.id)] = (
if str(message.from_user.id) not in yi_pro_player_dict:
yi_pro_player_dict[str(message.from_user.id)] = (
player_message # for the imuutable list
)
else:
player_message = yi_player_dict[str(message.from_user.id)]
player_message = yi_pro_player_dict[str(message.from_user.id)]
if m.strip() == "clear":
bot.reply_to(
message,