feat: support chatglm_turbo model #1443 (#1460)

2023-11-06 18:33:05 +08:00
parent a9ee18300e
commit 4dfbcd0b4e
3 changed files with 12 additions and 1 deletions
--- a/api/core/model_providers/providers/zhipuai_provider.py
+++ b/api/core/model_providers/providers/zhipuai_provider.py
@@ -26,6 +26,11 @@ class ZhipuAIProvider(BaseModelProvider):
    def _get_fixed_model_list(self, model_type: ModelType) -> list[dict]:
        if model_type == ModelType.TEXT_GENERATION:
            return [
+                {
+                    'id': 'chatglm_turbo',
+                    'name': 'chatglm_turbo',
+                    'mode': ModelMode.CHAT.value,
+                },
                {
                    'id': 'chatglm_pro',
                    'name': 'chatglm_pro',