wip

Signed-off-by: Diwank Tomer <[email protected]>
julep-ai · Aug 16, 2024 · be7471b · be7471b
1 parent a4e49a0
commit be7471b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/agents-api/agents_api/clients/temporal.py b/agents-api/agents_api/clients/temporal.py
@@ -8,6 +8,7 @@
     temporal_client_cert,
     temporal_namespace,
     temporal_private_key,
+    temporal_task_queue,
     temporal_worker_url,
 )
 from ..worker.codec import pydantic_data_converter
@@ -49,6 +50,6 @@ async def run_task_execution_workflow(
     return await client.start_workflow(
         TaskExecutionWorkflow.run,
         args=[execution_input, start, previous_inputs],
-        task_queue="memory-task-queue",
+        task_queue=temporal_task_queue,
         id=str(job_id),
     )
diff --git a/agents-api/pyproject.toml b/agents-api/pyproject.toml
@@ -63,7 +63,6 @@ check = [
     "format",
     "typecheck",
 ]
-test = "ward"
 codegen = """
 datamodel-codegen \
   --input ../openapi.yaml \
@@ -94,3 +93,7 @@ datamodel-codegen \
   --openapi-scopes schemas \
   --keep-model-order \
   --disable-timestamp"""
+
+[tool.poe.tasks.test]
+env = { AGENTS_API_TESTING = "true" }
+cmd = "ward test"