Update app.py

app.py

@@ -53,7 +53,7 @@ def agent_node(state: OrderState) -> OrderState:
 
     # Ensure we always have at least a system message
     if not state.get("messages", []):
-        return defaults | state | {"messages": []}
+        return defaults | state | {"messages": [SystemMessage(content=SYSINT), new_output]}
 
     try:
         # Prepend system instruction if not already present
@@ -69,6 +69,34 @@ def agent_node(state: OrderState) -> OrderState:
         # Fallback if LLM processing fails
         return defaults | state | {"messages": [AIMessage(content=f"I'm having trouble processing that. {str(e)}")]}
 
+def interactive_tools_node(state: OrderState) -> OrderState:
+    """Handles interactive tool calls."""
+    logging.info("order node")
+    tool_msg = state.get("messages", [])[-1]
+    order = state.get("order", [])
+    outbound_msgs = []
+
+    for tool_call in tool_msg.tool_calls:
+        tool_name = tool_call["name"]
+        tool_args = tool_call["args"]
+
+        if tool_name == "wikipedia_search":
+            page = wikipedia.page(title)
+            response = page.content[:300]
+
+        else:
+            raise NotImplementedError(f'Unknown tool call: {tool_name}')
+
+        outbound_msgs.append(
+            ToolMessage(
+                content=response,
+                name=tool_name,
+                tool_call_id=tool_call["id"],
+            )
+        )
+
+    return {"messages": outbound_msgs, "order": order, "finished": False}
+    
 def maybe_route_to_tools(state: OrderState) -> str:
     """Route between chat and tool nodes."""
     if not (msgs := state.get("messages", [])):
@@ -84,8 +112,11 @@ def maybe_route_to_tools(state: OrderState) -> str:
         if any(tool["name"] in tool_node.tools_by_name.keys() for tool in msg.tool_calls):
             print("from agent GOTO tools node")
             return "tools"
+        else:
+            logging.info("from chatbot GOTO order node")
+            return "interactive_tools"
 
-    print("tool call failed, letting agent try again")
+    print("tool call failed, quitting")
     return "human"
 
 def human_node(state: OrderState) -> OrderState:
@@ -127,11 +158,13 @@ graph_builder = StateGraph(OrderState)
 graph_builder.add_node("agent", agent_node)
 graph_builder.add_node("human", human_node)
 graph_builder.add_node("tools", tool_node)
+graph_builder.add_node("interactive_tools", interactive_tools_node)
 
 # Add edges and routing
 graph_builder.add_conditional_edges("agent", maybe_route_to_tools)
 graph_builder.add_conditional_edges("human", maybe_exit_human_node)
 graph_builder.add_edge("tools", "agent")
+graph_builder.add_edge("interactive_tools", "agent")
 graph_builder.add_edge(START, "human")
 
 # Compile the graph