Update agent.py

UPDATE THE SYSTEM

agent.py

@@ -173,8 +173,7 @@ def build_graph(provider: str = "groq"):
     # Bind tools to LLM
     llm_with_tools = llm.bind_tools(tools)
 
-    # Node
-     # Node
+    
     # Node
     def assistant(state: MessagesState):
         """Assistant node"""
@@ -192,12 +191,10 @@ def build_graph(provider: str = "groq"):
         tools_condition,
         {
             "tools": "tools",
-            None: "END"
+            None: END
         }
     )
     builder.add_edge("tools", "assistant")
 
     # Compile graph
-    return builder.compile()
-
-
+    return builder.compile()