Update app.py

app.py

@@ -23,11 +23,10 @@ def get_converted_speed(unit:str, speed:float)-> float: #it's import to specify
     elif unit == "mph": 
         converted_speed = speed / factor
     else:
-        return f"Error converting for {speed}[{unit}]" 
+        raise ValueError(f"Invalid unit: {unit}. Please use 'kph' or 'mph'.")
 
     return converted_speed
 
-
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -49,7 +48,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, get_converted_speed], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,