Update tools/WebSearchTool.py

tools/WebSearchTool.py

@@ -5,13 +5,13 @@ from googleapiclient.discovery import build
 class GoogleSearchTool(Tool):
     """
     A tool to perform web search Google searches using the Custom Search Engine (CSE) API.
-    Use this tool first to get the necessary information for calculation or further tools. 
+    Use this tool first to get the necessary information for calculation or further tools.
     """
     # Use a descriptive name to guide the agent
     name = "google_search"
     # Update the description to reflect the new functionality
     description = "Use Google to find current information and general knowledge. Returns a snippet and URL."
-    
+
     # Define the required input structure for the agent framework
     inputs = {
         "query": {"type": "string", "description": "The search term to look up."}
@@ -20,15 +20,15 @@ class GoogleSearchTool(Tool):
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
-        
+
         # Retrieve credentials from environment variables
         self.api_key = os.getenv("GOOGLE_API_KEY")
         self.cse_id = os.getenv("GOOGLE_CSE_ID")
-        
+
         # Check for mandatory credentials
         if not self.api_key or not self.cse_id:
             raise ValueError("GOOGLE_API_KEY or GOOGLE_CSE_ID secret not found. Check environment variables.")
-        
+
         # Initialize the Google Custom Search service
         # 'customsearch' is the API name, 'v1' is the version
         self.service = build(
@@ -38,10 +38,13 @@ class GoogleSearchTool(Tool):
     def forward(self, query: str) -> str:
         """
         Executes a Google search query and formats the top results (up to 3).
-        
+
+        The output is formatted to prioritize the Title and Snippet (informational content)
+        over the URL to address the confusion that it was only returning links.
+
         Args:
             query: The search term provided by the agent.
-            
+
         Returns:
             A formatted string of search results, or an error message.
         """
@@ -53,24 +56,25 @@ class GoogleSearchTool(Tool):
                 cx=self.cse_id,
                 num=3
             ).execute()
-            
+
             items = res.get('items', [])
-            
+
             if not items:
                 # Return the specific failure message expected by the agent
                 return "XX record info: No results found."
 
             search_results = []
-            for item in items:
-                # Format the output clearly for the LLM
+            for i, item in enumerate(items):
+                # We've adjusted the format here to put the snippet first,
+                # which is usually the most important information for the agent.
                 search_results.append(
-                    f"Title: {item.get('title')}\n"
-                    f"Snippet: {item.get('snippet')}\n"
-                    f"URL: {item.get('link')}"
+                    f"RESULT {i+1}: '{item.get('title')}'\n"
+                    f"CONTENT: {item.get('snippet')}\n"
+                    f"SOURCE: {item.get('link')}" # The URL is now clearly labeled as the SOURCE
                 )
-            
+
             # Join the results with a clear separator
-            return "\n---\n".join(search_results)
+            return "\n\n---SEPARATOR---\n\n".join(search_results)
 
         except Exception as e:
             # Provide an informative error message upon API failure