diff options
| author | Kyomotoi <[email protected]> | 2021-12-11 23:10:00 +0800 |
|---|---|---|
| committer | GitHub <[email protected]> | 2021-12-11 23:10:00 +0800 |
| commit | 42d6bb6e4d492aac530042a9b9d1a7e477c0f51b (patch) | |
| tree | e514db6e8099af691fc496a50ccf3fd7598634b3 /ATRI/plugins/chat | |
| parent | b19cf7768a70cf639853d9d33a6ab2a3d56e7045 (diff) | |
| parent | 37ad77a6f7966d0394a43134ab1339533dc16ecf (diff) | |
| download | ATRI-42d6bb6e4d492aac530042a9b9d1a7e477c0f51b.tar.gz ATRI-42d6bb6e4d492aac530042a9b9d1a7e477c0f51b.tar.bz2 ATRI-42d6bb6e4d492aac530042a9b9d1a7e477c0f51b.zip | |
✨ 合并PR #38 from zzj1804/main
新增chatbot插件,从群聊天学习,并随机回话 修改chat插件,无法回话时尝试chatterbot
Diffstat (limited to 'ATRI/plugins/chat')
| -rw-r--r-- | ATRI/plugins/chat/__init__.py | 7 |
1 files changed, 5 insertions, 2 deletions
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 |