Update app.py

add a joke tool

app.py

@@ -4,6 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from urllib.parse import quote
 
 from Gradio_UI import GradioUI
 
@@ -18,6 +19,78 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def get_relevant_joke(keyword: str = None) -> str:
+    """
+    Fetches a safe joke from the JokeAPI (v2.jokeapi.dev). 
+    It can return either a single-part or a two-part joke.
+
+    The joke can be optionally filtered by a keyword to provide some relevance.
+    If no keyword is provided, a random joke from the "Any" category is returned.
+
+    Args:
+        keyword: An optional keyword to search for within the joke text.
+
+    Returns:
+        A string containing the joke, or an error message if the fetch fails.
+    """
+    base_url = "https://v2.jokeapi.dev/joke/Any"
+    
+    # Removed 'type': 'single' to allow for both single and two-part jokes.
+    # We still enforce safe-mode for safety.
+    params = {
+        'safe-mode': ''
+    }
+
+    if keyword:
+        # URL-encode the keyword to handle spaces and special characters
+        safe_keyword = quote(keyword)
+        # The API uses the 'contains' parameter for keyword filtering
+        params['contains'] = safe_keyword
+
+    # We implement a simple retry mechanism with exponential backoff for robustness
+    max_retries = 3
+    for attempt in range(max_retries):
+        try:
+            # Make the API request
+            response = requests.get(base_url, params=params, timeout=10)
+            
+            # Raise an exception for bad status codes (4xx or 5xx)
+            response.raise_for_status()
+
+            data = response.json()
+
+            # The JokeAPI returns a 'success' field implicitly, but checks for 'error'
+            if data.get('error') is True:
+                # If the API returns an error, it often means no joke was found for the keyword
+                return (f"**[Joke Tool Error]** Sorry, I couldn't find a joke about '{keyword}'! "
+                        f"Perhaps try a more general term. (API Message: {data.get('message', 'Unknown error')})")
+            
+            # --- Logic to handle both single and two-part jokes ---
+            if data.get('type') == 'single':
+                # Case 1: Single-part joke
+                return data.get('joke')
+            elif data.get('type') == 'twopart':
+                # Case 2: Two-part joke (combine setup and delivery)
+                setup = data.get('setup', 'Setup missing...')
+                delivery = data.get('delivery', 'Punchline missing...')
+                return f"{setup}\n\n...{delivery}"
+            else:
+                return "The Joke API returned an unknown joke format."
+
+
+        except requests.exceptions.RequestException as e:
+            # Handle connection errors, timeouts, and bad status codes
+            print(f"Attempt {attempt + 1} failed: {e}")
+            if attempt < max_retries - 1:
+                # Wait before retrying (exponential backoff)
+                wait_time = 2 ** attempt
+                time.sleep(wait_time)
+            else:
+                return f"**[Joke Tool Error]** Failed to connect to the Joke API after {max_retries} attempts."
+        except Exception as e:
+            return f"**[Joke Tool Error]** An unexpected error occurred: {e}"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.