From 3debef578143251618625c45ed14b1530373b5e6 Mon Sep 17 00:00:00 2001 From: Laine Date: Wed, 19 Mar 2025 12:48:49 -0400 Subject: [PATCH] Add clarification re: format of start and end dates, and (probably?) fix non-determinism error caused by SHOW_CONFIRM --- tools/tool_registry.py | 9 +++++---- workflows/agent_goal_workflow.py | 8 ++++---- 2 files changed, 9 insertions(+), 8 deletions(-) diff --git a/tools/tool_registry.py b/tools/tool_registry.py index 79a4225..9886435 100644 --- a/tools/tool_registry.py +++ b/tools/tool_registry.py @@ -50,6 +50,7 @@ guess_location_tool = ToolDefinition( ), ], ) + # ----- Travel use cases tools ----- search_flights_tool = ToolDefinition( name="SearchFlights", @@ -198,12 +199,12 @@ future_pto_calc_tool = ToolDefinition( ToolArgument( name="start_date", type="string", - description="Start date of proposed PTO", + description="Start date of proposed PTO, sent in the form yyyy-mm-dd", ), ToolArgument( name="end_date", type="string", - description="End date of proposed PTO", + description="End date of proposed PTO, sent in the form yyyy-mm-dd", ), ToolArgument( name="email", @@ -221,12 +222,12 @@ book_pto_tool = ToolDefinition( ToolArgument( name="start_date", type="string", - description="Start date of proposed PTO", + description="Start date of proposed PTO, sent in the form yyyy-mm-dd", ), ToolArgument( name="end_date", type="string", - description="End date of proposed PTO", + description="End date of proposed PTO, sent in the form yyyy-mm-dd", ), ToolArgument( name="email", diff --git a/workflows/agent_goal_workflow.py b/workflows/agent_goal_workflow.py index 4699e86..86324ff 100644 --- a/workflows/agent_goal_workflow.py +++ b/workflows/agent_goal_workflow.py @@ -26,11 +26,11 @@ with workflow.unsafe.imports_passed_through(): # Constants MAX_TURNS_BEFORE_CONTINUE = 250 -SHOW_CONFIRM = True show_confirm_env = os.getenv("SHOW_CONFIRM") -if show_confirm_env is not None: - if show_confirm_env == "False": - SHOW_CONFIRM = False +if show_confirm_env is not None and show_confirm_env.lower() == "false": + SHOW_CONFIRM = False +else: + SHOW_CONFIRM = True #ToolData as part of the workflow is what's accessible to the UI - see LLMResponse.jsx for example class ToolData(TypedDict, total=False):