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

fix:ruff

Browse Source
This commit is contained in:
SengokuCola
2025-05-12 13:20:00 +08:00
parent 03589c6dcf
commit 11248cd0e3
2 changed files with 9 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/plugins/heartFC_chat/heartflow_prompt_builder.py
View File

@@ -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.person_info.relationship_manager import relationship_manager
from src.plugins.chat.utils import get_embedding from src.plugins.chat.utils import get_embedding
import time import time
from typing import Union, Optional, Deque, Dict, Any from typing import Union, Optional, Dict, Any
from ...common.database import db from ...common.database import db
from ..chat.utils import get_recent_group_speaker from ..chat.utils import get_recent_group_speaker
from ..moods.moods import MoodManager from ..moods.moods import MoodManager
@@ -15,7 +15,6 @@ from ..memory_system.Hippocampus import HippocampusManager
from ..schedule.schedule_generator import bot_schedule from ..schedule.schedule_generator import bot_schedule
from ..knowledge.knowledge_lib import qa_manager from ..knowledge.knowledge_lib import qa_manager
import traceback import traceback
from .heartFC_Cycleinfo import CycleDetail
logger = get_logger("prompt") logger = get_logger("prompt")

16
src/plugins/heartFC_chat/memory_activator.py
View File

@@ -13,6 +13,7 @@ from typing import List, Dict
logger = get_logger("memory_activator") logger = get_logger("memory_activator")
def init_prompt(): def init_prompt():
# --- Group Chat Prompt --- # --- Group Chat Prompt ---
memory_activator_prompt = """ memory_activator_prompt = """
@@ -83,20 +84,17 @@ class MemoryActivator:
# 激活时，所有已有记忆的duration+1，达到3则移除 # 激活时，所有已有记忆的duration+1，达到3则移除
for m in self.running_memory[:]: for m in self.running_memory[:]:
m['duration'] = m.get('duration', 1) + 1 m["duration"] = m.get("duration", 1) + 1
self.running_memory = [m for m in self.running_memory if m['duration'] < 4] self.running_memory = [m for m in self.running_memory if m["duration"] < 4]
if related_memory: if related_memory:
for topic, memory in related_memory: for topic, memory in related_memory:
# 检查是否已存在相同topic和content的记忆 # 检查是否已存在相同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: if not exists:
self.running_memory.append({ self.running_memory.append(
"topic": topic, {"topic": topic, "content": memory, "timestamp": datetime.now().isoformat(), "duration": 1}
"content": memory, )
"timestamp": datetime.now().isoformat(),
"duration": 1
})
logger.debug(f"添加新记忆: {topic} - {memory}") logger.debug(f"添加新记忆: {topic} - {memory}")
return self.running_memory return self.running_memory