From 37ad77a6f7966d0394a43134ab1339533dc16ecf Mon Sep 17 00:00:00 2001
From: zzj1804 <mehjz@foxmail.com>
Date: Thu, 2 Dec 2021 14:38:48 +0800
Subject: =?UTF-8?q?=E6=96=B0=E5=A2=9Echatbot=E6=8F=92=E4=BB=B6,=E4=BB=8E?=
 =?UTF-8?q?=E7=BE=A4=E8=81=8A=E5=A4=A9=E5=AD=A6=E4=B9=A0,=E5=B9=B6?=
 =?UTF-8?q?=E9=9A=8F=E6=9C=BA=E5=9B=9E=E8=AF=9D=20=E4=BF=AE=E6=94=B9chat?=
 =?UTF-8?q?=E6=8F=92=E4=BB=B6,=E6=97=A0=E6=B3=95=E5=9B=9E=E8=AF=9D?=
 =?UTF-8?q?=E6=97=B6=E5=B0=9D=E8=AF=95chatterbot?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/plugins/chat/__init__.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'ATRI/plugins/chat/__init__.py')

diff --git a/ATRI/plugins/chat/__init__.py b/ATRI/plugins/chat/__init__.py
index ab22db9..7436645 100644
--- a/ATRI/plugins/chat/__init__.py
+++ b/ATRI/plugins/chat/__init__.py
@@ -7,7 +7,7 @@ from ATRI.utils import CoolqCodeChecker
 from ATRI.utils.limit import FreqLimiter
 from ATRI.utils.apscheduler import scheduler
 from .data_source import Chat
-
+from ATRI.plugins.atri_chat_bot import ATRIChatBot
 
 _chat_flmt = FreqLimiter(3)
 _chat_flmt_notice = choice(["慢...慢一..点❤", "冷静1下", "歇会歇会~~", "我开始为你以后的伴侣担心了..."])
@@ -27,7 +27,10 @@ async def _chat(bot: Bot, event: MessageEvent):
     repo = await Chat().deal(msg, user_id)
     _chat_flmt.start_cd(user_id)
     try:
-        await chat.finish(repo)
+        if repo:
+            await chat.finish(repo)
+        else:  # 实在没话说就尝试 chatterbot
+            await chat.finish(await ATRIChatBot.get_response(msg))
     except Exception:
         return
 
-- 
cgit v1.2.3