Update GeneralAgent.py

GeneralAgent.py

@@ -47,8 +47,11 @@ WELCOME_MSG = "Welcome to my general-purpose AI agent. Type `q` to quit. How sha
 @tool
 def wikipedia_search_tool(title: str) -> str:
     """Provides an excerpt from a Wikipedia article with the given title."""
-    page = wikipedia.page(title, auto_suggest=False)
-    return page.content[:3000]
+    try:
+        page = wikipedia.page(title, auto_suggest=False)
+        return page.content[:3000]
+    except Exception as e:
+        return f"Error during processing: {e}"
 
 @tool
 def media_tool(file_path: str) -> str:
@@ -58,26 +61,35 @@ def media_tool(file_path: str) -> str:
 @tool
 def internet_search_tool(search_query: str) -> str:
     """Does a google search with using the input as the search query. Returns a long batch of textual information related to the query."""
-    search_tool = DuckDuckGoSearchTool()
-    result = search_tool(search_query)
-    return result
+    try:
+        search_tool = DuckDuckGoSearchTool()
+        result = search_tool(search_query)
+        return result
+    except Exception as e:
+        return f"Error during processing: {e}"
 
 @tool
 def webscraper_tool(url: str) -> str:
     """Returns the page's html content from the input url."""
-    response = requests.get(url, stream=True)
-    if response.status_code == 200:
-        soup = BeautifulSoup(response.content, 'html.parser')
-        html_text = soup.get_text()
-        return html_text
-    else:
-        raise Exception(f"Failed to retrieve the webpage. Status code: {response.status_code}")
+    try:
+        response = requests.get(url, stream=True)
+        if response.status_code == 200:
+            soup = BeautifulSoup(response.content, 'html.parser')
+            html_text = soup.get_text()
+            return html_text
+        else:
+            return f"Failed to retrieve the webpage. Status code: {response.status_code}"
+    except Exception as e:
+        return f"Error during processing: {e}"
 
 @tool
 def read_excel_tool(file_path: str) -> str:
     """Returns the contents of an Excel file as a Pandas dataframe."""
-    df = pd.read_excel(file_path, engine = "openpyxl")
-    return df.to_string(index=False)
+    try:
+        df = pd.read_excel(file_path, engine = "openpyxl")
+        return df.to_string(index=False)
+    except Exception as e:
+        return f"Error during processing: {e}"
 
 class AgenticAI:
     def __init__(self):
@@ -151,35 +163,50 @@ class AgenticAI:
             tool_args = tool_call["args"]
 
             if tool_name == "wikipedia_search_tool":
-                print(f"called wikipedia with {str(tool_args)}")
-                page = wikipedia.page(tool_args.get("title"), auto_suggest=False)
-                response = page.content[:3000]
+                try:
+                    print(f"called wikipedia with {str(tool_args)}")
+                    page = wikipedia.page(tool_args.get("title"), auto_suggest=False)
+                    response = page.content[:3000]
+                except Exception as e:
+                    response = e
             elif tool_name == "media_tool":
-                print(f"called media with {str(tool_args)}")
-                response = "This tool hasn't been implemented yet. Please return 0 if the task cannot be solved without knowing the contents of this file."
+                try:
+                    print(f"called media with {str(tool_args)}")
+                    response = "This tool hasn't been implemented yet. Please return 0 if the task cannot be solved without knowing the contents of this file."
+                except Exception as e:
+                    response = e
             elif tool_name == "internet_search_tool":
-                print(f"called internet with {str(tool_args)}")
-                question = tool_args.get("search_query")
-                search_tool = DuckDuckGoSearchTool()
-                response = search_tool(question)[:3000]
+                try:
+                    print(f"called internet with {str(tool_args)}")
+                    question = tool_args.get("search_query")
+                    search_tool = DuckDuckGoSearchTool()
+                    response = search_tool(question)[:3000]
+                except Exception as e:
+                    response = e
             elif tool_name == "webscraper_tool":
-                print(f"called webscraper with {str(tool_args)}")
-                url = tool_args.get("url")
-                response = requests.get(url, stream=True)
-                if response.status_code == 200:
-                    soup = BeautifulSoup(response.content, 'html.parser')
-                    html_text = soup.get_text()
-                    response = html_text
-                else:
-                    response = Exception(f"Failed to retrieve the webpage. Status code: {response.status_code}")
+                try:
+                    print(f"called webscraper with {str(tool_args)}")
+                    url = tool_args.get("url")
+                    response = requests.get(url, stream=True)
+                    if response.status_code == 200:
+                        soup = BeautifulSoup(response.content, 'html.parser')
+                        html_text = soup.get_text()
+                        response = html_text
+                    else:
+                        response = f"Failed to retrieve the webpage. Status code: {response.status_code}"
+                except Exception as e:
+                    response = e
             elif tool_name == "read_excel_tool":
-                print(f"called excel with {str(tool_args)}")
-                path = tool_args.get("file_path")
-                df = pd.read_excel(path, engine = "openpyxl")
-                response = df
+                try:
+                    print(f"called excel with {str(tool_args)}")
+                    path = tool_args.get("file_path")
+                    df = pd.read_excel(path, engine = "openpyxl")
+                    response = df
+                except Exception as e:
+                    response = e
 
             else:
-                raise NotImplementedError(f'Unknown tool call: {tool_name}')
+                response = f'Unknown tool call: {tool_name}'
             
             outbound_msgs.append(
             ToolMessage(