恢复工具调用

agent.py

@@ -202,18 +202,18 @@ def build_graph(tag: str='huggingface'):
 
     builder = StateGraph(MessagesState)
     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_conditional_edges(
-    #     'assistant',
-    #     tools_condition
-    # )
-    # builder.add_edge('tools', 'assistant')
-
-    builder.set_entry_point("retriever")
-    builder.set_finish_point("retriever")
+    builder.add_node('assistant', assistant)
+    builder.add_node('tools', ToolNode(tools))
+    builder.add_edge(START, 'retriever')
+    builder.add_edge('retriever', 'assistant')
+    builder.add_conditional_edges(
+        'assistant',
+        tools_condition
+    )
+    builder.add_edge('tools', 'assistant')
+
+    # builder.set_entry_point("retriever")
+    # builder.set_finish_point("retriever")
 
     return builder.compile()