From 744e350c4a0b4e14b25d8efc8cbce8c66c4e3902 Mon Sep 17 00:00:00 2001 From: weiensong Date: Fri, 17 Nov 2023 16:01:31 +0800 Subject: [PATCH] fix: dictionary traversal --- robot.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/robot.py b/robot.py index 9e0b29a..ba2cf2e 100644 --- a/robot.py +++ b/robot.py @@ -32,27 +32,27 @@ class Robot(Job): self.allContacts = self.getAllContacts() 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) - 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 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) - 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) else: self.LOG.warning('未配置模型') self.chat = None else: - if self.value_check(self.config.TIGERBOT.values()): + if self.value_check(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 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) - elif self.value_check(self.config.CHATGLM.values()): + elif self.value_check(self.config.CHATGLM): self.chat = ChatGLM(self.config.CHATGLM) else: self.LOG.warning('未配置模型')