diff --git a/constants.py b/constants.py index a1f87f1..8a909cc 100644 --- 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 diff --git a/robot.py b/robot.py index f9c8712..faaac39 100644 --- a/robot.py +++ b/robot.py @@ -46,8 +46,8 @@ class Robot(Job): self.chat = ChatGLM(self.config.CHATGLM) elif chat_type == ChatType.BardAssistant.value and BardAssistant.value_check(self.config.BardAssistant): self.chat = BardAssistant(self.config.BardAssistant) - elif chat_type == ChatType.ZhiPu.value and ZhiPu.value_check(self.config.ZHIPU): - self.chat = ZhiPu(self.config.ZHIPU) + elif chat_type == ChatType.ZhiPu.value and ZhiPu.value_check(self.config.ZhiPu): + self.chat = ZhiPu(self.config.ZhiPu) else: self.LOG.warning("未配置模型") self.chat = None