diff --git a/bot/xunfei/xunfei_spark_bot.py b/bot/xunfei/xunfei_spark_bot.py index e9065e1..30cab2f 100644 --- a/bot/xunfei/xunfei_spark_bot.py +++ b/bot/xunfei/xunfei_spark_bot.py @@ -3,7 +3,7 @@ import requests, json from bot.bot import Bot from bot.session_manager import SessionManager -from bot.baidu.baidu_wenxin_session import BaiduWenxinSession +from bot.chatgpt.chat_gpt_session import ChatGPTSession from bridge.context import ContextType, Context from bridge.reply import Reply, ReplyType from common.log import logger @@ -53,7 +53,7 @@ class XunFeiBot(Bot): self.host = urlparse(self.spark_url).netloc self.path = urlparse(self.spark_url).path # 和wenxin使用相同的session机制 - self.sessions = SessionManager(BaiduWenxinSession, model=const.XUNFEI) + self.sessions = SessionManager(ChatGPTSession, model=const.XUNFEI) def reply(self, query, context: Context = None) -> Reply: if context.type == ContextType.TEXT: