fix: Start workflows asynchronously

julep-ai · Sep 17, 2024 · a7ec9d5 · a7ec9d5
1 parent b1ba790
commit a7ec9d5
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/agents-api/agents_api/clients/temporal.py b/agents-api/agents_api/clients/temporal.py
@@ -75,21 +75,25 @@ async def get_workflow_handle(
 async def run_truncation_task(
     token_count_threshold: int, developer_id: UUID, session_id: UUID, job_id: UUID
 ):
+    from ..workflows.truncation import TruncationWorkflow
+
     client = await get_client()
 
-    await client.execute_workflow(
-        "TruncationWorkflow",
+    await client.start_workflow(
+        TruncationWorkflow.run,
         args=[str(developer_id), str(session_id), token_count_threshold],
         task_queue="memory-task-queue",
         id=str(job_id),
     )
 
 
 async def run_summarization_task(session_id: UUID, job_id: UUID):
+    from ..workflows.summarization import SummarizationWorkflow
+
     client = await get_client()
 
-    await client.execute_workflow(
-        "SummarizationWorkflow",
+    await client.start_workflow(
+        SummarizationWorkflow.run,
         args=[str(session_id)],
         task_queue="memory-task-queue",
         id=str(job_id),