gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(affinity-flow): 重构亲和流配置系统

Browse Source 
将硬编码的兴趣度评分参数迁移到集中式配置管理，新增AffinityFlowConfig配置类
- 从全局配置加载评分权重、阈值和计算参数
- 统一管理匹配奖励、关系分数和提及bot相关配置
- 更新配置文件模板包含完整的亲和流参数
- 确保各模块使用一致的配置值而非硬编码常量
This commit is contained in:
Windpicker-owo
2025-09-19 13:16:45 +08:00
parent 812e4d76f8
commit 96e4dc2946
7 changed files with 109 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/chat/interest_system/bot_interest_manager.py
View File

@@ -9,6 +9,7 @@ from datetime import datetime
import numpy as np
from src.common.logger import get_logger
from src.config.config import global_config
from src.common.data_models.bot_interest_data_model import (
BotPersonalityInterests, BotInterestTag, InterestMatchResult
)
@@ -433,9 +434,10 @@ class BotInterestManager:
low_similarity_count = 0
# 分级相似度阈值
high_threshold = 0.55
medium_threshold = 0.47
low_threshold = 0.3
affinity_config = global_config.affinity_flow
high_threshold = affinity_config.high_match_interest_threshold
medium_threshold = affinity_config.medium_match_interest_threshold
low_threshold = affinity_config.low_match_interest_threshold
logger.debug(f"🔍 使用分级相似度阈值: 高={high_threshold}, 中={medium_threshold}, 低={low_threshold}")
@@ -449,7 +451,7 @@ class BotInterestManager:
# 根据相似度等级应用不同的加成
if similarity > high_threshold:
# 高相似度：强加成
enhanced_score = weighted_score * 1.8
enhanced_score = weighted_score * affinity_config.high_match_keyword_multiplier
match_count += 1
high_similarity_count += 1
result.add_match(tag.tag_name, enhanced_score, [tag.tag_name])
@@ -457,7 +459,7 @@ class BotInterestManager:
elif similarity > medium_threshold:
# 中相似度：中等加成
enhanced_score = weighted_score * 1.4
enhanced_score = weighted_score * affinity_config.medium_match_keyword_multiplier
match_count += 1
medium_similarity_count += 1
result.add_match(tag.tag_name, enhanced_score, [tag.tag_name])
@@ -465,7 +467,7 @@ class BotInterestManager:
elif similarity > low_threshold:
# 低相似度：轻微加成
enhanced_score = weighted_score * 1.15
enhanced_score = weighted_score * affinity_config.low_match_keyword_multiplier
match_count += 1
low_similarity_count += 1
result.add_match(tag.tag_name, enhanced_score, [tag.tag_name])
@@ -506,6 +508,7 @@ class BotInterestManager:
if not keywords or not matched_tags:
return {}
affinity_config = global_config.affinity_flow
bonus_dict = {}
for tag_name in matched_tags:
@@ -518,21 +521,21 @@ class BotInterestManager:
# 完全匹配
if keyword_lower == tag_name_lower:
bonus += 0.3
logger.debug(f" 🎯 关键词完全匹配: '{keyword}' == '{tag_name}' (+0.3)")
bonus += affinity_config.high_match_interest_threshold * 0.6 # 使用高匹配阈值的60%作为完全匹配奖励
logger.debug(f" 🎯 关键词完全匹配: '{keyword}' == '{tag_name}' (+{affinity_config.high_match_interest_threshold * 0.6:.3f})")
# 包含匹配
elif keyword_lower in tag_name_lower or tag_name_lower in keyword_lower:
bonus += 0.15
logger.debug(f" 🎯 关键词包含匹配: '{keyword}' ⊃ '{tag_name}' (+0.15)")
bonus += affinity_config.medium_match_interest_threshold * 0.3 # 使用中匹配阈值的30%作为包含匹配奖励
logger.debug(f" 🎯 关键词包含匹配: '{keyword}' ⊃ '{tag_name}' (+{affinity_config.medium_match_interest_threshold * 0.3:.3f})")
# 部分匹配（编辑距离）
elif self._calculate_partial_match(keyword_lower, tag_name_lower):
bonus += 0.08
logger.debug(f" 🎯 关键词部分匹配: '{keyword}' ≈ '{tag_name}' (+0.08)")
bonus += affinity_config.low_match_interest_threshold * 0.4 # 使用低匹配阈值的40%作为部分匹配奖励
logger.debug(f" 🎯 关键词部分匹配: '{keyword}' ≈ '{tag_name}' (+{affinity_config.low_match_interest_threshold * 0.4:.3f})")
if bonus > 0:
bonus_dict[tag_name] = min(bonus, 0.5) # 最大奖励限制为0.5
bonus_dict[tag_name] = min(bonus, affinity_config.max_match_bonus) # 使用配置的最大奖励限制
return bonus_dict