From 6b02bae9577352517f38dbbbd3c6f1b6cdc9ea07 Mon Sep 17 00:00:00 2001 From: KurolVeko <128911459+KuroIVeko@users.noreply.github.com> Date: Thu, 5 Sep 2024 10:59:57 +0800 Subject: [PATCH 1/2] Update bridge.py --- bridge/bridge.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bridge/bridge.py b/bridge/bridge.py index b7b3ebf..a602159 100644 --- a/bridge/bridge.py +++ b/bridge/bridge.py @@ -38,7 +38,7 @@ class Bridge(object): self.btype["chat"] = const.QWEN_DASHSCOPE if model_type and model_type.startswith("gemini"): self.btype["chat"] = const.GEMINI - if model_type in [const.ZHIPU_AI]: + if model_type and model_type.startswith("glm"): self.btype["chat"] = const.ZHIPU_AI if model_type and model_type.startswith("claude-3"): self.btype["chat"] = const.CLAUDEAPI From 111ad44029e2fa6e52a36a8acfe2b52a8a420146 Mon Sep 17 00:00:00 2001 From: KurolVeko <128911459+KuroIVeko@users.noreply.github.com> Date: Thu, 5 Sep 2024 11:07:06 +0800 Subject: [PATCH 2/2] Update const.py --- common/const.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/common/const.py b/common/const.py index e2e6a00..b099030 100644 --- a/common/const.py +++ b/common/const.py @@ -57,11 +57,22 @@ GEMINI_PRO = "gemini-1.0-pro" GEMINI_15_flash = "gemini-1.5-flash" GEMINI_15_PRO = "gemini-1.5-pro" +GLM_4 = "glm-4" +GLM_4_PLUS = "glm-4-plus" +GLM_4_flash = "glm-4-flash" +GLM_4_LONG = "glm-4-long" +GLM_4_ALLTOOLS = "glm-4-alltools" +GLM_4_0520 = "glm-4-0520" +GLM_4_AIR = "glm-4-air" +GLM_4_AIRX = "glm-4-airx" + MODEL_LIST = [ GPT35, GPT35_0125, GPT35_1106, "gpt-3.5-turbo-16k", GPT_4o, GPT_4O_0806, GPT_4o_MINI, 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, ZHIPU_AI, MOONSHOT, MiniMax, + XUNFEI, + ZHIPU_AI, GLM_4, GLM_4_PLUS, GLM_4_flash, GLM_4_LONG, GLM_4_ALLTOOLS, GLM_4_0520, GLM_4_AIR, GLM_4_AIRX, + MOONSHOT, MiniMax, GEMINI, GEMINI_PRO, GEMINI_15_flash, GEMINI_15_PRO, "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",