Update agent.py

agent.py

@@ -16,32 +16,32 @@ import time
 import shutil
 import traceback
 
-# Simulated additional tools (implementation depends on external APIs or setup)
-#@tool
-#def GoogleSearchTool(query: str) -> str:
-#    """Tool for performing Google searches using Custom Search JSON API
-#    Args:
-#        query (str): Search query string
-#    Returns:
-#        str: Formatted search results
-#    """
-#    cse_id = os.environ.get("GOOGLE_CSE_ID")
-#    if not api_key or not cse_id:
-#        raise ValueError("GOOGLE_API_KEY and GOOGLE_CSE_ID must be set in environment variables.")
-#    url = "https://www.googleapis.com/customsearch/v1"
-#    params = {
-#        "key": api_key,
-#        "cx": cse_id,
-#        "q": query,
-#        "num": 5  # Number of results to return
-#    }
-#    try:
-#        response = requests.get(url, params=params)
-#        response.raise_for_status()
-#        results = response.json().get("items", [])
-#        return "\n".join([f"{item['title']}: {item['link']}" for item in results]) or "No results found."
-#    except Exception as e:
-#        return f"Error performing Google search: {str(e)}"
+
+@tool
+def GoogleSearchTool(query: str) -> str:
+    """Tool for performing Google searches using Custom Search JSON API
+    Args:
+        query (str): Search query string
+    Returns:
+        str: Formatted search results
+    """
+    cse_id = os.environ.get("GOOGLE_CSE_ID")
+    if not api_key or not cse_id:
+        raise ValueError("GOOGLE_API_KEY and GOOGLE_CSE_ID must be set in environment variables.")
+    url = "https://www.googleapis.com/customsearch/v1"
+    params = {
+        "key": api_key,
+        "cx": cse_id,
+        "q": query,
+        "num": 5  # Number of results to return
+    }
+    try:
+        response = requests.get(url, params=params)
+        response.raise_for_status()
+        results = response.json().get("items", [])
+        return "\n".join([f"{item['title']}: {item['link']}" for item in results]) or "No results found."
+    except Exception as e:
+        return f"Error performing Google search: {str(e)}"
 
 #@tool
 #def ImageAnalysisTool(question: str, model: LiteLLMModel) -> str:
@@ -64,6 +64,7 @@ import traceback
 #    }
 #    try:
 #        response = requests.get(image_url, headers=headers)
+
 #        response.raise_for_status()
 #        image = Image.open(BytesIO(response.content)).convert("RGB")
 #    except Exception as e:
@@ -83,6 +84,29 @@ import traceback
 #    
 #    return f"The image description: '{response}'"
 
+class VisitWebpageTool(Tool):
+    name = "visit_webpage"
+    description = "Visits a webpage at the given url and reads its content as a markdown string. Use this to browse webpages."
+    inputs = {'url': {'type': 'string', 'description': 'The url of the webpage to visit.'}}
+    output_type = "string"
+
+    def forward(self, url: str) -> str:
+        try:
+            response = requests.get(url, timeout=20)
+            response.raise_for_status()
+            markdown_content = markdownify(response.text).strip()
+            markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
+            from smolagents.utils import truncate_content
+            return truncate_content(markdown_content, 10000)
+        except requests.exceptions.Timeout:
+            return "The request timed out. Please try again later or check the URL."
+        except requests.exceptions.RequestException as e:
+            return f"Error fetching the webpage: {str(e)}"
+        except Exception as e:
+            return f"An unexpected error occurred: {str(e)}"
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False
 
 class DownloadTaskAttachmentTool(Tool):
     name = "download_file"
@@ -254,12 +278,13 @@ class MagAgent:
             model= model,
             tools=[
                 DownloadTaskAttachmentTool(rate_limiter=rate_limiter),
-                DuckDuckGoSearchTool(),
+#                DuckDuckGoSearchTool(),
                 WikipediaSearchTool(),
                 SpeechToTextTool,
-                ExcelReaderTool(),
-                PythonCodeReaderTool()
-#                GoogleSearchTool,
+                ExcelReaderTool,
+                VisitWebpageTool,
+                PythonCodeReaderTool,
+                GoogleSearchTool,
 #                ImageAnalysisTool,
             ],
             verbosity_level=3,