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

fix: dictionary traversal

Browse Source
This commit is contained in:
weiensong
2023-11-17 16:01:31 +08:00
parent d3f7152710
commit 744e350c4a
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
robot.py
View File

@@ -32,27 +32,27 @@ class Robot(Job):
self.allContacts = self.getAllContacts() self.allContacts = self.getAllContacts()
if ChatType.is_in_chat_types(chat_type): if ChatType.is_in_chat_types(chat_type):
if chat_type == ChatType.TIGER_BOT.value and self.value_check(self.config.TIGERBOT.values()): if chat_type == ChatType.TIGER_BOT.value and self.value_check(self.config.TIGERBOT):
self.chat = TigerBot(self.config.TIGERBOT) self.chat = TigerBot(self.config.TIGERBOT)
elif chat_type == ChatType.CHATGPT.value and self.value_check(self.config.CHATGPT.values()): elif chat_type == ChatType.CHATGPT.value and self.value_check(self.config.CHATGPT):
cgpt = self.config.CHATGPT cgpt = self.config.CHATGPT
self.chat = ChatGPT(cgpt.get("key"), cgpt.get("api"), cgpt.get("proxy"), cgpt.get("prompt")) self.chat = ChatGPT(cgpt.get("key"), cgpt.get("api"), cgpt.get("proxy"), cgpt.get("prompt"))
elif chat_type == ChatType.XINGHUO_WEB.value and self.value_check(self.config.XINGHUO_WEB.values()): elif chat_type == ChatType.XINGHUO_WEB.value and self.value_check(self.config.XINGHUO_WEB):
self.chat = XinghuoWeb(self.config.XINGHUO_WEB) self.chat = XinghuoWeb(self.config.XINGHUO_WEB)
elif chat_type == ChatType.CHATGLM.value and self.value_check(self.config.CHATGLM.values()): elif chat_type == ChatType.CHATGLM.value and self.value_check(self.config.CHATGLM):
self.chat = ChatGLM(self.config.CHATGLM) self.chat = ChatGLM(self.config.CHATGLM)
else: else:
self.LOG.warning('未配置模型') self.LOG.warning('未配置模型')
self.chat = None self.chat = None
else: else:
if self.value_check(self.config.TIGERBOT.values()): if self.value_check(self.config.TIGERBOT):
self.chat = TigerBot(self.config.TIGERBOT) self.chat = TigerBot(self.config.TIGERBOT)
elif self.value_check(self.config.CHATGPT.values()): elif self.value_check(self.config.CHATGPT):
cgpt = self.config.CHATGPT cgpt = self.config.CHATGPT
self.chat = ChatGPT(cgpt.get("key"), cgpt.get("api"), cgpt.get("proxy"), cgpt.get("prompt")) self.chat = ChatGPT(cgpt.get("key"), cgpt.get("api"), cgpt.get("proxy"), cgpt.get("prompt"))
elif self.value_check(self.config.XINGHUO_WEB.values()): elif self.value_check(self.config.XINGHUO_WEB):
self.chat = XinghuoWeb(self.config.XINGHUO_WEB) self.chat = XinghuoWeb(self.config.XINGHUO_WEB)
elif self.value_check(self.config.CHATGLM.values()): elif self.value_check(self.config.CHATGLM):
self.chat = ChatGLM(self.config.CHATGLM) self.chat = ChatGLM(self.config.CHATGLM)
else: else:
self.LOG.warning('未配置模型') self.LOG.warning('未配置模型')