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 69f2ee64d9
commit e62843ea58
7 changed files with 113 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/chat/affinity_flow/relationship_tracker.py
View File

@@ -8,7 +8,7 @@ import time
from typing import Dict, List, Optional
from src.common.logger import get_logger
from src.config.config import model_config
from src.config.config import model_config, global_config
from src.llm_models.utils_model import LLMRequest
from src.common.database.sqlalchemy_database_api import get_db_session
from src.common.database.sqlalchemy_models import UserRelationships, Messages
@@ -79,7 +79,7 @@ class UserRelationshipTracker:
del self.tracking_users[oldest_user]
# 获取当前关系分
current_relationship_score = 0.3 # 默认值
current_relationship_score = global_config.affinity_flow.base_relationship_score # 默认值
if self.interest_scoring_system:
current_relationship_score = self.interest_scoring_system.get_user_relationship(user_id)
@@ -139,7 +139,7 @@ Bot回复: {interaction["bot_reply"]}
# 清理LLM响应，移除可能的格式标记
cleaned_response = self._clean_llm_json_response(llm_response)
response_data = json.loads(cleaned_response)
new_score = max(0.0, min(1.0, float(response_data.get("new_relationship_score", 0.3))))
new_score = max(0.0, min(1.0, float(response_data.get("new_relationship_score", global_config.affinity_flow.base_relationship_score))))
if self.interest_scoring_system:
self.interest_scoring_system.update_user_relationship(
@@ -258,7 +258,7 @@ Bot回复: {interaction["bot_reply"]}
# 检查缓存是否过期
cache_time = cache_data.get("last_tracked", 0)
if time.time() - cache_time < self.cache_expiry_hours * 3600:
return cache_data.get("relationship_score", 0.3)
return cache_data.get("relationship_score", global_config.affinity_flow.base_relationship_score)
# 缓存过期或不存在，从数据库获取
relationship_data = self._get_user_relationship_from_db(user_id)
@@ -266,13 +266,13 @@ Bot回复: {interaction["bot_reply"]}
# 更新缓存
self.user_relationship_cache[user_id] = {
"relationship_text": relationship_data.get("relationship_text", ""),
"relationship_score": relationship_data.get("relationship_score", 0.3),
"relationship_score": relationship_data.get("relationship_score", global_config.affinity_flow.base_relationship_score),
"last_tracked": time.time(),
}
return relationship_data.get("relationship_score", 0.3)
return relationship_data.get("relationship_score", global_config.affinity_flow.base_relationship_score)
# 数据库中也没有，返回默认值
return 0.3
return global_config.affinity_flow.base_relationship_score
def _get_user_relationship_from_db(self, user_id: str) -> Optional[Dict]:
"""从数据库获取用户关系数据"""
@@ -357,7 +357,7 @@ Bot回复: {interaction["bot_reply"]}
# 获取当前关系数据
current_relationship = self._get_user_relationship_from_db(user_id)
current_score = current_relationship.get("relationship_score", 0.3) if current_relationship else 0.3
current_score = current_relationship.get("relationship_score", global_config.affinity_flow.base_relationship_score) if current_relationship else global_config.affinity_flow.base_relationship_score
current_text = current_relationship.get("relationship_text", "新用户") if current_relationship else "新用户"
# 使用LLM分析并更新关系