diff --git a/src/chat/focus_chat/expressors/exprssion_learner.py b/src/chat/focus_chat/expressors/exprssion_learner.py index 2cc1b0b34..f4107a459 100644 --- a/src/chat/focus_chat/expressors/exprssion_learner.py +++ b/src/chat/focus_chat/expressors/exprssion_learner.py @@ -111,14 +111,43 @@ class ExpressionLearner: async def learn_and_store_expression(self) -> List[Tuple[str, str, str]]: """ 学习并存储表达方式，分别学习语言风格和句法特点 + 同时对所有已存储的表达方式进行全局衰减 """ + current_time = time.time() + + # 全局衰减所有已存储的表达方式 + for type in ["style", "grammar"]: + base_dir = os.path.join("data", "expression", f"learnt_{type}") + if not os.path.exists(base_dir): + continue + + for chat_id in os.listdir(base_dir): + file_path = os.path.join(base_dir, chat_id, "expressions.json") + if not os.path.exists(file_path): + continue + + try: + with open(file_path, "r", encoding="utf-8") as f: + expressions = json.load(f) + + # 应用全局衰减 + decayed_expressions = self.apply_decay_to_expressions(expressions, current_time) + + # 保存衰减后的结果 + with open(file_path, "w", encoding="utf-8") as f: + json.dump(decayed_expressions, f, ensure_ascii=False, indent=2) + except Exception as e: + logger.error(f"全局衰减{type}表达方式失败: {e}") + continue + + # 学习新的表达方式（这里会进行局部衰减） for i in range(3): - learnt_style: Optional[List[Tuple[str, str, str]]] = await self.learn_and_store(type="style", num=15) + learnt_style: Optional[List[Tuple[str, str, str]]] = await self.learn_and_store(type="style", num=25) if not learnt_style: return [] - for i in range(1): - learnt_grammar: Optional[List[Tuple[str, str, str]]] = await self.learn_and_store(type="grammar", num=15) + for j in range(1): + learnt_grammar: Optional[List[Tuple[str, str, str]]] = await self.learn_and_store(type="grammar", num=10) if not learnt_grammar: return [] @@ -126,29 +155,29 @@ class ExpressionLearner: def calculate_decay_factor(self, time_diff_days: float) -> float: """ - 计算衰减因子 - 当时间差为0天或30天时，衰减值为0.01 - 当时间差为7天时，衰减值为1.0 + 计算衰减值 + 当时间差为0天时，衰减值为0.001 + 当时间差为7天时，衰减值为0 + 当时间差为30天时，衰减值为0.001 使用二次函数进行曲线插值 """ if time_diff_days <= 0 or time_diff_days >= DECAY_DAYS: - return DECAY_MIN + return 0.001 # 使用二次函数进行插值 # 将7天作为顶点，0天和30天作为两个端点 # 使用顶点式：y = a(x-h)^2 + k，其中(h,k)为顶点 h = 7.0 # 顶点x坐标 - k = 1.0 # 顶点y坐标 + k = 0.001 # 顶点y坐标 - # 计算a值，使得x=0和x=30时y=0.01 - # 0.01 = a(0-7)^2 + 1 - # 0.01 = a(30-7)^2 + 1 - # 解得a = -0.99/49 - a = -0.99 / 49 + # 计算a值，使得x=0和x=30时y=0.001 + # 0.001 = a(0-7)^2 + 0.001 + # 解得a = 0 + a = 0 - # 计算衰减因子 + # 计算衰减值 decay = a * (time_diff_days - h) ** 2 + k - return max(DECAY_MIN, min(1.0, decay)) + return min(0.001, decay) def apply_decay_to_expressions(self, expressions: List[Dict[str, Any]], current_time: float) -> List[Dict[str, Any]]: """ @@ -157,11 +186,15 @@ class ExpressionLearner: """ result = [] for expr in expressions: - last_active = expr.get("last_active_time", current_time) + # 确保last_active_time存在，如果不存在则使用current_time + if "last_active_time" not in expr: + expr["last_active_time"] = current_time + + last_active = expr["last_active_time"] time_diff_days = (current_time - last_active) / (24 * 3600) # 转换为天 - decay_factor = self.calculate_decay_factor(time_diff_days) - expr["count"] = expr.get("count", 1) * decay_factor + decay_value = self.calculate_decay_factor(time_diff_days) + expr["count"] = max(0.01, expr.get("count", 1) - decay_value) if expr["count"] > 0: result.append(expr) @@ -225,7 +258,7 @@ class ExpressionLearner: old_data = [] # 应用衰减 - old_data = self.apply_decay_to_expressions(old_data, current_time) + # old_data = self.apply_decay_to_expressions(old_data, current_time) # 合并逻辑 for new_expr in expr_list: diff --git a/src/chat/focus_chat/info_processors/relationship_processor.py b/src/chat/focus_chat/info_processors/relationship_processor.py index d0614ad2e..0e7a6d927 100644 --- a/src/chat/focus_chat/info_processors/relationship_processor.py +++ b/src/chat/focus_chat/info_processors/relationship_processor.py @@ -27,9 +27,9 @@ def init_prompt(): {chat_observe_info} 现在请你根据现有的信息，总结你和群里的人的关系 -1. 当聊天记录中提到你时，请输出你和这个人之间的关系 -2. 当聊天记录中提到其他人时，请输出你和这个人之间的关系 -3. 如果没有特别需要提及的关系，就不用输出这个人 +1. 根据聊天记录的需要，精简你和其他人的关系并输出 +2. 根据聊天记录，如果需要提及你和某个人的关系，请输出你和这个人之间的关系 +3. 如果没有特别需要提及的关系，就不用输出这个人的关系 输出内容平淡一些，说中文。 请注意不要输出多余内容(包括前后缀，括号()，表情包，at或 @等 )。只输出关系内容，记得明确说明这是你的关系。 diff --git a/src/individuality/expression_style.py b/src/individuality/expression_style.py index 20d20bfcc..5ce48a952 100644 --- a/src/individuality/expression_style.py +++ b/src/individuality/expression_style.py @@ -1,4 +1,6 @@ import random + +from reportportal_client import current from src.common.logger_manager import get_logger from src.llm_models.utils_model import LLMRequest from src.config.config import global_config @@ -7,7 +9,6 @@ from typing import List, Tuple import os import json from datetime import datetime -from src.individuality.individuality import individuality logger = get_logger("expressor") @@ -89,21 +90,18 @@ class PersonalityExpression: os.makedirs(os.path.dirname(self.expressions_file_path), exist_ok=True) current_style_text = global_config.expression.expression_style - current_personality = individuality.get_personality_prompt(x_person=2, level=2) - current_identity = individuality.get_identity_prompt(x_person=2, level=2) + current_personality = global_config.personality.personality_core meta_data = self._read_meta_data() last_style_text = meta_data.get("last_style_text") last_personality = meta_data.get("last_personality") - last_identity = meta_data.get("last_identity") count = meta_data.get("count", 0) # 检查是否有任何变化 if (current_style_text != last_style_text or - current_personality != last_personality or - current_identity != last_identity): - logger.info(f"检测到变化：\n风格: '{last_style_text}' -> '{current_style_text}'\n人格: '{last_personality}' -> '{current_personality}'\n身份: '{last_identity}' -> '{current_identity}'") + current_personality != last_personality): + logger.info(f"检测到变化：\n风格: '{last_style_text}' -> '{current_style_text}'\n人格: '{last_personality}' -> '{current_personality}'") count = 0 if os.path.exists(self.expressions_file_path): try: @@ -119,7 +117,6 @@ class PersonalityExpression: { "last_style_text": current_style_text, "last_personality": current_personality, - "last_identity": current_identity, "count": count, "last_update_time": meta_data.get("last_update_time"), } @@ -142,7 +139,6 @@ class PersonalityExpression: { "last_style_text": current_style_text, "last_personality": current_personality, - "last_identity": current_identity, "count": count, "last_update_time": meta_data.get("last_update_time"), } @@ -200,7 +196,6 @@ class PersonalityExpression: { "last_style_text": current_style_text, "last_personality": current_personality, - "last_identity": current_identity, "count": count, "last_update_time": current_time }