Update agent.py

agent.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,  WikipediaSearchTool, LiteLLMModel, tool, load_tool # HfApiModel, OpenAIServerModel
+from smolagents import CodeAgent, DuckDuckSearchTool, WikipediaSearchTool, LiteLLMModel, tool, load_tool # HfApiModel, OpenAIServerModel
 import asyncio
 import os
 import yaml
@@ -7,32 +7,31 @@ import requests
 from io import BytesIO
 
 # 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
-    """
-    api_key = os.environ.get("GOOGLE_API_KEY")
-    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(image_path: str) -> str:
@@ -103,7 +102,8 @@ class MagAgent:
         self.agent = CodeAgent(
             model= model,
             tools=[
-                GoogleSearchTool,
+#                GoogleSearchTool,
+                DuckDuckSearchTool(),
                 WikipediaSearchTool(),
                 ImageAnalysisTool,
 #                LocalFileAudioTool()