Update app.py

app.py

@@ -37,21 +37,41 @@ if openai_api_key and tavily_api_key:
         st.sidebar.write("Agent Response:", response)
         return {"messages": [response]}
 
+    # Tool Execution Logic
+    def ToolExecutor(state: State):
+        last_message = state["messages"][-1]
+        if "tool_calls" in last_message.additional_kwargs:
+            tool_call = last_message.additional_kwargs["tool_calls"][0]
+            tool_name = tool_call["function"]["name"]
+            tool_args = tool_call["function"]["arguments"]
+
+            st.sidebar.write("Tool Call Detected:", tool_name, tool_args)
+
+            # Execute the tool with provided arguments
+            if tool_name == "tavily_search_results_json":
+                query = eval(tool_args)["query"]  # Convert stringified arguments to dict
+                tool_response = tool.invoke({"query": query})
+                st.sidebar.write("Tool Response:", tool_response)
+
+                # Return the tool response in a new ToolMessage
+                return {"messages": [ToolMessage(content=str(tool_response), tool_call_id=tool_call["id"])]}
+
+        return state  # Fallback
+
     # Memory Checkpoint
     memory = MemorySaver()
 
     # Build the Graph
     graph = StateGraph(State)
-    tool_node = ToolNode(tools=[tool])
-
     graph.add_node("Agent", Agent)
-    graph.add_node("tools", tool_node)
-    graph.add_conditional_edges("Agent", tools_condition)
-    graph.add_edge("tools", "Agent")
+    graph.add_node("ToolExecutor", ToolExecutor)
+
+    graph.add_conditional_edges("Agent", tools_condition, {"True": "ToolExecutor"})
+    graph.add_edge("ToolExecutor", "Agent")
     graph.set_entry_point("Agent")
 
     # Compile Graph
-    app = graph.compile(checkpointer=memory, interrupt_before=["tools"])
+    app = graph.compile(checkpointer=memory, interrupt_before=["ToolExecutor"])
 
     # Display Graph Visualization
     st.subheader("Graph Workflow")