Update app.py

app.py

@@ -5,7 +5,7 @@ import streamlit as st
 from dotenv import load_dotenv 
 from langchain_groq import ChatGroq 
 from langgraph.graph import StateGraph, START, END
-from langgraph.prebuilt import ToolNode, tools_condition
+# from langgraph.prebuilt import ToolNode, tools_condition
 from langgraph.checkpoint.memory import MemorySaver
 from langchain_core.messages import AIMessage, HumanMessage
 from langgraph.graph.message import add_messages
@@ -41,11 +41,12 @@ def ai_assistance(state: State):
 # Build LangGraph
 builder = StateGraph(State)
 builder.add_node("AI_Assistance", ai_assistance)
-builder.add_node("tools", ToolNode(tools))
+# builder.add_node("tools", ToolNode(tools))
  
 builder.add_edge(START, "AI_Assistance")
-builder.add_conditional_edges("AI_Assistance", tools_condition)
-builder.add_edge("tools", "AI_Assistance")
+# builder.add_conditional_edges("AI_Assistance", tools_condition)
+# builder.add_edge("tools", "AI_Assistance")
+builder.add_edge("AI_Assistance", END)
 graph = builder.compile(checkpointer=memory)
  
 # Streamlit UI setup