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

Add Google Bard

Browse Source
This commit is contained in:
aki66938
2023-12-14 09:46:04 +08:00
parent 8f5a0d9d61
commit c653cda0ff
6 changed files with 49 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
robot.py
View File

@@ -41,6 +41,8 @@ class Robot(Job):
self.chat = XinghuoWeb(self.config.XINGHUO_WEB)
elif chat_type == ChatType.CHATGLM.value and ChatGLM.value_check(self.config.CHATGLM):
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)
else:
self.LOG.warning("未配置模型")
self.chat = None
@@ -53,6 +55,8 @@ class Robot(Job):
self.chat = XinghuoWeb(self.config.XINGHUO_WEB)
elif ChatGLM.value_check(self.config.CHATGLM):
self.chat = ChatGLM(self.config.CHATGLM)
elif BardAssistant.value_check(self.config.BardAssistant):
self.chat = BardAssistant(self.config.BardAssistant)
else:
self.LOG.warning("未配置模型")
self.chat = None