fix: Fix arg list

julep-ai · May 16, 2024 · 8cc62d4 · 8cc62d4
1 parent aa808cd
commit 8cc62d4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/agents-api/agents_api/workflows/embed_docs.py b/agents-api/agents_api/workflows/embed_docs.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 
 
+from datetime import timedelta
 from temporalio import workflow
 
 with workflow.unsafe.imports_passed_through():
@@ -13,7 +14,6 @@ class EmbedDocsWorkflow:
     async def run(self, doc_id: str, title: str, content: list[str]) -> None:
         return await workflow.execute_activity(
             embed_docs,
-            doc_id,
-            title,
-            content,
+            args=[doc_id, title, content],
+            schedule_to_close_timeout=timedelta(seconds=600),
         )