mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-12-18 17:35:46 +08:00
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import asyncio
|
|
|
|
from none import (
|
|
on_command, CommandSession,
|
|
on_natural_language, NLPSession, NLPResult
|
|
)
|
|
|
|
|
|
@on_command('tuling', aliases=('聊天', '对话'))
|
|
async def tuling(session: CommandSession):
|
|
message = session.get('message', prompt='我已经准备好啦，来跟我聊天吧~')
|
|
|
|
finish = message in ('结束', '拜拜', '再见')
|
|
if finish:
|
|
asyncio.ensure_future(session.send('拜拜啦，你忙吧，下次想聊天随时找我哦~'))
|
|
return
|
|
|
|
# call tuling api
|
|
reply = f'你说了：{message}'
|
|
|
|
one_time = session.get_optional('one_time', False)
|
|
if one_time:
|
|
asyncio.ensure_future(session.send(reply))
|
|
else:
|
|
del session.args['message']
|
|
session.get('message', prompt=reply)
|
|
|
|
|
|
@tuling.args_parser
|
|
async def _(session: CommandSession):
|
|
if session.current_key == 'message':
|
|
session.args[session.current_key] = session.current_arg_text.strip()
|
|
|
|
|
|
@on_natural_language
|
|
async def _(session: NLPSession):
|
|
return NLPResult(60.0, 'tuling', {
|
|
'message': session.msg,
|
|
'one_time': True
|
|
})
|