fix: async flush the memory when agent update

crestalnetwork · Feb 25, 2025 · 62f2215 · 62f2215
1 parent aaa04d9
commit 62f2215
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/app/admin/api.py b/app/admin/api.py
@@ -1,3 +1,4 @@
+import asyncio
 import json
 import logging
 
@@ -66,8 +67,9 @@ async def _process_agent(
         if agent_data and agent_data.cdp_wallet_data:
             has_wallet = True
             wallet_data = json.loads(agent_data.cdp_wallet_data)
-        # Clean agent memory
-        await clean_agent_memory(latest_agent.id, clean_agent_memory=True)
+        # Run clean_agent_memory in background
+        import asyncio
+        asyncio.create_task(clean_agent_memory(latest_agent.id, clean_agent_memory=True))
 
     if not has_wallet:
         # create the wallet

diff --git a/app/core/engine.py b/app/core/engine.py
@@ -679,6 +679,7 @@ async def clean_agent_memory(
 
             await db.commit()
 
+            logger.info(f"Agent [{agent_id}] data cleaned up successfully.")
             return "Agent data cleaned up successfully."
         except SQLAlchemyError as e:
             # Handle other SQLAlchemy-related errors