fix use zhipu model error

2024-09-11 01:18:04 +08:00
parent 4c7afd6adb
commit 40be945fab
2 changed files with 3 additions and 3 deletions
--- a/constants.py
+++ b/constants.py
@@ -15,7 +15,7 @@ class ChatType(IntEnum):
    def is_in_chat_types(chat_type: int) -> bool:
        if chat_type in [ChatType.TIGER_BOT.value, ChatType.CHATGPT.value,
                         ChatType.XINGHUO_WEB.value, ChatType.CHATGLM.value,
-                         ChatType.BardAssistant.value]:
+                         ChatType.BardAssistant.value, ChatType.ZhiPu.value]:
            return True
        return False