mirror of https://github.com/Mai-with-u/MaiBot.git
modified: src/plugins/group_nickname/nickname_utils.py
parent
a96c2720dc
commit
5886be4ac9
|
|
@ -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 初始化完成。")
|
||||
|
|
|
|||
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Reference in New Issue