From 35d6368779fc371ca8c0e52c112ce9372e435c96 Mon Sep 17 00:00:00 2001 From: Yi Zhao Date: Fri, 29 Mar 2024 23:09:28 +0800 Subject: [PATCH] enlarge context to 8K --- handlers/claude.py | 5 ++--- handlers/gemini.py | 5 ++--- handlers/yi.py | 3 +-- 3 files changed, 5 insertions(+), 8 deletions(-) diff --git a/handlers/claude.py b/handlers/claude.py index 85e6e5d..ffae8e3 100644 --- a/handlers/claude.py +++ b/handlers/claude.py @@ -38,8 +38,7 @@ def claude_handler(message: Message, bot: TeleBot) -> None: else: player_message = claude_player_dict[str(message.from_user.id)] - q = m.strip() - if q == "clear" or len(q) == 0: + if m.strip() == "clear": bot.reply_to( message, "just clear you claude messages history", @@ -119,7 +118,7 @@ def claude_pro_handler(message: Message, bot: TeleBot) -> None: # tricky player_message.pop() r = client.messages.create( - max_tokens=4096, + max_tokens=8192, messages=player_message, model=ANTHROPIC_PRO_MODEL, stream=True, diff --git a/handlers/gemini.py b/handlers/gemini.py index 0f1c2ba..8a89f2f 100644 --- a/handlers/gemini.py +++ b/handlers/gemini.py @@ -15,7 +15,7 @@ generation_config = { "temperature": 0.7, "top_p": 1, "top_k": 1, - "max_output_tokens": 4096, + "max_output_tokens": 8192, } safety_settings = [ @@ -49,8 +49,7 @@ def gemini_handler(message: Message, bot: TeleBot) -> None: gemini_player_dict[str(message.from_user.id)] = player else: player = gemini_player_dict[str(message.from_user.id)] - q = m.strip() - if q == "clear" or len(q) == 0: + if m.strip() == "clear": bot.reply_to( message, "just clear you gemini messages history", diff --git a/handlers/yi.py b/handlers/yi.py index 3f3230b..b22b570 100644 --- a/handlers/yi.py +++ b/handlers/yi.py @@ -35,8 +35,7 @@ def yi_handler(message: Message, bot: TeleBot) -> None: ) else: player_message = yi_player_dict[str(message.from_user.id)] - q = m.strip() - if q == "clear" or len(q) == 0: + if m.strip() == "clear": bot.reply_to( message, "just clear your yi messages history",