From 6d267947bba707706868c1c6aae1fcaa3d222485 Mon Sep 17 00:00:00 2001
From: Rid7 <lizongxi1@live.com>
Date: Mon, 15 May 2023 15:12:50 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=9E=E7=8E=B0Claude=E8=81=8A=E5=A4=A9?=
 =?UTF-8?q?=E5=8A=9F=E8=83=BD=E9=85=8D=E7=BD=AE=E9=A1=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 config.py                 |  7 ++++++-
 request_llm/bridge_all.py | 12 ++++++++++++
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/config.py b/config.py
index 2617aff..99b72d9 100644
--- a/config.py
+++ b/config.py
@@ -46,7 +46,7 @@ MAX_RETRY = 2
 
 # OpenAI模型选择是（gpt4现在只对申请成功的人开放，体验gpt-4可以试试api2d）
 LLM_MODEL = "gpt-3.5-turbo" # 可选 ↓↓↓
-AVAIL_LLM_MODELS = ["gpt-3.5-turbo", "api2d-gpt-3.5-turbo", "gpt-4", "api2d-gpt-4", "chatglm", "moss", "newbing"]
+AVAIL_LLM_MODELS = ["gpt-3.5-turbo", "api2d-gpt-3.5-turbo", "gpt-4", "api2d-gpt-4", "chatglm", "moss", "newbing", "claude"]
 
 # 本地LLM模型如ChatGLM的执行方式 CPU/GPU
 LOCAL_MODEL_DEVICE = "cpu" # 可选 "cuda"
@@ -75,3 +75,8 @@ NEWBING_STYLE = "creative"  # ["creative", "balanced", "precise"]
 NEWBING_COOKIES = """
 your bing cookies here
 """
+
+# slack-claude bot
+# 下面的id怎么填写具体参见https://zhuanlan.zhihu.com/p/627485689
+CLAUDE_BOT_ID = '' 
+SLACK_USER_TOKEN = ''
\ No newline at end of file
diff --git a/request_llm/bridge_all.py b/request_llm/bridge_all.py
index 9dbcf79..5560592 100644
--- a/request_llm/bridge_all.py
+++ b/request_llm/bridge_all.py
@@ -22,6 +22,9 @@ from .bridge_chatglm import predict as chatglm_ui
 from .bridge_newbing import predict_no_ui_long_connection as newbing_noui
 from .bridge_newbing import predict as newbing_ui
 
+from .bridge_claude import predict_no_ui_long_connection as claude_noui
+from .bridge_claude import predict as claude_ui
+
 # from .bridge_tgui import predict_no_ui_long_connection as tgui_noui
 # from .bridge_tgui import predict as tgui_ui
 
@@ -130,6 +133,15 @@ model_info = {
         "tokenizer": tokenizer_gpt35,
         "token_cnt": get_token_num_gpt35,
     },
+    # claude
+    "claude": {
+        "fn_with_ui": claude_ui,
+        "fn_without_ui": claude_noui,
+        "endpoint": None,
+        "max_token": 4096,
+        "tokenizer": tokenizer_gpt35,
+        "token_cnt": get_token_num_gpt35,
+    },
 }