Merge pull request #3101 from xinnan-tech/fix-title-summary

fix:修复PowerMem和Mem0AI记忆模型不会总结会话标题的问题
This commit is contained in:
wengzh
2026-04-16 09:04:58 +08:00
committed by GitHub
2 changed files with 73 additions and 68 deletions
@@ -2,6 +2,7 @@ import json
import traceback import traceback
from ..base import MemoryProviderBase, logger from ..base import MemoryProviderBase, logger
from config.manage_api_client import generate_and_save_chat_summary
from mem0 import MemoryClient from mem0 import MemoryClient
from core.utils.util import check_model_key from core.utils.util import check_model_key
@@ -30,12 +31,8 @@ class MemoryProvider(MemoryProviderBase):
self.use_mem0 = False self.use_mem0 = False
async def save_memory(self, msgs, session_id=None): async def save_memory(self, msgs, session_id=None):
if not self.use_mem0:
return None
if len(msgs) < 2:
return None
try: try:
if self.use_mem0 and len(msgs) >= 2:
# Format the content as a message list for mem0 # Format the content as a message list for mem0
messages = [] messages = []
for message in msgs: for message in msgs:
@@ -61,6 +58,12 @@ class MemoryProvider(MemoryProviderBase):
logger.bind(tag=TAG).debug(f"Save memory result: {result}") logger.bind(tag=TAG).debug(f"Save memory result: {result}")
except Exception as e: except Exception as e:
logger.bind(tag=TAG).error(f"保存记忆失败: {str(e)}") logger.bind(tag=TAG).error(f"保存记忆失败: {str(e)}")
# Generate and save chat summary (SLM summarizes session title)
# This is independent of Mem0AI's memory saving
if session_id:
await generate_and_save_chat_summary(session_id)
return None return None
async def query_memory(self, query: str) -> str: async def query_memory(self, query: str) -> str:
@@ -16,6 +16,7 @@ import traceback
from typing import Optional, Dict, Any from typing import Optional, Dict, Any
from ..base import MemoryProviderBase, logger from ..base import MemoryProviderBase, logger
from config.manage_api_client import generate_and_save_chat_summary
TAG = __name__ TAG = __name__
@@ -162,15 +163,8 @@ class MemoryProvider(MemoryProviderBase):
Returns: Returns:
Result from PowerMem API or None if failed Result from PowerMem API or None if failed
""" """
if not self.use_powermem or self.memory_client is None:
logger.bind(tag=TAG).warning("PowerMem is not available, skipping save_memory")
return None
if len(msgs) < 2:
logger.bind(tag=TAG).debug("Not enough messages to save (need at least 2)")
return None
try: try:
if self.use_powermem and self.memory_client is not None and len(msgs) >= 2:
# Format the content as a message list for PowerMem # Format the content as a message list for PowerMem
messages = [] messages = []
for message in msgs: for message in msgs:
@@ -208,12 +202,20 @@ class MemoryProvider(MemoryProviderBase):
if result.get('profile_extracted'): if result.get('profile_extracted'):
self.last_profile_content = result.get('profile_content', '') self.last_profile_content = result.get('profile_content', '')
logger.bind(tag=TAG).debug(f"User profile extracted: {self.last_profile_content}") logger.bind(tag=TAG).debug(f"User profile extracted: {self.last_profile_content}")
else:
return result if not self.use_powermem or self.memory_client is None:
logger.bind(tag=TAG).warning("PowerMem is not available, skipping save_memory")
elif len(msgs) < 2:
logger.bind(tag=TAG).debug("Not enough messages to save (need at least 2)")
except Exception as e: except Exception as e:
logger.bind(tag=TAG).error(f"Error saving memory: {str(e)}") logger.bind(tag=TAG).error(f"Error saving memory: {str(e)}")
logger.bind(tag=TAG).debug(f"Detailed error: {traceback.format_exc()}") logger.bind(tag=TAG).debug(f"Detailed error: {traceback.format_exc()}")
# Generate and save chat summary (SLM summarizes session title)
# This is independent of PowerMem's memory saving
if session_id:
await generate_and_save_chat_summary(session_id)
return None return None
async def query_memory(self, query: str) -> str: async def query_memory(self, query: str) -> str: