Add solve_equation tool

app.py

@@ -4,19 +4,68 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import re
+from typing import List, Union
+import sympy as sp
+from sympy.core.expr import Expr
 
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+def solve_equation(equation_str: str) -> List[Union[Expr, float]]:
+    """
+    A tool for solving single-variable equation provided as a string.
+    
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        equation_str: A string representing the equation in the format 
+                      "<expression> = <expression>". Expressions can include numbers, 
+                      variables (e.g., 'x'), and arithmetic operators (+, -, *, /). 
+                      Exponentiation can be specified using Python's '**' operator or the 
+                      LaTeX-style '^{...}' (e.g., "x^{1/11}" is interpreted as x**(1/11)).
+    
+    Returns:
+        A list of solutions (each a sympy.Expr or float) for the variable found in the equation.
+    
+    Raises:
+        ValueError: If the equation does not contain an '=' sign or if no variable is found.
+    
+    Example:
+        >>> eq_str = "x^{1/11} + (x^{10/11} / x) = 4.25"
+        >>> solutions = solve_equation(eq_str)
+        >>> for sol in solutions:
+        ...     print(sol)
     """
-    return "What magic will you build ?"
+    # Convert LaTeX-style exponentiation (e.g., x^{1/11}) to Python-style (x**(1/11))
+    equation_str = re.sub(r'\^\{\s*([^}]*)\s*\}', r'**(\1)', equation_str)
+    # Replace any remaining '^' with '**'
+    equation_str = equation_str.replace('^', '**')
+    
+    # Split the string into left-hand side and right-hand side
+    if '=' not in equation_str:
+        raise ValueError("The equation must contain an '=' sign.")
+    lhs_str, rhs_str = equation_str.split('=')
+    
+    # Convert the string expressions to sympy expressions
+    lhs: Expr = sp.sympify(lhs_str)
+    rhs: Expr = sp.sympify(rhs_str)
+    
+    # Form the equation lhs == rhs
+    eq = sp.Eq(lhs, rhs)
+    
+    # Determine the free symbols (variables) in the equation
+    variables = list(eq.free_symbols)
+    if not variables:
+        raise ValueError("No variable found in the equation.")
+    elif len(variables) > 1:
+        # For multiple variables, the first one is chosen for solving.
+        var = variables[0]
+    else:
+        var = variables[0]
+    
+    # Solve the equation for the chosen variable
+    solutions: List[Union[Expr, float]] = sp.solve(eq, var)
+    return solutions
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +104,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, solve_equation], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,