mirror of https://github.com/Mai-with-u/MaiBot.git
153 lines
6.3 KiB
Python
153 lines
6.3 KiB
Python
import random
|
||
from typing import Tuple
|
||
|
||
# 导入新插件系统
|
||
from src.plugin_system import BaseAction, ActionActivationType, ChatMode
|
||
|
||
# 导入依赖的系统组件
|
||
from src.common.logger import get_logger
|
||
|
||
# 导入API模块 - 标准Python包方式
|
||
from src.plugin_system.apis import emoji_api, llm_api, message_api
|
||
# NoReplyAction已集成到heartFC_chat.py中，不再需要导入
|
||
from src.config.config import global_config
|
||
|
||
|
||
logger = get_logger("emoji")
|
||
|
||
|
||
class EmojiAction(BaseAction):
|
||
"""表情动作 - 发送表情包"""
|
||
|
||
activation_type = ActionActivationType.RANDOM
|
||
random_activation_probability = global_config.emoji.emoji_chance
|
||
mode_enable = ChatMode.ALL
|
||
parallel_action = True
|
||
|
||
# 动作基本信息
|
||
action_name = "emoji"
|
||
action_description = "发送表情包辅助表达情绪"
|
||
|
||
# LLM判断提示词
|
||
llm_judge_prompt = """
|
||
判定是否需要使用表情动作的条件：
|
||
1. 用户明确要求使用表情包
|
||
2. 这是一个适合表达强烈情绪的场合
|
||
3. 不要发送太多表情包，如果你已经发送过多个表情包则回答"否"
|
||
|
||
请回答"是"或"否"。
|
||
"""
|
||
|
||
# 动作参数定义
|
||
action_parameters = {}
|
||
|
||
# 动作使用场景
|
||
action_require = [
|
||
"发送表情包辅助表达情绪",
|
||
"表达情绪时可以选择使用",
|
||
"不要连续发送，如果你已经发过[表情包]，就不要选择此动作",
|
||
]
|
||
|
||
# 关联类型
|
||
associated_types = ["emoji"]
|
||
|
||
async def execute(self) -> Tuple[bool, str]:
|
||
# sourcery skip: assign-if-exp, introduce-default-else, swap-if-else-branches, use-named-expression
|
||
"""执行表情动作"""
|
||
logger.info(f"{self.log_prefix} 决定发送表情")
|
||
|
||
try:
|
||
# 1. 获取发送表情的原因
|
||
reason = self.action_data.get("reason", "表达当前情绪")
|
||
logger.info(f"{self.log_prefix} 发送表情原因: {reason}")
|
||
|
||
# 2. 随机获取20个表情包
|
||
sampled_emojis = await emoji_api.get_random(30)
|
||
if not sampled_emojis:
|
||
logger.warning(f"{self.log_prefix} 无法获取随机表情包")
|
||
return False, "无法获取随机表情包"
|
||
|
||
# 3. 准备情感数据
|
||
emotion_map = {}
|
||
for b64, desc, emo in sampled_emojis:
|
||
if emo not in emotion_map:
|
||
emotion_map[emo] = []
|
||
emotion_map[emo].append((b64, desc))
|
||
|
||
available_emotions = list(emotion_map.keys())
|
||
|
||
if not available_emotions:
|
||
logger.warning(f"{self.log_prefix} 获取到的表情包均无情感标签, 将随机发送")
|
||
emoji_base64, emoji_description, _ = random.choice(sampled_emojis)
|
||
else:
|
||
# 获取最近的5条消息内容用于判断
|
||
recent_messages = message_api.get_recent_messages(chat_id=self.chat_id, limit=5)
|
||
messages_text = ""
|
||
if recent_messages:
|
||
# 使用message_api构建可读的消息字符串
|
||
messages_text = message_api.build_readable_messages(
|
||
messages=recent_messages,
|
||
timestamp_mode="normal_no_YMD",
|
||
truncate=False,
|
||
show_actions=False,
|
||
)
|
||
|
||
# 4. 构建prompt让LLM选择情感
|
||
prompt = f"""
|
||
你是一个正在进行聊天的网友，你需要根据一个理由和最近的聊天记录，从一个情感标签列表中选择最匹配的一个。
|
||
这是最近的聊天记录：
|
||
{messages_text}
|
||
|
||
这是理由：“{reason}”
|
||
这里是可用的情感标签：{available_emotions}
|
||
请直接返回最匹配的那个情感标签，不要进行任何解释或添加其他多余的文字。
|
||
"""
|
||
|
||
if global_config.debug.show_prompt:
|
||
logger.info(f"{self.log_prefix} 生成的LLM Prompt: {prompt}")
|
||
else:
|
||
logger.debug(f"{self.log_prefix} 生成的LLM Prompt: {prompt}")
|
||
|
||
# 5. 调用LLM
|
||
models = llm_api.get_available_models()
|
||
chat_model_config = models.get("utils_small") # 使用字典访问方式
|
||
if not chat_model_config:
|
||
logger.error(f"{self.log_prefix} 未找到'utils_small'模型配置，无法调用LLM")
|
||
return False, "未找到'utils_small'模型配置"
|
||
|
||
success, chosen_emotion, _, _ = await llm_api.generate_with_model(
|
||
prompt, model_config=chat_model_config, request_type="emoji"
|
||
)
|
||
|
||
if not success:
|
||
logger.error(f"{self.log_prefix} LLM调用失败: {chosen_emotion}")
|
||
return False, f"LLM调用失败: {chosen_emotion}"
|
||
|
||
chosen_emotion = chosen_emotion.strip().replace('"', "").replace("'", "")
|
||
logger.info(f"{self.log_prefix} LLM选择的情感: {chosen_emotion}")
|
||
|
||
# 6. 根据选择的情感匹配表情包
|
||
if chosen_emotion in emotion_map:
|
||
emoji_base64, emoji_description = random.choice(emotion_map[chosen_emotion])
|
||
logger.info(f"{self.log_prefix} 找到匹配情感 '{chosen_emotion}' 的表情包: {emoji_description}")
|
||
else:
|
||
logger.warning(
|
||
f"{self.log_prefix} LLM选择的情感 '{chosen_emotion}' 不在可用列表中, 将随机选择一个表情包"
|
||
)
|
||
emoji_base64, emoji_description, _ = random.choice(sampled_emojis)
|
||
|
||
# 7. 发送表情包
|
||
success = await self.send_emoji(emoji_base64)
|
||
|
||
if not success:
|
||
logger.error(f"{self.log_prefix} 表情包发送失败")
|
||
return False, "表情包发送失败"
|
||
|
||
# no_reply计数器现在由heartFC_chat.py统一管理，无需在此重置
|
||
|
||
return True, f"发送表情包: {emoji_description}"
|
||
|
||
except Exception as e:
|
||
logger.error(f"{self.log_prefix} 表情动作执行失败: {e}", exc_info=True)
|
||
return False, f"表情发送失败: {str(e)}"
|