Update agent.py

agent.py

@@ -186,11 +186,12 @@ def build_graph(provider: str = "groq"):
         return {"messages": [sys_msg] + state["messages"] + [example_msg]}
 
     builder = StateGraph(MessagesState)
-    builder.add_node("retriever", retriever)
+    # builder.add_node("retriever", retriever)
     builder.add_node("assistant", assistant)
     builder.add_node("tools", ToolNode(tools))
-    builder.add_edge(START, "retriever")
-    builder.add_edge("retriever", "assistant")
+    # builder.add_edge(START, "retriever")
+    # builder.add_edge("retriever", "assistant")
+    builder.add_edge(START, "assistant")
     builder.add_conditional_edges(
         "assistant",
         tools_condition,