diff --git a/tools/planner.py b/tools/planner.py index bd83d34..1150aa5 100644 --- a/tools/planner.py +++ b/tools/planner.py @@ -495,10 +495,10 @@ class PlannerTool(WorkflowTool): # Convert generic status names to planner-specific ones tool_name = self.get_name() status_mapping = { - f"{tool_name}_in_progress": "planning_success", - f"pause_for_{tool_name}": f"pause_for_{tool_name}", # Keep the full tool name for workflow consistency - f"{tool_name}_required": f"{tool_name}_required", # Keep the full tool name for workflow consistency - f"{tool_name}_complete": f"{tool_name}_complete", # Keep the full tool name for workflow consistency + f"{tool_name}_in_progress": "planning_in_progress", + f"pause_for_{tool_name}": f"pause_for_planning", + f"{tool_name}_required": f"planning_required", + f"{tool_name}_complete": "planning_complete", } if response_data["status"] in status_mapping: diff --git a/tools/tracer.py b/tools/tracer.py index d701b0f..6f65f70 100644 --- a/tools/tracer.py +++ b/tools/tracer.py @@ -535,9 +535,9 @@ class TracerTool(WorkflowTool): tool_name = self.get_name() status_mapping = { f"{tool_name}_in_progress": "tracing_in_progress", - f"pause_for_{tool_name}": f"pause_for_{tool_name}", - f"{tool_name}_required": f"{tool_name}_required", - f"{tool_name}_complete": f"{tool_name}_complete", + f"pause_for_{tool_name}": f"pause_for_tracing", + f"{tool_name}_required": f"tracing_required", + f"{tool_name}_complete": f"tracing_complete", } if response_data["status"] in status_mapping: