From be7471b771061fb07b0ebdd9cb0f0d522204e74b Mon Sep 17 00:00:00 2001 From: Diwank Tomer Date: Fri, 16 Aug 2024 16:20:50 -0400 Subject: [PATCH] wip Signed-off-by: Diwank Tomer --- agents-api/agents_api/clients/temporal.py | 3 ++- agents-api/pyproject.toml | 5 ++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/agents-api/agents_api/clients/temporal.py b/agents-api/agents_api/clients/temporal.py index 2130c3947..61f53c71c 100644 --- 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 index a80350a5c..88068f962 100644 --- 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" \ No newline at end of file