add new tools

agent_smolagent.py

@@ -14,20 +14,7 @@ from smolagents.default_tools import (DuckDuckGoSearchTool,
                                       SpeechToTextTool,
                                       PythonInterpreterTool)
 #from final_answer import FinalAnswerTool, check_reasoning, ensure_formatting
-from tools_smolagent import (read_file, 
-                             extract_text_from_image, 
-                             analyze_csv_file, 
-                             analyze_excel_file, 
-                             youtube_transcribe, 
-                             transcribe_audio, 
-                             wikipedia_search)
-
-#(use_vision_model, youtube_frames_to_images,
-#                         read_file, 
-#                         extract_text_from_image, analyze_csv_file, 
-#                         analyze_excel_file, youtube_transcribe,
-#                         transcribe_audio, review_youtube_video)
-#from model_provider import create_react_model
+from tools_smolagent import level1_tools
 
 load_dotenv()
 
@@ -62,17 +49,7 @@ def build_agent():
             VisitWebpageTool(max_output_length=500000), 
             #WikipediaSearchTool(extract_format='HTML'),
             SpeechToTextTool(),
-            read_file, 
-            extract_text_from_image, 
-            analyze_csv_file, 
-            analyze_excel_file, 
-            youtube_transcribe,
-            transcribe_audio,
-            wikipedia_search
-            #youtube_transcribe,
-            #    use_vision_model,
-            #    youtube_frames_to_images,
-            #transcribe_audio,
+            level1_tools,
             ],
         managed_agents=[],
         additional_authorized_imports=['os', 'pandas', 'numpy', 'PIL', 'tempfile', 'PIL.Image'],

tools_smolagent.py

@@ -346,4 +346,118 @@ def wikipedia_search(query: str) -> dict:
         return {"wiki_results": formatted}
     
     except Exception as e:
-        return {"wiki_results": f"Error during Wikipedia search: {str(e)}"}
+        return {"wiki_results": f"Error during Wikipedia search: {str(e)}"}
+    
+    
+#Mathematical tools
+@tool
+def multiply(a: float, b: float) -> float:
+    """Multiply two numbers.
+    Args:
+        a: first number
+        b: second number
+    Returns: 
+        Multiplication result
+    """
+    return a * b
+
+
+@tool
+def add(a: float, b: float) -> float:
+    """Add two numbers.
+    Args:
+        a: first number
+        b: second number
+    Returns: 
+        Addition result
+    """
+    return a + b
+
+
+@tool
+def subtract(a: float, b: float) -> float:
+    """Subtract two numbers.
+    Args:
+        a: first number
+        b: second number
+    Returns: 
+        Subtraction result
+    """
+    return a - b
+
+
+@tool
+def divide(a: float, b: float) -> float:
+    """Divide two numbers.
+    Args:
+        a: first number
+        b: second number
+    Returns: 
+        Division result
+    """
+    if b == 0:
+        raise ValueError("Cannot divide by zero.")
+    return a / b
+
+
+@tool
+def modulus(a: int, b: int) -> int:
+    """Get the modulus of two numbers.
+    Args:
+        a: first number
+        b: second number
+    Returns:
+        Modulus result
+    """
+    return a % b
+
+from langchain_core.tools import tool
+
+
+@tool
+def convert_units(value: float, from_unit: str, to_unit: str) -> float:
+    """
+    Converts a value from one unit to another.
+
+    Args:
+        value: The numerical value to convert.
+        from_unit: The original unit (e.g. 'miles', 'kg', 'celsius').
+        to_unit: The target unit (e.g. 'kilometers', 'lb', 'fahrenheit').
+
+    Supported conversions:
+        - miles <-> kilometers
+        - kilograms <-> pounds
+        - celsius <-> fahrenheit
+
+    Returns:
+        The converted value result.
+    """
+    conversions = {
+        ("miles", "kilometers"): lambda v: v * 1.60934,
+        ("kilometers", "miles"): lambda v: v / 1.60934,
+        ("kilograms", "pounds"): lambda v: v * 2.20462,
+        ("pounds", "kilograms"): lambda v: v / 2.20462,
+        ("celsius", "fahrenheit"): lambda v: (v * 9/5) + 32,
+        ("fahrenheit", "celsius"): lambda v: (v - 32) * 5/9,
+    }
+
+    key = (from_unit.lower(), to_unit.lower())
+    if key not in conversions:
+        raise ValueError(f"Conversion from {from_unit} to {to_unit} not supported.")
+
+    return conversions[key](value)
+
+level1_tools = [
+    multiply,
+    add,
+    subtract,
+    divide,
+    modulus,
+    read_file, 
+    extract_text_from_image, 
+    analyze_csv_file, 
+    analyze_excel_file, 
+    youtube_transcribe, 
+    transcribe_audio, 
+    wikipedia_search
+]