mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
284 lines
9.8 KiB
Python
284 lines
9.8 KiB
Python
import asyncio
|
||
from pathlib import Path
|
||
import random
|
||
|
||
from httpx import HTTPStatusError
|
||
from nonebot import on_message
|
||
from nonebot.adapters import Bot, Event
|
||
from nonebot.plugin import PluginMetadata
|
||
from nonebot_plugin_alconna import UniMsg, Voice
|
||
from nonebot_plugin_uninfo import Uninfo
|
||
|
||
from zhenxun.configs.config import BotConfig
|
||
from zhenxun.configs.path_config import IMAGE_PATH
|
||
from zhenxun.configs.utils import (
|
||
AICallableParam,
|
||
AICallableProperties,
|
||
AICallableTag,
|
||
PluginExtraData,
|
||
RegisterConfig,
|
||
)
|
||
from zhenxun.services.log import logger
|
||
from zhenxun.services.plugin_init import PluginInit
|
||
from zhenxun.utils.depends import CheckConfig, UserName
|
||
from zhenxun.utils.message import MessageUtils
|
||
|
||
from .bym_gift import ICON_PATH
|
||
from .bym_gift.data_source import send_gift
|
||
from .bym_gift.gift_reg import driver
|
||
from .config import Arparma, FunctionParam
|
||
from .data_source import ChatManager, base_config, split_text
|
||
from .exception import GiftRepeatSendException, NotResultException
|
||
from .goods_register import driver # noqa: F401
|
||
from .models.bym_chat import BymChat
|
||
|
||
__plugin_meta__ = PluginMetadata(
|
||
name="BYM_AI",
|
||
description=f"{BotConfig.self_nickname}想成为人类...",
|
||
usage=f"""
|
||
你问小真寻的愿望？
|
||
{BotConfig.self_nickname}说她想成为人类！
|
||
""".strip(),
|
||
extra=PluginExtraData(
|
||
author="Chtholly & HibiKier",
|
||
version="0.3",
|
||
ignore_prompt=True,
|
||
configs=[
|
||
RegisterConfig(
|
||
key="BYM_AI_CHAT_URL",
|
||
value=None,
|
||
help="ai聊天接口地址，可以填入url和平台名称，当你使用平台名称时，默认使用平台官方api, 目前有[gemini, DeepSeek, 硅基流动, 阿里云百炼, 百度智能云, 字节火山引擎], 填入对应名称即可, 如 gemini",
|
||
),
|
||
RegisterConfig(
|
||
key="BYM_AI_CHAT_TOKEN",
|
||
value=None,
|
||
help="ai聊天接口密钥，使用列表",
|
||
type=list[str],
|
||
),
|
||
RegisterConfig(
|
||
key="BYM_AI_CHAT_MODEL",
|
||
value=None,
|
||
help="ai聊天接口模型",
|
||
),
|
||
RegisterConfig(
|
||
key="BYM_AI_TOOL_MODEL",
|
||
value=None,
|
||
help="ai工具接口模型",
|
||
),
|
||
RegisterConfig(
|
||
key="BYM_AI_CHAT",
|
||
value=True,
|
||
help="是否开启伪人回复",
|
||
default_value=True,
|
||
type=bool,
|
||
),
|
||
RegisterConfig(
|
||
key="BYM_AI_CHAT_RATE",
|
||
value=0.05,
|
||
help="伪人回复概率 0-1",
|
||
default_value=0.05,
|
||
type=float,
|
||
),
|
||
RegisterConfig(
|
||
key="BYM_AI_CHAT_SMART",
|
||
value=False,
|
||
help="是否开启智能模式",
|
||
default_value=False,
|
||
type=bool,
|
||
),
|
||
RegisterConfig(
|
||
key="BYM_AI_TTS_URL",
|
||
value=None,
|
||
help="tts接口地址",
|
||
),
|
||
RegisterConfig(
|
||
key="BYM_AI_TTS_TOKEN",
|
||
value=None,
|
||
help="tts接口密钥",
|
||
),
|
||
RegisterConfig(
|
||
key="BYM_AI_TTS_VOICE",
|
||
value=None,
|
||
help="tts接口音色",
|
||
),
|
||
RegisterConfig(
|
||
key="ENABLE_IMPRESSION",
|
||
value=True,
|
||
help="使用签到数据作为基础好感度",
|
||
default_value=True,
|
||
type=bool,
|
||
),
|
||
RegisterConfig(
|
||
key="GROUP_CACHE_SIZE",
|
||
value=40,
|
||
help="群组内聊天记录数据大小",
|
||
default_value=40,
|
||
type=int,
|
||
),
|
||
RegisterConfig(
|
||
key="CACHE_SIZE",
|
||
value=40,
|
||
help="私聊下缓存聊天记录数据大小（每位用户）",
|
||
default_value=40,
|
||
type=int,
|
||
),
|
||
RegisterConfig(
|
||
key="ENABLE_GROUP_CHAT",
|
||
value=True,
|
||
help="在群组中时共用缓存",
|
||
default_value=True,
|
||
type=bool,
|
||
),
|
||
],
|
||
smart_tools=[
|
||
AICallableTag(
|
||
name="call_send_gift",
|
||
description="想给某人送礼物时，调用此方法，并且将返回值发送",
|
||
parameters=AICallableParam(
|
||
type="object",
|
||
properties={
|
||
"user_id": AICallableProperties(
|
||
type="string", description="用户的id"
|
||
),
|
||
},
|
||
required=["user_id"],
|
||
),
|
||
func=send_gift,
|
||
)
|
||
],
|
||
).to_dict(),
|
||
)
|
||
|
||
|
||
async def rule(event: Event, session: Uninfo) -> bool:
|
||
if event.is_tome():
|
||
"""at自身必定回复"""
|
||
return True
|
||
if not base_config.get("BYM_AI_CHAT"):
|
||
return False
|
||
if event.is_tome() and not session.group:
|
||
"""私聊过滤"""
|
||
return False
|
||
rate = base_config.get("BYM_AI_CHAT_RATE") or 0
|
||
return random.random() <= rate
|
||
|
||
|
||
_matcher = on_message(priority=998, rule=rule)
|
||
|
||
|
||
@_matcher.handle(parameterless=[CheckConfig(config="BYM_AI_CHAT_TOKEN")])
|
||
async def _(
|
||
bot: Bot,
|
||
event: Event,
|
||
message: UniMsg,
|
||
session: Uninfo,
|
||
uname: str = UserName(),
|
||
):
|
||
if not message.extract_plain_text().strip():
|
||
if event.is_tome():
|
||
await MessageUtils.build_message(ChatManager.hello()).finish()
|
||
return
|
||
fun_param = FunctionParam(
|
||
bot=bot,
|
||
event=event,
|
||
arparma=Arparma(head_result="BYM_AI"),
|
||
session=session,
|
||
message=message,
|
||
)
|
||
group_id = session.group.id if session.group else None
|
||
is_bym = not event.is_tome()
|
||
try:
|
||
try:
|
||
result = await ChatManager.get_result(
|
||
bot, session, group_id, uname, message, is_bym, fun_param
|
||
)
|
||
except HTTPStatusError as e:
|
||
logger.error("BYM AI 请求失败", "BYM_AI", session=session, e=e)
|
||
return await MessageUtils.build_message(
|
||
f"请求失败了哦，code: {e.response.status_code}"
|
||
).send(reply_to=True)
|
||
except NotResultException:
|
||
return await MessageUtils.build_message("请求没有结果呢...").send(
|
||
reply_to=True
|
||
)
|
||
if is_bym:
|
||
"""伪人回复，切割文本"""
|
||
if result:
|
||
for r, delay in split_text(result):
|
||
await MessageUtils.build_message(r).send()
|
||
await asyncio.sleep(delay)
|
||
else:
|
||
try:
|
||
if result:
|
||
await MessageUtils.build_message(result).send(
|
||
reply_to=bool(group_id)
|
||
)
|
||
if tts_data := await ChatManager.tts(result):
|
||
await MessageUtils.build_message(Voice(raw=tts_data)).send()
|
||
elif not base_config.get("BYM_AI_CHAT_SMART"):
|
||
await MessageUtils.build_message(ChatManager.no_result()).send()
|
||
else:
|
||
await MessageUtils.build_message(
|
||
f"{BotConfig.self_nickname}并不想理你..."
|
||
).send(reply_to=True)
|
||
if (
|
||
event.is_tome()
|
||
and result
|
||
and (plain_text := message.extract_plain_text())
|
||
):
|
||
await BymChat.create(
|
||
user_id=session.user.id,
|
||
group_id=group_id,
|
||
plain_text=plain_text,
|
||
result=result,
|
||
)
|
||
logger.info(
|
||
f"BYM AI 问题: {message} | 回答: {result}",
|
||
"BYM_AI",
|
||
session=session,
|
||
)
|
||
except HTTPStatusError as e:
|
||
logger.error("BYM AI 请求失败", "BYM_AI", session=session, e=e)
|
||
await MessageUtils.build_message(
|
||
f"请求失败了哦，code: {e.response.status_code}"
|
||
).send(reply_to=True)
|
||
except NotResultException:
|
||
await MessageUtils.build_message("请求没有结果呢...").send(
|
||
reply_to=True
|
||
)
|
||
except GiftRepeatSendException:
|
||
logger.warning("BYM AI 重复发送礼物", "BYM_AI", session=session)
|
||
await MessageUtils.build_message(
|
||
f"今天已经收过{BotConfig.self_nickname}的礼物了哦~"
|
||
).finish(reply_to=True)
|
||
except Exception as e:
|
||
logger.error("BYM AI 其他错误", "BYM_AI", session=session, e=e)
|
||
await MessageUtils.build_message("发生了一些异常，想要休息一下...").finish(
|
||
reply_to=True
|
||
)
|
||
|
||
|
||
RESOURCE_FILES = [
|
||
IMAGE_PATH / "shop_icon" / "reload_ai_card.png",
|
||
IMAGE_PATH / "shop_icon" / "reload_ai_card1.png",
|
||
]
|
||
|
||
GIFT_FILES = [ICON_PATH / "wallet.png", ICON_PATH / "hairpin.png"]
|
||
|
||
|
||
class MyPluginInit(PluginInit):
|
||
async def install(self):
|
||
for res_file in RESOURCE_FILES + GIFT_FILES:
|
||
res = Path(__file__).parent / res_file.name
|
||
if res.exists():
|
||
if res_file.exists():
|
||
res_file.unlink()
|
||
res.rename(res_file)
|
||
logger.info(f"更新 BYM_AI 资源文件成功 {res} -> {res_file}")
|
||
|
||
async def remove(self):
|
||
for res_file in RESOURCE_FILES + GIFT_FILES:
|
||
if res_file.exists():
|
||
res_file.unlink()
|
||
logger.info(f"删除 BYM_AI 资源文件成功 {res_file}")
|