Fix hitl for fastapi langgraph

Author: maxkorp

typescript-sdk/integrations/langgraph/python/ag_ui_langgraph/examples/agents/human_in_the_loop.py

@@ -19,7 +19,7 @@
 DEFINE_TASK_TOOL = {
     "type": "function",
     "function": {
-        "name": "generate_task_steps",
+        "name": "plan_execution_steps",
         "description": "Make up 10 steps (only a couple of words per step) that are required for a task. The step should be in imperative form (i.e. Dig hole, Open door, ...)",
         "parameters": {
             "type": "object",
@@ -79,7 +79,7 @@ async def chat_node(state: Dict[str, Any], config: RunnableConfig):
     """
     system_prompt = """
     You are a helpful assistant that can perform any task.
-    You MUST call the `generate_task_steps` function when the user asks you to perform a task.
+    You MUST call the `plan_execution_steps` function when the user asks you to perform a task.
     Always make sure you will provide tasks based on the user query
     """
 
@@ -93,7 +93,7 @@ async def chat_node(state: Dict[str, Any], config: RunnableConfig):
     # Use "predict_state" metadata to set up streaming for the write_document tool
     config["metadata"]["predict_state"] = [{
         "state_key": "steps",
-        "tool": "generate_task_steps",
+        "tool": "plan_execution_steps",
         "tool_argument": "steps"
     }]
 
@@ -130,7 +130,7 @@ async def chat_node(state: Dict[str, Any], config: RunnableConfig):
             args = tool_call.get("args", {})
             tool_call_args = args if not isinstance(args, str) else json.loads(args)
 
-        if tool_call_name == "generate_task_steps":
+        if tool_call_name == "plan_execution_steps":
             # Get the steps from the tool call
             steps_raw = tool_call_args.get("steps", [])
 
@@ -181,7 +181,7 @@ async def chat_node(state: Dict[str, Any], config: RunnableConfig):
                 }
             )
 
-    # If no tool calls or not generate_task_steps, return to END with the updated messages
+    # If no tool calls or not plan_execution_steps, return to END with the updated messages
     return Command(
         goto=END,
         update={