Update tools/get_pokemon_info.py

tools/get_pokemon_info.py

@@ -9,17 +9,22 @@ class GetPokemonInfoTool(Tool):
     name = "Get Pokemon Info"
     description = "Retrieves complete information on a given Pokemon. Use this to get all info for a pokemon."
     inputs = {'name': {'type': 'string', 'description': 'The name of the pokemon.'}}
-    output_type = "string"
+    output_type = "object"
 
-    def forward(self, url: str) -> str:
+    def forward(self, url: str) -> object:
         try:
             import pokebase as pb
             pokemon = pb.APIResource('pokemon', name)
-
-            from smolagents.utils import truncate_content
         except ImportError as e:
             raise ImportError(
                 "You must install package `pokebase` to run this tool: for instance run `pip install pokebase`."
             ) from e
+        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)}"
+            
     def __init__(self, *args, **kwargs):
         self.is_initialized = False