Update agent_v2.py

agent_v2.py

@@ -794,10 +794,19 @@ def build_graph(provider: str = "huggingface"):
     builder.add_node("assistant", assistant)
     builder.add_node("tools", ToolNode(tools))
     builder.add_node("extractor", extractor)
+    
     builder.add_edge(START, "assistant")
-    builder.add_conditional_edges("assistant", tools_condition)
+    
+    builder.add_conditional_edges(
+        "assistant",
+        tools_condition,
+        {
+            "tools": "tools",
+            "none": "extractor",  # Only go to extractor when no tool calls exist
+        }
+    )
+    
     builder.add_edge("tools", "assistant")
-    builder.add_edge("assistant", "extractor")
     builder.set_finish_point("extractor")