Merge pull request #312 from SuanmoSuanyangTechnology/fix/release_memory_bug

Fix/release memory bug

api/app/core/memory/agent/langgraph_graph/routing/write_router.py

@@ -18,7 +18,6 @@ from app.services.memory_konwledges_server import write_rag
 from app.services.task_service import get_task_memory_write_result
 from app.tasks import write_message_task
 from app.utils.config_utils import resolve_config_id
-
 logger = get_agent_logger(__name__)
 template_root = os.path.join(PROJECT_ROOT_, 'memory', 'agent', 'utils', 'prompt')
 
@@ -95,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):
     with get_db_context() as db_session:
-        try:
         repo = LongTermMemoryRepository(db_session)
         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)
@@ -113,11 +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)
             repo.upsert(end_user_id, long_messages)
             logger.info(f'写入短长期：')
-            # yield db_session
-        finally:
-            if db_session.in_transaction():
-                db_session.rollback()
-            db_session.close()
+
 
 
 '''根据窗口'''

api/app/core/memory/agent/langgraph_graph/write_graph.py

@@ -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))
     # 获取数据库会话
     with get_db_context() as db_session:
-        try:
         config_service = MemoryConfigService(db_session)
         memory_config = config_service.load_memory_config(
             config_id=memory_config,  # 改为整数
@@ -62,10 +61,6 @@ async def long_term_storage(long_term_type:str="chunk",langchain_messages:list=[
         if  long_term_type=='aggregate':
             """方案三：聚合判断"""
             await aggregate_judgment(end_user_id, langchain_messages, memory_config)
-        finally:
-            if db_session.in_transaction():
-                db_session.rollback()
-            db_session.close()