Update agent.py

agent.py

@@ -78,23 +78,23 @@ def build_graph(provider: str = "google"):
     else:
         raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
     
-# Bind tools to LLM
-llm_with_tools = llm.bind_tools(tools)
+    # Bind tools to LLM
+    llm_with_tools = llm.bind_tools(tools)
 
-        # Node
-def assistant(state: MessagesState):
-    """Assistant node"""
-    return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])]}
+    # Node
+    def assistant(state: MessagesState):
+        """Assistant node"""
+        return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])]}
 
-builder = StateGraph(MessagesState)
-builder.add_node("assistant", assistant)
-builder.add_node("tools", ToolNode(tools))
-builder.add_edge(START, "assistant")
-builder.add_conditional_edges(
-    "assistant",
-    tools_condition,
-)
-builder.add_edge("tools", "assistant")
+    builder = StateGraph(MessagesState)
+    builder.add_node("assistant", assistant)
+    builder.add_node("tools", ToolNode(tools))
+    builder.add_edge(START, "assistant")
+    builder.add_conditional_edges(
+        "assistant",
+        tools_condition,
+    )
+    builder.add_edge("tools", "assistant")
 
-# Compile graph
-return builder.compile()
+    # Compile graph
+    return builder.compile()