feat: adding new tools

app.py

@@ -1,4 +1,6 @@
+import json
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from bs4 import BeautifulSoup
 from duckduckgo_search import DDGS
 import datetime
 import requests
@@ -36,7 +38,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 @tool
 def search_for_something(query: str) -> str:
-    """A tool that fetches the 5 first websites that appear in a query, with their title, url and snippets.
+    """A tool that fetches the 5 first websites that appear in a query, with their title, url and snippets. Can be used to get websites urls to be used in the request_get tool.
     Args:
         query: String that represent what is being searched
     """
@@ -53,6 +55,30 @@ def search_for_something(query: str) -> str:
         return "\n".join(formatted_results)
     except Exception as e:
         return f"Error searching for {query} in the Web: {str(e)}"
+    
+@tool
+def request_get(url: str) -> str:
+    """Tool that is used to make get requests in any API or Website needed and get its response. Can be used after the search_for_something tool to get more specific things in the websites returned.
+    Args:
+        url: Url that is used to reach the API or Website.
+    """
+    
+    try:
+        r = requests.get(url)
+        if r.status_code == 200:
+            try:
+                data = r.json()
+                return json.dumps(data, indent=2)
+            except ValueError:
+                soup = BeautifulSoup(r.text, 'html.parser')
+                title = soup.title.string if soup.title else "No title available"
+                body_text = ' '.join([p.text for p in soup.find_all('p')[:5]])
+                return f"Title: {title}\n\nContent:\n{body_text}"
+        else:
+            return f"Error: Received status code {r.status_code} from {url}"
+        
+    except Exception as e:
+        return f"Error in request: {str(e)}"
 
 
 final_answer = FinalAnswerTool()
@@ -76,7 +102,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, search_for_something], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, search_for_something, image_generation_tool, request_get], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,