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] 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 b7b3ebf8..a6021596 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