Update app.py

app.py

@@ -33,6 +33,31 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def convert_units(value: float, from_unit: str, to_unit: str) -> str:
+    """Converts common units (length, weight, temperature).
+    Args:
+        value: The numerical value to convert
+        from_unit: The original unit (e.g., 'km', 'kg', 'C')
+        to_unit: The target unit (e.g., 'miles', 'lb', 'F')
+    """
+    conversions - {
+        ('km', 'miles'): lambda x: x * 0.621371,
+        ('miles', 'km'): lambda x: x / 0.621371,
+        ('kg', 'lb'): lambda x: x * 2.20462,
+        ('lb', 'kg'): lambda x: x / 2.20462,
+        ('C', 'F'): lambda x: (x * 9/5) + 32,
+        ('F', 'C'): lambda x: (x - 32) * 5/9
+    }
+    # dictionary which maps (from_unit, to_unit) tuples to simple math functions (lambda) that perform the actual conversion
+    try:
+        func = conversions[(from_unit.lower(), to_unit.lower())]
+        result = func(value)
+        return f"(value) {from_unit} = {result:.2f} {to_unit}"
+    except KeyError:
+        return f"Unsupported conversion from {from_unit} to {to_unit}."
+    except Exception as e:
+        return f"An unexpected error occurred: {str(e)}"
 
 final_answer = FinalAnswerTool()
 
@@ -55,7 +80,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, convert_units], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,