优化autogen的使用

2023-10-23 01:56:18 +08:00
parent e2b3c47186
commit 27c4e3ef4f
9 changed files with 108 additions and 54 deletions
--- a/request_llm/bridge_all.py
+++ b/request_llm/bridge_all.py
@@ -182,6 +182,15 @@ model_info = {
        "token_cnt": get_token_num_gpt4,
    },

+    "api2d-gpt-3.5-turbo-16k": {
+        "fn_with_ui": chatgpt_ui,
+        "fn_without_ui": chatgpt_noui,
+        "endpoint": api2d_endpoint,
+        "max_token": 1024*16,
+        "tokenizer": tokenizer_gpt35,
+        "token_cnt": get_token_num_gpt35,
+    },
+
    # 将 chatglm 直接对齐到 chatglm2
    "chatglm": {
        "fn_with_ui": chatglm_ui,