Update agent.py

agent.py

@@ -75,6 +75,19 @@ def duckduckgo_search(query: str) -> str:
         return f"An error occurred during DuckDuckGo search: {e}"
         
 
+@tool
+def arithmetic(expression: str) -> str:
+    """执行数学计算并返回结果。支持基本的算术运算如加减乘除、幂运算等。"""
+    try:
+        # 使用Python的eval函数安全地计算表达式
+        # 限制只能使用基本算术运算，不允许导入模块或执行其他危险操作
+        allowed_names = {"__builtins__": {}}
+        allowed_symbols = {}
+        result = eval(expression, allowed_names, allowed_symbols)
+        return str(result)
+    except Exception as e:
+        return f"计算表达式时出错: {e}"
+        
 # load the system prompt from the file
 # Ensure this file exists and has the content from Step 2
 with open("system_prompt.txt", "r", encoding="utf-8") as f:
@@ -84,7 +97,8 @@ sys_msg = SystemMessage(content=system_prompt)
 tools = [
     wiki_search,
     duckduckgo_search,
-    web_search, # *** ADDED TAVILY WEB SEARCH ***
+    web_search,
+    arithmetic,
 ]