fix: async bug

crestalnetwork · Feb 4, 2025 · ec08240 · ec08240
1 parent 8363c23
commit ec08240
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/app/core/engine.py b/app/core/engine.py
@@ -343,13 +343,13 @@ async def execute_agent(
     )
 
     # cold start or needs reinitialization
-    if aid not in agents or needs_reinit:
-        await initialize_agent(aid)
+    if (aid not in agents) or needs_reinit:
         resp_debug.append(
             "[ Agent cold start ... ]"
             if aid not in agents
             else "[ Agent reinitialized ... ]"
         )
+        await initialize_agent(aid)
         resp_debug.append(
             f"\n------------------- start cost: {time.perf_counter() - last:.3f} seconds\n"
         )
@@ -371,7 +371,7 @@ async def execute_agent(
         try:
             resp_debug_append = "\n===================\n\n[ system ]\n"
             resp_debug_append += agent_prompt(agent)
-            snap = executor.get_state(stream_config)
+            snap = await executor.aget_state(stream_config)
             if snap.values and "messages" in snap.values:
                 for msg in snap.values["messages"]:
                     resp_debug_append += f"[ {msg.type} ]\n{str(msg.content)}\n\n"

diff --git a/models/skill.py b/models/skill.py
@@ -83,7 +83,7 @@ async def save(self) -> None:
                 db.add(existing)
             else:
                 db.add(self)
-            db.commit()
+            await db.commit()
 
     @classmethod
     async def clean_data(cls, agent_id: str):
@@ -175,7 +175,7 @@ async def save(self) -> None:
                 db.add(existing)
             else:
                 db.add(self)
-            db.commit()
+            await db.commit()
 
     @classmethod
     async def clean_data(cls, agent_id: str, thread_id: str):