|
|
from typing import Any, Optional |
|
|
from smolagents.tools import Tool |
|
|
import requests |
|
|
import markdownify |
|
|
import smolagents |
|
|
import pokebase as pb |
|
|
|
|
|
class GetPokemonInfoTool(Tool): |
|
|
name = "get_pokemon_info" |
|
|
description = """" |
|
|
Retrieves complete information on a given Pokemon. Use this to get information for a pokemon. It will return a python object where you can then find the following object properties, eacho one is described as name, description and type: |
|
|
Name Description Type |
|
|
id The identifier for this resource. integer |
|
|
name The name for this resource. string |
|
|
base_experience The base experience gained for defeating this Pokémon. integer |
|
|
height The height of this Pokémon in decimetres. integer |
|
|
is_default Set for exactly one Pokémon used as the default for each species. boolean |
|
|
order Order for sorting. Almost national order, except families are grouped together. integer |
|
|
weight The weight of this Pokémon in hectograms. integer |
|
|
abilities A list of abilities this Pokémon could potentially have. list PokemonAbility |
|
|
forms A list of forms this Pokémon can take on. list NamedAPIResource (PokemonForm) |
|
|
game_indices A list of game indices relevent to Pokémon item by generation. list VersionGameIndex |
|
|
held_items A list of items this Pokémon may be holding when encountered. list PokemonHeldItem |
|
|
location_area_encounters A link to a list of location areas, as well as encounter details pertaining to specific versions. string |
|
|
moves A list of moves along with learn methods and level details pertaining to specific version groups. list PokemonMove |
|
|
past_types A list of details showing types this pokémon had in previous generations list PokemonTypePast |
|
|
sprites A set of sprites used to depict this Pokémon in the game. A visual representation of the various sprites can be found at PokeAPI/sprites PokemonSprites |
|
|
cries A set of cries used to depict this Pokémon in the game. A visual representation of the various cries can be found at PokeAPI/cries PokemonCries |
|
|
species The species this Pokémon belongs to. NamedAPIResource (PokemonSpecies) |
|
|
stats A list of base stat values for this Pokémon. list PokemonStat |
|
|
types A list of details showing types this Pokémon has. list PokemonType |
|
|
""" |
|
|
|
|
|
|
|
|
inputs = {'name': {'type': 'string', 'description': 'The name of the pokemon.'}} |
|
|
output_type = "object" |
|
|
|
|
|
def __init__(self, *args, **kwargs): |
|
|
super().__init__() |
|
|
try: |
|
|
import pokebase as pb |
|
|
except ImportError as e: |
|
|
raise ImportError( |
|
|
"You must install package `pokebase` to run this tool: for instance run `pip install pokebase`." |
|
|
) from e |
|
|
|
|
|
def forward(self, name: str) -> object: |
|
|
try: |
|
|
pokemon = pb.APIResource('pokemon', name) |
|
|
except requests.exceptions.Timeout: |
|
|
return "The request timed out. Please try again later." |
|
|
except RequestException as e: |
|
|
return "Error fetching contents from the pokemon remote API." |
|
|
except Exception as e: |
|
|
return f"An unexpected error occurred: {str(e)}" |
|
|
return pokemon |
|
|
|
