From 87e22d336057e8e0f56b420b84a6945e8e60b0ee Mon Sep 17 00:00:00 2001 From: Rob Holland Date: Thu, 13 Feb 2025 13:23:06 +0000 Subject: [PATCH] Use legacy queue for both train APIs. --- workflows/tool_workflow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/workflows/tool_workflow.py b/workflows/tool_workflow.py index 4a4aa7b..97893e3 100644 --- a/workflows/tool_workflow.py +++ b/workflows/tool_workflow.py @@ -49,7 +49,7 @@ class ToolWorkflow: """Execute a tool after confirmation and handle its result.""" workflow.logger.info(f"Confirmed. Proceeding with tool: {current_tool}") - task_queue = TEMPORAL_LEGACY_TASK_QUEUE if current_tool == "BookTrain" else None + task_queue = TEMPORAL_LEGACY_TASK_QUEUE if current_tool in ["SearchTrain", "BookTrain"] else None try: dynamic_result = await workflow.execute_activity(