Update agent.py

agent.py

@@ -112,24 +112,24 @@ def build_graph(provider: str):
                 temperature=0,
             ),
         )
-  def assistant(state: MessagesState):
-      """ Use the tools to answer the query. you have add,subtract,multiply,divide,web_search,wikipedia_search,arxiv_search tools."""
-      response = llm_with_tools.invoke([system_message]+state["messages"])
-      time.sleep(4) # 무료 티어의 한계
-      return {"messages": state["messages"] + [response]}
-
-
-  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")
-
-  return builder.compile()
+    def assistant(state: MessagesState):
+        """ Use the tools to answer the query. you have add,subtract,multiply,divide,web_search,wikipedia_search,arxiv_search tools."""
+        response = llm_with_tools.invoke([system_message]+state["messages"])
+        #time.sleep(4) # 무료 티어의 한계
+        return {"messages": state["messages"] + [response]}
+
+
+    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")
+
+    return builder.compile()
 
 # test
 if __name__ == "__main__":