Create tools.py

tools.py

@@ -0,0 +1,87 @@
+from smolagents import tool
+import re
+import math
+
+@tool
+def calculator(expression: str) -> str:
+    """Evaluate a mathematical expression safely.
+    
+    Args:
+        expression: A mathematical expression (e.g., '2 + 2 * 3', 'sqrt(16)')
+    
+    Returns:
+        The result of the evaluation
+    """
+    try:
+        # Clean the expression - remove any non-math characters for safety
+        # Allow only numbers, basic operators, parentheses, and math functions
+        safe_chars = r'[\d\s\+\-\*\/\(\)\.\^]|sqrt|pow|log|sin|cos|tan'
+        
+        # Extract safe parts
+        safe_parts = re.findall(safe_chars, expression.lower())
+        safe_expression = ''.join(safe_parts)
+        
+        if not safe_expression:
+            return "Error: No valid mathematical expression found"
+        
+        # Replace ^ with ** for exponentiation
+        safe_expression = safe_expression.replace('^', '**')
+        
+        # Define safe functions
+        safe_dict = {
+            'sqrt': math.sqrt,
+            'pow': math.pow,
+            'log': math.log,
+            'log10': math.log10,
+            'sin': math.sin,
+            'cos': math.cos,
+            'tan': math.tan,
+            'pi': math.pi,
+            'e': math.e
+        }
+        
+        # Evaluate safely using eval with limited globals
+        result = eval(safe_expression, {"__builtins__": {}}, safe_dict)
+        
+        return f"Result of '{expression}': {result}"
+    
+    except ZeroDivisionError:
+        return "Error: Division by zero"
+    except Exception as e:
+        return f"Error evaluating expression: {str(e)}. Try: '2 + 2', '3 * 5', 'sqrt(16)', '2 ^ 3'"
+
+@tool
+def get_hf_dataset_info(dataset_name: str) -> str:
+    """Get information about a Hugging Face dataset.
+    
+    Args:
+        dataset_name: Name of the dataset (e.g., 'mnist', 'glue')
+    
+    Returns:
+        Dataset information
+    """
+    try:
+        from huggingface_hub import get_dataset_info
+        
+        # Handle common dataset names
+        if dataset_name.lower() == "mnist":
+            dataset_name = "mnist"
+        elif dataset_name.lower() == "glue":
+            dataset_name = "glue"
+        
+        info = get_dataset_info(dataset_name)
+        
+        response = f"📊 **Dataset: {info.id}**\n\n"
+        response += f"**Description:** {info.description[:300]}...\n\n"
+        response += f"**Downloads:** {info.downloads:,}\n"
+        response += f"**Last Modified:** {info.lastModified}\n"
+        response += f"**License:** {info.cardData.get('license', 'Not specified')}\n"
+        
+        # Show dataset size if available
+        if hasattr(info, 'size') and info.size:
+            response += f"**Size:** {info.size}\n"
+        
+        return response
+        
+    except Exception as e:
+        return f"Error getting dataset info: {str(e)}\n\nTry: 'mnist', 'glue', or other Hugging Face dataset names"