Update agent.py

agent.py

@@ -153,31 +153,20 @@ def transcribe_audio(file_path: str, language: str = "en") -> str:
 
 
 @tool
-def multiply(a: int, b: int) -> int:
-    """Multiplies two numbers."""
-    return a * b
+def simple_calculator(operation: str, a: float, b: float) -> float:
+    """Basic maths: add, subtract, multiply, divide and modulo."""
+    if operation == "add":
+        return a + b
+    if operation == "subtract":
+        return a - b
+    if operation == "multiply":
+        return a * b
+    if operation == "divide":
+        return a / b if b else float("inf")
+    if operation == "modulo":
+        return a % b
+    return 0.0
 
-@tool
-def add(a: int, b: int) -> int:
-    """Adds two numbers."""
-    return a + b
-
-@tool
-def subtract(a: int, b: int) -> int:
-    """Subtracts two numbers."""
-    return a - b
-
-@tool
-def divide(a: int, b: int) -> float:
-    """Divides two numbers."""
-    if b == 0:
-        raise ValueError("Cannot divide by zero.")
-    return a / b
-
-@tool
-def modulo(a: int, b: int) -> int:
-    """Returns the remainder of dividing two numbers."""
-    return a % b
 
 @tool
 def wiki_search(query: str) -> str:
@@ -250,8 +239,7 @@ llm = ChatGoogleGenerativeAI(
     model="gemini-2.0-flash",
     google_api_key=GOOGLE_API_KEY,
     temperature=0,
-    max_output_tokens=2048,
-    system_message=system_prompt,
+    max_output_tokens=2048
 )
 
 # === Tools in LLM einbinden ===
@@ -292,11 +280,10 @@ def safe_llm_invoke(messages):
 
 # === Nodes für LangGraph ===
 def assistant(state: MessagesState):
-    """
-    Assistant node mit eingebautem Retry bei Tool-Fehlern.
-    """
-    result_msg = safe_llm_invoke(state["messages"])
-    return {"messages": [result_msg]}
+    msgs = state["messages"]
+    if not msgs or msgs[0].type != "system":
+        msgs = [system_prompt] + msgs
+    return {"messages": [llm_with_tools.invoke(msgs)]}
 
 # === LangGraph bauen ===
 builder = StateGraph(MessagesState)