added pokedex access

app.py

@@ -38,20 +38,64 @@ final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
-model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
+model_id='mistralai/Mistral-7B-instruct-v0.3',# it is possible that this model may be overloaded
 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
+    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)}"
+    return pokemon_info
+
+web_search_tool = DuckDuckGoSearchTool()
 
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_pokemon_information, web_search_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,