Update agent.py

agent.py

@@ -157,10 +157,17 @@ def planner(state: MessagesState):
     msgs = state["messages"]
     if msgs[0].type != "system":
         msgs = [SYSTEM_PROMPT] + msgs
+
     resp = with_backoff(lambda: gemini_llm.invoke(msgs))
-    content = resp.content.strip()
-    finished = not getattr(resp, "tool_calls", None) and "\n" not in content
-    return {"messages": msgs + [resp], "should_end": finished}
+
+    # WICHTIG: Gib tool_calls weiter – sie lösen im ToolNode die Ausführung aus
+    return {
+        "messages": msgs + [resp],
+        "should_end": (
+            not getattr(resp, "tool_calls", None)  # kein Tool gewünscht
+            and "\n" not in resp.content           # einfache Heuristik
+        )
+    }
 
 def route(state):
     return "END" if state["should_end"] else "tools"
@@ -169,10 +176,17 @@ def route(state):
 TOOLS = [web_search, wiki_search, parse_csv, parse_excel, python_repl]
 
 graph = StateGraph(MessagesState)
+
 graph.add_node("planner", planner)
 graph.add_node("tools", ToolNode(TOOLS))
+
 graph.add_edge(START, "planner")
-graph.add_conditional_edges("planner", route, {"tools": "tools", "END": END})
+graph.add_edge("tools", "planner")  # 🔁 Rücksprung zum Planner nach Tool-Ausführung
+
+graph.add_conditional_edges("planner", route, {
+    "tools": "tools",
+    "END": END,
+})
 
 # compile → LangGraph-Executor
 agent_executor = graph.compile()