mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
from typing import List
|
|
|
|
from nonebot import on_message
|
|
from nonebot.plugin import PluginMetadata
|
|
from nonebot.rule import to_me
|
|
from nonebot_plugin_alconna import UniMsg
|
|
from nonebot_plugin_session import EventSession
|
|
|
|
from zhenxun.configs.config import BotConfig, Config
|
|
from zhenxun.configs.utils import PluginExtraData, RegisterConfig
|
|
from zhenxun.models.friend_user import FriendUser
|
|
from zhenxun.models.group_member_info import GroupInfoUser
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.depends import UserName
|
|
from zhenxun.utils.message import MessageUtils
|
|
|
|
from .data_source import get_chat_result, hello, no_result
|
|
|
|
__plugin_meta__ = PluginMetadata(
|
|
name="AI",
|
|
description="屑Ai",
|
|
usage=f"""
|
|
与{BotConfig.self_nickname}普普通通的对话吧！
|
|
""".strip(),
|
|
extra=PluginExtraData(
|
|
author="HibiKier",
|
|
version="0.1",
|
|
configs=[
|
|
RegisterConfig(
|
|
module="alapi",
|
|
key="ALAPI_TOKEN",
|
|
value=None,
|
|
help="在 https://admin.alapi.cn/user/login 登录后获取token",
|
|
),
|
|
RegisterConfig(key="TL_KEY", value=[], help="图灵Key", type=List[str]),
|
|
RegisterConfig(
|
|
key="ALAPI_AI_CHECK",
|
|
value=False,
|
|
help="是否检测青云客骂娘回复",
|
|
default_value=False,
|
|
type=bool,
|
|
),
|
|
RegisterConfig(
|
|
key="TEXT_FILTER",
|
|
value=["鸡", "口交"],
|
|
help="文本过滤器，将敏感词更改为*",
|
|
type=List[str],
|
|
),
|
|
],
|
|
).dict(),
|
|
)
|
|
|
|
|
|
ai = on_message(rule=to_me(), priority=998)
|
|
|
|
|
|
@ai.handle()
|
|
async def _(message: UniMsg, session: EventSession, uname: str = UserName()):
|
|
if not message or message.extract_plain_text() in [
|
|
"你好啊",
|
|
"你好",
|
|
"在吗",
|
|
"在不在",
|
|
"您好",
|
|
"您好啊",
|
|
"你好",
|
|
"在",
|
|
]:
|
|
await hello().finish()
|
|
if not session.id1:
|
|
await Text("用户id不存在...").finish()
|
|
gid = session.id3 or session.id2
|
|
if gid:
|
|
nickname = await GroupInfoUser.get_user_nickname(session.id1, gid)
|
|
else:
|
|
nickname = await FriendUser.get_user_nickname(session.id1)
|
|
if not nickname:
|
|
nickname = uname
|
|
result = await get_chat_result(message, session.id1, nickname)
|
|
logger.info(f"问题：{message} ---- 回答：{result}", "ai", session=session)
|
|
if result:
|
|
result = str(result)
|
|
for t in Config.get_config("ai", "TEXT_FILTER"):
|
|
result = result.replace(t, "*")
|
|
await MessageUtils.build_message(result).finish()
|
|
else:
|
|
await no_result().finish()
|