Update agent.py

agent.py

@@ -152,31 +152,14 @@ SYSTEM_PROMPT = SystemMessage(
 # ---------------------------------------------------------------------
 # 7) LangGraph – Planner + Tools + Router
 # ---------------------------------------------------------------------
-def extract_tool_call(text: str) -> tuple[str, str] | None:
-    """Parse Gemini output like: 'Tool: xyz\nInput: abc'."""
-    match = re.search(r"Tool:\s*(\w+)\s*Input:\s*(.+)", text, re.DOTALL)
-    if match:
-        return match.group(1).strip(), match.group(2).strip()
-    return None
 
 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()
-
-    parsed = extract_tool_call(content)
-    if parsed:
-        tool_name, tool_input = parsed
-        tool = {t.name: t for t in TOOLS}.get(tool_name)
-        if tool:
-            result = tool.invoke(tool_input)
-            new_msg = HumanMessage(content=f"Tool result:\n{result}")
-            return {"messages": msgs + [resp, new_msg], "should_end": False}
-
-    finished = "\n" not in content  # einfache Heuristik
+    finished = not getattr(resp, "tool_calls", None) and "\n" not in content
     return {"messages": msgs + [resp], "should_end": finished}
 
 def route(state):