add pokemon tools

app.py

@@ -139,6 +139,73 @@ def search_wikipedia(query: str) -> str:
     except requests.exceptions.RequestException as e:
         return f"Error fetching Wikipedia data: {str(e)}"
 
+@tool
+def get_pokemon_data(pokemon_name: str) -> dict:
+    """Fetches a summarized version of Pokémon data from the PokéAPI.
+
+    Args:
+        pokemon_name: The name of the Pokémon (case insensitive).
+    
+    Returns:
+        A dictionary with relevant Pokémon information: name, ID, types, abilities, height, weight, and sprite.
+    """
+    try:
+        pokemon_name = pokemon_name.lower()
+        response = requests.get(f"https://pokeapi.co/api/v2/pokemon/{pokemon_name}")
+
+        if response.status_code != 200:
+            return {"error": f"Pokémon '{pokemon_name}' not found."}
+
+        data = response.json()
+
+        # Extract relevant information
+        return {
+            "name": data["name"].capitalize(),
+            "id": data["id"],
+            "types": [t["type"]["name"].capitalize() for t in data["types"]],
+            "abilities": [a["ability"]["name"].replace("-", " ").capitalize() for a in data["abilities"]],
+            "height": data["height"] / 10,  # Convert from decimeters to meters
+            "weight": data["weight"] / 10,  # Convert from hectograms to kilograms
+            "sprite": data["sprites"]["front_default"]  # URL of Pokémon image
+        }
+
+    except Exception as e:
+        return {"error": f"Failed to retrieve Pokémon data: {e}"}
+
+
+@tool
+def get_pokemon_move(move_name: str) -> dict:
+    """Fetches a summarized version of Pokémon move data from the PokéAPI.
+
+    Args:
+        move_name: The name of the move (case insensitive).
+    
+    Returns:
+        A dictionary with relevant move information: name, power, accuracy, PP, type, and effect.
+    """
+    try:
+        move_name = move_name.lower().replace(" ", "-")
+        response = requests.get(f"https://pokeapi.co/api/v2/move/{move_name}")
+
+        if response.status_code != 200:
+            return {"error": f"Move '{move_name}' not found."}
+
+        data = response.json()
+
+        # Extract relevant information
+        return {
+            "name": data["name"].capitalize(),
+            "power": data.get("power", "N/A"),  # Some moves have no power
+            "accuracy": data.get("accuracy", "N/A"),  # Some moves have no accuracy
+            "pp": data["pp"],
+            "type": data["type"]["name"].capitalize(),
+            "effect": data["effect_entries"][0]["short_effect"] if data["effect_entries"] else "No effect description available."
+        }
+
+    except Exception as e:
+        return {"error": f"Failed to retrieve move data: {e}"}
+
+
 search_tool = DuckDuckGoSearchTool()
 
 final_answer = FinalAnswerTool()
@@ -147,7 +214,7 @@ final_answer = FinalAnswerTool()
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
 model = HfApiModel(
-max_tokens=2096,
+max_tokens=5024,
 temperature=0.5,
 model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
 custom_role_conversions=None,
@@ -171,7 +238,9 @@ agent = CodeAgent(
         get_joke,
         get_time_in_timezone,
         get_random_fact,
-        search_wikipedia
+        search_wikipedia,
+        get_pokemon_data,
+        get_pokemon_move,
     ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,