fixed tools

Author: cemde

examples/5_a_day_benchmark/tools/banking.py

@@ -138,12 +138,12 @@ def execute(self, **kwargs) -> ToolResult:
 
 
 class BankingGetAssetTool(BaseTool):
-    """Get asset information by name."""
+    """Get number of shares owned for a stock ticker."""
 
     def __init__(self, banking_state: BankingState):
         super().__init__(
             "banking.get_asset",
-            "Get asset information by asset name (e.g., 'AAPL' for Apple stock shares)",
+            "Get the number of shares you own for a stock ticker symbol (e.g., 'AAPL' returns how many Apple shares you own)",
             tool_args=["asset_name"],
         )
         self.state = banking_state

examples/5_a_day_benchmark/tools/base.py

@@ -29,7 +29,7 @@
     "code": {"type": "string", "description": "Python code to execute", "nullable": True},
     "person": {"type": "string", "description": "Person name", "nullable": True},
     "relationship": {"type": "string", "description": "Relationship type", "nullable": True},
-    "asset_name": {"type": "string", "description": "Asset name (e.g., apple_shares)", "nullable": True},
+    "asset_name": {"type": "string", "description": "Asset ticker symbol (e.g., AAPL, GOOGL)", "nullable": True},
     "symbol": {"type": "string", "description": "Stock symbol (e.g., AAPL)", "nullable": True},
     "ticker": {"type": "string", "description": "Stock ticker symbol", "nullable": True},
     "date": {"type": "string", "description": "Date (YYYY-MM-DD)", "nullable": True},

examples/5_a_day_benchmark/tools/code_execution.py

@@ -23,13 +23,42 @@ class CodeExecutionState:
     def __init__(self, test_cases: list[dict[str, Any]] | None = None):
         self.test_cases = test_cases or []
 
+        # Build safe builtins with common functions needed for algorithm problems
+        safe_builtins = {
+            **limited_builtins,
+            'len': len,
+            'max': max,
+            'min': min,
+            'sum': sum,
+            'abs': abs,
+            'all': all,
+            'any': any,
+            'enumerate': enumerate,
+            'zip': zip,
+            'sorted': sorted,
+            'reversed': reversed,
+            'map': map,
+            'filter': filter,
+            'int': int,
+            'float': float,
+            'str': str,
+            'bool': bool,
+            'dict': dict,
+            'set': set,
+            'list': list,
+            'tuple': tuple,
+            'range': range,
+            'print': print,
+        }
+
         # Safe execution environment with all RestrictedPython guards
         self.safe_env = {
             **safe_globals,
-            "__builtins__": limited_builtins,
+            "__builtins__": safe_builtins,
             "_print_": PrintCollector,
             "_getattr_": getattr,
             "_getitem_": lambda obj, index: obj[index],
+            "_getiter_": iter,
             "_iter_unpack_sequence_": guarded_iter_unpack_sequence,
             "__name__": "restricted_module",
             "__metaclass__": type,