pokeinformation returns moves and evolution chains

app.py

@@ -5,6 +5,7 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 import os
+import pokebase as pb
 
 from Gradio_UI import GradioUI
 
@@ -46,46 +47,64 @@ custom_role_conversions=None,
 
 @tool
 def get_pokemon_information(pokemon_name: str) -> dict:
-    '''
+    """
     This tool gives information about a pokemon.
-    The information given is 
-    Moves
-    Abilities
-    Types
-    Pokémon Evolution Chains.
-    base_experience
-    stats
-    weight
-    height
-    sprites-> image url of the pokemon could be shiny
+    Just pass the name of the pokemon as an argument in the string format.
+
+    The information given is:
+        - Moves
+        - Abilities
+        - Types
+        - Pokémon Evolution Chains
+        - base_experience
+        - stats
+        - weight
+        - height
+
     Use this tool to get the information about a pokemon else use the web search tool.
+
     Args:
         pokemon_name: The name of the pokemon.
-    Returns: dict
-        The information about the pokemon.
-    '''
-    try:
-        response = requests.get(f"https://pokeapi.co/api/v2/pokemon/{pokemon_name}")
-        response.raise_for_status()
-        pokemon_info_json = response.json()
-
-        pokemon_info = {
-            "moves": pokemon_info_json["moves"],
-            "abilities": pokemon_info_json["abilities"],
-            "types": pokemon_info_json["types"],
-            "evolution_chains": pokemon_info_json["evolution_chain"],
-            "base_experience": pokemon_info_json["base_experience"],
-            "stats": pokemon_info_json["stats"],
-            "weight": pokemon_info_json["weight"],
-            "height": pokemon_info_json["height"],
-            "sprites": pokemon_info_json["sprites"],
-            "pokedex_number": pokemon_info_json["id"],
-            "location_area_url": pokemon_info_json["location_area_encounters"]
-        }
-    except requests.exceptions.RequestException as e:
-        return f"Error fetching the Pokemon number: {str(e)}"
-    except Exception as e:
-        return f"An unexpected error occurred: {str(e)}"
+
+    Returns:
+        dict: A dictionary conaining The information about the pokemon like moves, abilities, types, evolution chains, base experience, stats, weight, height, sprites
+    """
+    if type(pokemon_name) != str:
+        return "Please pass the pokemon name as a string"
+    
+    client = pb
+    pokemon = client.pokemon(pokemon_name)
+    if pokemon is None:
+        return "Please pass a valid pokemon name"
+
+
+    evolution_number = client.pokemon_species(pokemon_name).evolution_chain.id
+    e_chain = client.evolution_chain(evolution_number).chain
+
+
+    e2 = "none"
+    e3 = "none"
+    e1 = e_chain.species.name
+    if len(e_chain.evolves_to)>0:
+        e2 = e_chain.evolves_to[0].species.name
+        if len(e_chain.evolves_to[0].evolves_to)>0:
+            e3 = e_chain.evolves_to[0].evolves_to[0].species.name
+
+    chain_str = f"Evolution chain is: {e1} -> {e2} -> {e3}"
+
+        
+    pokemon_info = {
+        "moves": str([move.name for move in pb.move(pokemon_name).results]),
+        "types": str([type.type.name for type in pokemon.types]),
+        "stats": str({ stats.stat.name: pokemon.stats[i].base_stat for i, stats in enumerate(pokemon.stats)}),
+        "expierience": pokemon.base_experience,
+        "species": pokemon.species,
+        "weight": pokemon.weight,
+        "height": pokemon.height,        
+        "abilities": str([ability.ability for ability in pokemon.abilities]),
+        "evolution_chain": chain_str
+    }
+
     return pokemon_info
 
 web_search_tool = DuckDuckGoSearchTool()