liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix use zhipu model error

Browse Source
This commit is contained in:
chjs
2024-09-11 01:18:04 +08:00
parent 4c7afd6adb
commit 40be945fab
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
constants.py
View File

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

4
robot.py
View File

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