Customize Agent Tools

Added:
search_tool,
get_current_time_in_timezone,
native_language_translator,
weather_history,
weather_forecast,
suissknife_advisor_tool

app.py

@@ -18,6 +18,22 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def suissknife_advisor_tool()->str:
+    """A tool that returns an indication of what to do at the next step - it is useful in cases where there is no other suitable tool at the user's disposal and the user does not know what he wants
+    May be used prior final answer to remember about additional steps
+    Args: (no args)    
+    """
+    advice = '''
+    Try to undertand from context what is country or region user from. 
+    Translate your answer to his state or region language (if this is not clear then use `be` translation as a target translation for your answer. 
+    If you do not understand for sure what exacly user wants then try to search for a funny or encourage story about Mother Language Day
+    Duplicate your final answer with `en` and `bn` language translation for the case when user mentonied any stuff outside Europe 
+    Use `native_language_translator` to translate your final answer to the target language  
+    
+    '''
+    return advice
+    
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -34,6 +50,86 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+@tool
+def weather_history(lat: float, lon: float, start: str, end: str) -> str:
+    """
+    Fetches historical weather data for a given location and date range using the Meteostat library.
+    
+    Args:
+        lat: Latitude of the location.
+        lon: Longitude of the location.
+        start: Start date in YYYY-MM-DD format.
+        end: End date in YYYY-MM-DD format.
+        
+    Returns:
+        A JSON string containing the historical weather data or an error message.
+    """
+    from meteostat import Point, Daily
+    try:
+        # Parse the date strings into datetime objects
+        start_date = datetime.strptime(start, "%Y-%m-%d")
+        end_date = datetime.strptime(end, "%Y-%m-%d")
+        
+        # Define the location using latitude and longitude
+        location = Point(lat, lon)
+        
+        # Fetch daily historical data for the location between start_date and end_date
+        data = Daily(location, start_date, end_date)
+        data = data.fetch()
+        
+        # Return the data as a JSON string
+        return data.to_json()
+    except Exception as e:
+        return f"Error fetching weather history: {str(e)}"
+
+
+@tool
+def weather_forecast(lat: float, lon: float) -> str:
+    """
+    Fetches weather forecast data for the next 7 days for a given location using the Open-Meteo API.
+    
+    Args:
+        lat: Latitude of the location.
+        lon: Longitude of the location.
+        
+    Returns:
+        A JSON string containing the weather forecast data or an error message.
+    """
+    try:
+        # Construct the API URL with daily forecast for max and min temperatures.
+        url = (
+            f"https://api.open-meteo.com/v1/forecast?"
+            f"latitude={lat}&longitude={lon}"
+            f"&daily=temperature_2m_max,temperature_2m_min,precipitation_sum"
+            f"&timezone=auto"
+        )
+        
+        # Request the forecast data
+        response = requests.get(url)
+        if response.status_code == 200:
+            return response.text
+        else:
+            return f"Error fetching forecast: {response.status_code}"
+    except Exception as e:
+        return f"Error fetching forecast: {str(e)}"
+
+
+@tool
+def native_language_translator(text: str, target_lang: str) -> str:
+    """Translates text to a specified native language
+    Args:
+        text: Input text to translate
+        target_lang: Target language code (e.g., 'be', 'es', 'fr', 'zh')
+    """
+    try:
+        from googletrans import Translator
+        translator = Translator()
+        translation = translator.translate(text, dest=target_lang)
+        return f"Translated to {target_lang}: {translation.text}"
+    except Exception as e:
+        return f"Translation failed: {str(e)}"
+
+search_tool = DuckDuckGoSearchTool()
 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:
@@ -43,6 +139,7 @@ model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
 model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+fallback_model_id='meta-llama/Meta-Llama-3-70B-Instruct',  # Fallback model
 custom_role_conversions=None,
 )
 
@@ -55,7 +152,15 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[
+        search_tool,
+        get_current_time_in_timezone,
+        native_language_translator,
+        weather_history,
+        weather_forecast,
+        suissknife_advisor_tool,
+        final_answer,
+    ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,