mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2025-04-12 23:29:37 +08:00
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
from typing import List
|
|
|
|
import httpx
|
|
from nonebot import on_command
|
|
from nonebot.adapters.onebot.v11 import Message, MessageEvent
|
|
from nonebot.adapters.onebot.v11.helpers import Numbers
|
|
from nonebot.params import CommandArg
|
|
from nonebot.plugin import PluginMetadata
|
|
from nonebot.typing import T_State
|
|
|
|
__plugin_meta__ = PluginMetadata(
|
|
name="对对联",
|
|
description="人工智能和你对对联",
|
|
usage="对联<内容>(数量)",
|
|
extra={
|
|
'type': '娱乐',
|
|
"author": "惜月 <277073121@qq.com>",
|
|
"version": "1.0.0",
|
|
},
|
|
)
|
|
|
|
couplets = on_command('对联', aliases={'对对联'}, priority=13, block=True, state={
|
|
'pm_usage': '对联<内容>(数量)',
|
|
'pm_describe': '人工智能和你对对联',
|
|
'pm_priority': 15
|
|
})
|
|
|
|
|
|
@couplets.handle()
|
|
async def _(event: MessageEvent, state: T_State, msg: Message = CommandArg(), num: List[float] = Numbers()):
|
|
msg = msg.extract_plain_text().strip()
|
|
if msg:
|
|
for n in num:
|
|
msg = msg.replace(str(int(n)), '')
|
|
if msg:
|
|
state['text'] = msg
|
|
if num:
|
|
state['num'] = int(num[0]) if num[0] <= 10 else 10
|
|
else:
|
|
state['num'] = 1
|
|
|
|
|
|
@couplets.got('text', prompt='请输入上联内容')
|
|
async def couplets_handler(event: MessageEvent, state: T_State):
|
|
url = f'https://seq2seq-couplet-model.rssbrain.com/v0.2/couplet/{state["text"]}'
|
|
async with httpx.AsyncClient() as client:
|
|
res = await client.get(url=url)
|
|
res = res.json()
|
|
result = '\n'.join(['➤' + res['output'][n] for n in range(state['num'])])
|
|
await couplets.finish(f'上联：\n➤{state["text"]}\n下联：\n{result}')
|