From 095f9cc108b12355c4760a8f6bdb1b2faeb88895 Mon Sep 17 00:00:00 2001 From: 6vision Date: Mon, 24 Jun 2024 11:20:50 +0800 Subject: [PATCH] feat: support claude-3.5-sonnet model --- bot/claudeapi/claude_api_bot.py | 2 ++ common/const.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/bot/claudeapi/claude_api_bot.py b/bot/claudeapi/claude_api_bot.py index dfb4b06..e5452e5 100644 --- a/bot/claudeapi/claude_api_bot.py +++ b/bot/claudeapi/claude_api_bot.py @@ -130,4 +130,6 @@ class ClaudeAPIBot(Bot, OpenAIImage): return "claude-3-sonnet-20240229" elif model == "claude-3-haiku": return "claude-3-haiku-20240307" + elif model == "claude-3.5-sonnet": + return "claude-3-5-sonnet-20240620" return model diff --git a/common/const.py b/common/const.py index 638c728..655848f 100644 --- a/common/const.py +++ b/common/const.py @@ -57,7 +57,7 @@ MODEL_LIST = [ GPT_4o, GPT4_TURBO, GPT4_TURBO_PREVIEW, GPT4_TURBO_01_25, GPT4_TURBO_11_06, GPT4, GPT4_32k, GPT4_06_13, GPT4_32k_06_13, WEN_XIN, WEN_XIN_4, XUNFEI, GEMINI, ZHIPU_AI, MOONSHOT, - "claude", "claude-3-haiku", "claude-3-sonnet", "claude-3-opus", "claude-3-opus-20240229", + "claude", "claude-3-haiku", "claude-3-sonnet", "claude-3-opus", "claude-3-opus-20240229", "claude-3.5-sonnet", "moonshot-v1-8k", "moonshot-v1-32k", "moonshot-v1-128k", QWEN, QWEN_TURBO, QWEN_PLUS, QWEN_MAX, MiniMax,