From cd2185c7d622438bee1907f68e26e02af747bc14 Mon Sep 17 00:00:00 2001 From: yihong0618 Date: Thu, 18 Apr 2024 11:12:41 +0800 Subject: [PATCH] fix: change gpt4 model and claude latest Signed-off-by: yihong0618 --- handlers/chatgpt.py | 2 +- handlers/claude.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/handlers/chatgpt.py b/handlers/chatgpt.py index 24578a7..3234100 100644 --- a/handlers/chatgpt.py +++ b/handlers/chatgpt.py @@ -16,7 +16,7 @@ markdown_symbol.link = "🔗" # If you want, Customizing the link symbol CHATGPT_API_KEY = environ.get("OPENAI_API_KEY") CHATGPT_BASE_URL = environ.get("OPENAI_API_BASE") or "https://api.openai.com/v1" CHATGPT_MODEL = "gpt-3.5-turbo" -CHATGPT_PRO_MODEL = "gpt-4-turbo" +CHATGPT_PRO_MODEL = "gpt-4-turbo-2024-04-09" client = OpenAI(api_key=CHATGPT_API_KEY, base_url=CHATGPT_BASE_URL, timeout=20) diff --git a/handlers/claude.py b/handlers/claude.py index a2c094f..3ee5fc1 100644 --- a/handlers/claude.py +++ b/handlers/claude.py @@ -117,8 +117,9 @@ def claude_pro_handler(message: Message, bot: TeleBot) -> None: reply_id = bot_reply_first(message, who, bot) player_message.append({"role": "user", "content": m}) - # keep the last 5, every has two ask and answer. - if len(player_message) > 10: + # keep the last 2, every has two ask and answer. + # its too expensive + if len(player_message) > 4: player_message = player_message[2:] try: