updating args

app.py

@@ -19,7 +19,7 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 
 @tool
-def capital_of_country(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def capital_of_country(country: str)-> str:
     """A tool for getting capital of country
     Args:
         country: string of country