ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter2/prompt-engineering/tau_bench/envs/airline/tools/calculate.py

@@ -0,0 +1,35 @@
+# Copyright Sierra
+
+from typing import Any, Dict
+from tau_bench.envs.tool import Tool
+
+
+class Calculate(Tool):
+    @staticmethod
+    def invoke(data: Dict[str, Any], expression: str) -> str:
+        if not all(char in "0123456789+-*/(). " for char in expression):
+            return "Error: invalid characters in expression"
+        try:
+            return str(round(float(eval(expression, {"__builtins__": None}, {})), 2))
+        except Exception as e:
+            return f"Error: {e}"
+
+    @staticmethod
+    def get_info() -> Dict[str, Any]:
+        return {
+            "type": "function",
+            "function": {
+                "name": "calculate",
+                "description": "Calculate the result of a mathematical expression.",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "expression": {
+                            "type": "string",
+                            "description": "The mathematical expression to calculate, such as '2 + 2'. The expression can contain numbers, operators (+, -, *, /), parentheses, and spaces.",
+                        },
+                    },
+                    "required": ["expression"],
+                },
+            },
+        }