Update agent.py

agent.py

@@ -175,22 +175,29 @@ def build_graph(provider: str = "groq"):
 
     # Node
      # Node
+    # Node
     def assistant(state: MessagesState):
         """Assistant node"""
-        return {"messages": [llm_with_tools.invoke(state["messages"])]}
+        return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])]}
     
-    # Skip the retriever and just use the LLM directly
+    # Build the graph
     builder = StateGraph(MessagesState)
     builder.add_node("assistant", assistant)
     builder.add_node("tools", ToolNode(tools))
     
-    builder.set_entry_point(START)
-    builder.add_edge(START, "assistant")
+    # Set entry point and edges
+    builder.set_entry_point("assistant")
     builder.add_conditional_edges(
         "assistant",
         tools_condition,
+        {
+            "tools": "tools",
+            None: "END"
+        }
     )
     builder.add_edge("tools", "assistant")
 
     # Compile graph
-    return builder.compile()
+    return builder.compile()
+
+