adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

memory_BUG_long_term

Browse Source
This commit is contained in:
lixinyue
2026-02-04 14:24:50 +08:00
parent 62aba2dd38
commit 72b5e5cf8e
1 changed files with 22 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/app/core/memory/agent/langgraph_graph/write_graph.py
View File

@@ -7,7 +7,7 @@ from contextlib import asynccontextmanager
from langgraph.constants import END, START from langgraph.constants import END, START
from langgraph.graph import StateGraph from langgraph.graph import StateGraph
from app.db import get_db from app.db import get_db, get_db_context
from app.core.logging_config import get_agent_logger from app.core.logging_config import get_agent_logger
from app.core.memory.agent.utils.llm_tools import WriteState from app.core.memory.agent.utils.llm_tools import WriteState
from app.core.memory.agent.langgraph_graph.nodes.write_nodes import write_node from app.core.memory.agent.langgraph_graph.nodes.write_nodes import write_node
@@ -46,7 +46,8 @@ async def long_term_storage(long_term_type:str="chunk",langchain_messages:list=[
from app.core.memory.agent.utils.redis_tool import write_store from app.core.memory.agent.utils.redis_tool import write_store
write_store.save_session_write(end_user_id, (langchain_messages)) write_store.save_session_write(end_user_id, (langchain_messages))
# 获取数据库会话 # 获取数据库会话
db_session = next(get_db()) with get_db_context() as db_session:
try:
config_service = MemoryConfigService(db_session) config_service = MemoryConfigService(db_session)
memory_config = config_service.load_memory_config( memory_config = config_service.load_memory_config(
config_id=memory_config, # 改为整数 config_id=memory_config, # 改为整数
@@ -61,6 +62,11 @@ async def long_term_storage(long_term_type:str="chunk",langchain_messages:list=[
if long_term_type=='aggregate': if long_term_type=='aggregate':
"""方案三：聚合判断""" """方案三：聚合判断"""
await aggregate_judgment(end_user_id, langchain_messages, memory_config) await aggregate_judgment(end_user_id, langchain_messages, memory_config)
finally:
if db_session.in_transaction():
db_session.rollback()
db_session.close()
async def write_long_term(storage_type,end_user_id,message_chat,aimessages,user_rag_memory_id,actual_config_id): async def write_long_term(storage_type,end_user_id,message_chat,aimessages,user_rag_memory_id,actual_config_id):