From 40be945fab918ddbf2d053caf3577c0655c3bae9 Mon Sep 17 00:00:00 2001
From: chjs <461646668@qq.com>
Date: Wed, 11 Sep 2024 01:18:04 +0800
Subject: [PATCH] fix use zhipu model error

---
 constants.py | 2 +-
 robot.py     | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

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