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

knowledge_retrieval/bug/fix

Browse Source
This commit is contained in:
lixinyue
2026-02-04 18:06:56 +08:00
parent 918e7285c4
commit 34276e2066
2 changed files with 31 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/app/core/memory/agent/langgraph_graph/routing/write_router.py
View File

@@ -94,7 +94,6 @@ async def write(storage_type, end_user_id, user_message, ai_message, user_rag_me
async def term_memory_save(long_term_messages,actual_config_id,end_user_id,type,scope): async def term_memory_save(long_term_messages,actual_config_id,end_user_id,type,scope):
with get_db_context() as db_session: with get_db_context() as db_session:
try:
repo = LongTermMemoryRepository(db_session) repo = LongTermMemoryRepository(db_session)
await long_term_storage(long_term_type=AgentMemory_Long_Term.STRATEGY_CHUNK, langchain_messages=long_term_messages, await long_term_storage(long_term_type=AgentMemory_Long_Term.STRATEGY_CHUNK, langchain_messages=long_term_messages,
memory_config=actual_config_id, end_user_id=end_user_id, scope=scope) memory_config=actual_config_id, end_user_id=end_user_id, scope=scope)
@@ -112,9 +111,7 @@ async def term_memory_save(long_term_messages,actual_config_id,end_user_id,type,
long_messages = await messages_parse(long_time_data) long_messages = await messages_parse(long_time_data)
repo.upsert(end_user_id, long_messages) repo.upsert(end_user_id, long_messages)
logger.info(f'写入短长期：') logger.info(f'写入短长期：')
# yield db_session
finally:
db_session.close()
'''根据窗口''' '''根据窗口'''

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

@@ -47,7 +47,6 @@ async def long_term_storage(long_term_type:str="chunk",langchain_messages:list=[
write_store.save_session_write(end_user_id, (langchain_messages)) write_store.save_session_write(end_user_id, (langchain_messages))
# 获取数据库会话 # 获取数据库会话
with get_db_context() as db_session: 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, # 改为整数
@@ -62,8 +61,6 @@ 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:
db_session.close()