cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-11-04 08:46:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: change gpt4 model and claude latest

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong0618
2024-04-18 11:12:41 +08:00
parent 385df6c821
commit cd2185c7d6
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
handlers/chatgpt.py
View File

@ -16,7 +16,7 @@ markdown_symbol.link = "🔗" # If you want, Customizing the link symbol
CHATGPT_API_KEY = environ.get("OPENAI_API_KEY") CHATGPT_API_KEY = environ.get("OPENAI_API_KEY")
CHATGPT_BASE_URL = environ.get("OPENAI_API_BASE") or "https://api.openai.com/v1" CHATGPT_BASE_URL = environ.get("OPENAI_API_BASE") or "https://api.openai.com/v1"
CHATGPT_MODEL = "gpt-3.5-turbo" 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) client = OpenAI(api_key=CHATGPT_API_KEY, base_url=CHATGPT_BASE_URL, timeout=20)

5
handlers/claude.py
View File

@ -117,8 +117,9 @@ def claude_pro_handler(message: Message, bot: TeleBot) -> None:
reply_id = bot_reply_first(message, who, bot) reply_id = bot_reply_first(message, who, bot)
player_message.append({"role": "user", "content": m}) player_message.append({"role": "user", "content": m})
# keep the last 5, every has two ask and answer. # keep the last 2, every has two ask and answer.
if len(player_message) > 10: # its too expensive
if len(player_message) > 4:
player_message = player_message[2:] player_message = player_message[2:]
try: try: