mirror of https://github.com/Mai-with-u/MaiBot.git
164 lines
6.4 KiB
Python
164 lines
6.4 KiB
Python
from datetime import datetime
|
||
import time
|
||
import asyncio
|
||
from typing import Dict
|
||
|
||
from src.chat.utils.chat_message_builder import (
|
||
get_raw_msg_by_timestamp_with_chat,
|
||
build_readable_messages,
|
||
)
|
||
from src.chat.utils.prompt_builder import Prompt, global_prompt_manager
|
||
from src.config.config import global_config, model_config
|
||
from src.llm_models.utils_model import LLMRequest
|
||
from src.common.logger import get_logger
|
||
from src.plugin_system.apis import frequency_api
|
||
|
||
|
||
def init_prompt():
|
||
Prompt(
|
||
"""{name_block}
|
||
{time_block}
|
||
你现在正在聊天，请根据下面的聊天记录判断是否有用户觉得你的发言过于频繁或者发言过少
|
||
{message_str}
|
||
|
||
如果用户觉得你的发言过于频繁，请输出"过于频繁"，否则输出"正常"
|
||
如果用户觉得你的发言过少，请输出"过少"，否则输出"正常"
|
||
**你只能输出以下三个词之一，不要输出任何其他文字、解释或标点：**
|
||
- 正常
|
||
- 过于频繁
|
||
- 过少
|
||
""",
|
||
"frequency_adjust_prompt",
|
||
)
|
||
|
||
|
||
logger = get_logger("frequency_control")
|
||
|
||
|
||
class FrequencyControl:
|
||
"""简化的频率控制类，仅管理不同chat_id的频率值"""
|
||
|
||
def __init__(self, chat_id: str):
|
||
self.chat_id = chat_id
|
||
# 发言频率调整值
|
||
self.talk_frequency_adjust: float = 1.0
|
||
|
||
self.last_frequency_adjust_time: float = 0.0
|
||
self.frequency_model = LLMRequest(
|
||
model_set=model_config.model_task_config.utils_small, request_type="frequency.adjust"
|
||
)
|
||
# 频率调整锁，防止并发执行
|
||
self._adjust_lock = asyncio.Lock()
|
||
|
||
def get_talk_frequency_adjust(self) -> float:
|
||
"""获取发言频率调整值"""
|
||
return self.talk_frequency_adjust
|
||
|
||
def set_talk_frequency_adjust(self, value: float) -> None:
|
||
"""设置发言频率调整值"""
|
||
self.talk_frequency_adjust = max(0.1, min(5.0, value))
|
||
|
||
async def trigger_frequency_adjust(self) -> None:
|
||
# 使用异步锁防止并发执行
|
||
async with self._adjust_lock:
|
||
# 在锁内检查，避免并发触发
|
||
current_time = time.time()
|
||
previous_adjust_time = self.last_frequency_adjust_time
|
||
|
||
msg_list = get_raw_msg_by_timestamp_with_chat(
|
||
chat_id=self.chat_id,
|
||
timestamp_start=previous_adjust_time,
|
||
timestamp_end=current_time,
|
||
)
|
||
|
||
if current_time - previous_adjust_time < 160 or len(msg_list) <= 20:
|
||
return
|
||
|
||
# 立即更新调整时间，防止并发触发
|
||
self.last_frequency_adjust_time = current_time
|
||
|
||
try:
|
||
new_msg_list = get_raw_msg_by_timestamp_with_chat(
|
||
chat_id=self.chat_id,
|
||
timestamp_start=previous_adjust_time,
|
||
timestamp_end=current_time,
|
||
limit=20,
|
||
limit_mode="latest",
|
||
)
|
||
|
||
message_str = build_readable_messages(
|
||
new_msg_list,
|
||
replace_bot_name=True,
|
||
timestamp_mode="relative",
|
||
read_mark=0.0,
|
||
show_actions=False,
|
||
)
|
||
time_block = f"当前时间：{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
|
||
bot_name = global_config.bot.nickname
|
||
bot_nickname = (
|
||
f",也有人叫你{','.join(global_config.bot.alias_names)}" if global_config.bot.alias_names else ""
|
||
)
|
||
name_block = f"你的名字是{bot_name}{bot_nickname}，请注意哪些是你自己的发言。"
|
||
|
||
prompt = await global_prompt_manager.format_prompt(
|
||
"frequency_adjust_prompt",
|
||
name_block=name_block,
|
||
time_block=time_block,
|
||
message_str=message_str,
|
||
)
|
||
response, (reasoning_content, _, _) = await self.frequency_model.generate_response_async(
|
||
prompt,
|
||
)
|
||
|
||
# logger.info(f"频率调整 prompt: {prompt}")
|
||
# logger.info(f"频率调整 response: {response}")
|
||
|
||
if global_config.debug.show_prompt:
|
||
logger.info(f"频率调整 prompt: {prompt}")
|
||
logger.info(f"频率调整 response: {response}")
|
||
logger.info(f"频率调整 reasoning_content: {reasoning_content}")
|
||
|
||
final_value_by_api = frequency_api.get_current_talk_value(self.chat_id)
|
||
|
||
# LLM依然输出过多内容时取消本次调整。合法最多4个字，但有的模型可能会输出一些markdown换行符等，需要长度宽限
|
||
if len(response) < 20:
|
||
if "过于频繁" in response:
|
||
logger.info(f"频率调整: 过于频繁，调整值到{final_value_by_api}")
|
||
self.talk_frequency_adjust = max(0.1, min(1.5, self.talk_frequency_adjust * 0.8))
|
||
elif "过少" in response:
|
||
logger.info(f"频率调整: 过少，调整值到{final_value_by_api}")
|
||
self.talk_frequency_adjust = max(0.1, min(1.5, self.talk_frequency_adjust * 1.2))
|
||
except Exception as e:
|
||
logger.error(f"频率调整失败: {e}")
|
||
# 即使失败也保持时间戳更新，避免频繁重试
|
||
|
||
|
||
class FrequencyControlManager:
|
||
"""频率控制管理器，管理多个聊天流的频率控制实例"""
|
||
|
||
def __init__(self):
|
||
self.frequency_control_dict: Dict[str, FrequencyControl] = {}
|
||
|
||
def get_or_create_frequency_control(self, chat_id: str) -> FrequencyControl:
|
||
"""获取或创建指定聊天流的频率控制实例"""
|
||
if chat_id not in self.frequency_control_dict:
|
||
self.frequency_control_dict[chat_id] = FrequencyControl(chat_id)
|
||
return self.frequency_control_dict[chat_id]
|
||
|
||
def remove_frequency_control(self, chat_id: str) -> bool:
|
||
"""移除指定聊天流的频率控制实例"""
|
||
if chat_id in self.frequency_control_dict:
|
||
del self.frequency_control_dict[chat_id]
|
||
return True
|
||
return False
|
||
|
||
def get_all_chat_ids(self) -> list[str]:
|
||
"""获取所有有频率控制的聊天ID"""
|
||
return list(self.frequency_control_dict.keys())
|
||
|
||
|
||
init_prompt()
|
||
|
||
# 创建全局实例
|
||
frequency_control_manager = FrequencyControlManager()
|