modified: src/plugins/group_nickname/nickname_utils.py

pull/937/head
Bakadax 2025-05-04 18:08:32 +08:00
parent a96c2720dc
commit 5886be4ac9
2 changed files with 5 additions and 5 deletions

View File

@ -103,7 +103,7 @@ class NicknameManager:
logger.info("正在初始化 NicknameManager 组件...")
self.config = global_config
self.is_enabled = self.config.ENABLE_NICKNAME_MAPPING
self.is_enabled = self.config.enable_nickname_mapping
# 数据库处理器
person_info_collection = getattr(db, "person_info", None)
@ -138,12 +138,12 @@ class NicknameManager:
self.is_enabled = False
# 队列和线程
self.queue_max_size = getattr(self.config, "NICKNAME_QUEUE_MAX_SIZE", 100)
self.queue_max_size = getattr(self.config, "nickname_queue_max_size", 100)
# 使用 asyncio.Queue
self.nickname_queue: asyncio.Queue = asyncio.Queue(maxsize=self.queue_max_size)
self._stop_event = threading.Event() # stop_event 仍然使用 threading.Event因为它是由另一个线程设置的
self._nickname_thread: Optional[threading.Thread] = None
self.sleep_interval = getattr(self.config, "NICKNAME_PROCESS_SLEEP_INTERVAL", 0.5) # 超时时间
self.sleep_interval = getattr(self.config, "nickname_process_sleep_interval", 0.5) # 超时时间
self._initialized = True
logger.info("NicknameManager 初始化完成。")

View File

@ -25,7 +25,7 @@ def select_nicknames_for_prompt(all_nicknames_info: Dict[str, List[Dict[str, int
return []
candidates = [] # 存储 (用户名, 绰号, 次数, 权重)
smoothing_factor = getattr(global_config, "NICKNAME_PROBABILITY_SMOOTHING", 1.0) # 平滑因子避免权重为0
smoothing_factor = getattr(global_config, "nickname_probability_smoothing", 1.0) # 平滑因子避免权重为0
for user_name, nicknames in all_nicknames_info.items():
if nicknames and isinstance(nicknames, list):
@ -48,7 +48,7 @@ def select_nicknames_for_prompt(all_nicknames_info: Dict[str, List[Dict[str, int
return []
# 确定需要选择的数量
max_nicknames = getattr(global_config, "MAX_NICKNAMES_IN_PROMPT", 5)
max_nicknames = getattr(global_config, "max_nicknames_in_prompt", 5)
num_to_select = min(max_nicknames, len(candidates))
try: