From 11248cd0e3035efc012e9c07e0a806f8aae420a3 Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Mon, 12 May 2025 13:20:00 +0800 Subject: [PATCH] fix:ruff --- .../heartFC_chat/heartflow_prompt_builder.py | 3 +-- src/plugins/heartFC_chat/memory_activator.py | 18 ++++++++---------- 2 files changed, 9 insertions(+), 12 deletions(-) diff --git a/src/plugins/heartFC_chat/heartflow_prompt_builder.py b/src/plugins/heartFC_chat/heartflow_prompt_builder.py index 39ccce53e..4d37268d6 100644 --- a/src/plugins/heartFC_chat/heartflow_prompt_builder.py +++ b/src/plugins/heartFC_chat/heartflow_prompt_builder.py @@ -7,7 +7,7 @@ from src.plugins.utils.chat_message_builder import build_readable_messages, get_ from src.plugins.person_info.relationship_manager import relationship_manager from src.plugins.chat.utils import get_embedding import time -from typing import Union, Optional, Deque, Dict, Any +from typing import Union, Optional, Dict, Any from ...common.database import db from ..chat.utils import get_recent_group_speaker from ..moods.moods import MoodManager @@ -15,7 +15,6 @@ from ..memory_system.Hippocampus import HippocampusManager from ..schedule.schedule_generator import bot_schedule from ..knowledge.knowledge_lib import qa_manager import traceback -from .heartFC_Cycleinfo import CycleDetail logger = get_logger("prompt") diff --git a/src/plugins/heartFC_chat/memory_activator.py b/src/plugins/heartFC_chat/memory_activator.py index aa60869d8..67fac8ab6 100644 --- a/src/plugins/heartFC_chat/memory_activator.py +++ b/src/plugins/heartFC_chat/memory_activator.py @@ -13,6 +13,7 @@ from typing import List, Dict logger = get_logger("memory_activator") + def init_prompt(): # --- Group Chat Prompt --- memory_activator_prompt = """ @@ -83,23 +84,20 @@ class MemoryActivator: # 激活时,所有已有记忆的duration+1,达到3则移除 for m in self.running_memory[:]: - m['duration'] = m.get('duration', 1) + 1 - self.running_memory = [m for m in self.running_memory if m['duration'] < 4] + m["duration"] = m.get("duration", 1) + 1 + self.running_memory = [m for m in self.running_memory if m["duration"] < 4] if related_memory: for topic, memory in related_memory: # 检查是否已存在相同topic和content的记忆 - exists = any(m['topic'] == topic and m['content'] == memory for m in self.running_memory) + exists = any(m["topic"] == topic and m["content"] == memory for m in self.running_memory) if not exists: - self.running_memory.append({ - "topic": topic, - "content": memory, - "timestamp": datetime.now().isoformat(), - "duration": 1 - }) + self.running_memory.append( + {"topic": topic, "content": memory, "timestamp": datetime.now().isoformat(), "duration": 1} + ) logger.debug(f"添加新记忆: {topic} - {memory}") return self.running_memory -init_prompt() \ No newline at end of file +init_prompt()