add test with api

app.py

@@ -18,6 +18,32 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+
+import requests
+
+@tool
+def get_book():
+    """A tool that fetches the Harry Potter books with information about them.
+    Args:
+    """
+    response = requests.get('https://potterapi-fedeperin.vercel.app/en/books')
+    books = response.json()
+    return books
+
+@tool
+def filter_book(libro, campo, valor, operador='=='):
+    """
+    Filtra libros según un campo, valor y operador especificados
+    operadores permitidos: ==, >, <, >=, <=, contains
+    """
+    if operador == 'contains':
+        return [libro for libro in libros if valor.lower() in str(getattr(libro, campo)).lower()]
+    elif operador in ['==', '>', '<', '>=', '<=']:
+        return [libro for libro in libros if eval(f"{getattr(libro, campo)} {operador} {valor}")]
+    else:
+        raise ValueError("Operador no válido")
+
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -55,7 +81,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, get_book, filter_book], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,