feat(tool): add tool for summing integer lists

ShrewdAgent.py

@@ -13,6 +13,7 @@ from langgraph.pregel import PregelProtocol
 from loguru import logger
 from pydantic import SecretStr
 
+from tools import maths
 from tools.excel_to_text import excel_to_text
 from tools.execute_python_code_from_file import execute_python_code_from_file
 from tools.produce_classifier import produce_classifier
@@ -50,6 +51,7 @@ class ShrewdAgent:
             sort_words_alphabetically,
             excel_to_text,
             execute_python_code_from_file,
+            maths,
         ]
         self.llm = ChatOpenAI(
             model="gpt-4o-mini",

tools/maths.py

@@ -0,0 +1,33 @@
+from typing import List
+
+from langchain_core.tools import tool
+from loguru import logger
+
+
+@tool("add_integers_tool", parse_docstring=True)
+def add_integers(numbers: List[int]) -> int:
+    """
+    Add a list of integers together and return their sum.
+
+    This tool takes a list of integers and calculates their sum. It's useful for
+    performing basic arithmetic operations on multiple numbers at once.
+
+    Args:
+        numbers (List[int]): A list of integers to be summed.
+
+    Returns:
+        int: The sum of all integers in the input list.
+    """
+    logger.info(f"use add_integers with param: {numbers}")
+    if not numbers:
+        raise ValueError("Input list cannot be empty")
+
+    try:
+        return sum(numbers)
+    except TypeError as e:
+        raise TypeError("All elements in the list must be integers") from e
+
+
+if __name__ == "__main__":
+    result = add_integers.invoke({"numbers": [10577, 10037, 10107, 9888, 10089, 10061, 10089, 9917, 10031]})
+    print(result)