Merge pull request #2288 from KuroIVeko/patch-3

Support more models from Zhipu AI
This commit is contained in:
vision
2024-09-25 22:28:16 +08:00
committed by GitHub
2 changed files with 13 additions and 2 deletions

View File

@@ -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