refactor(tools): add multi-provider support to web_search tool (aden-hive#795)

* feat(tools): add Google Custom Search as alternative to Brave Search

Adds google_search tool using Google Custom Search API as an alternative
to the existing web_search tool (Brave Search).

Changes:
- Add google_search_tool with full implementation
- Register Google credentials (GOOGLE_API_KEY, GOOGLE_CSE_ID)
- Register tool in tools/__init__.py
- Add README with setup instructions

Closes aden-hive#793

* test(tools): add unit tests for google_search tool

Adds 7 tests mirroring web_search_tool test patterns:
- Missing API key error handling
- Missing CSE ID error handling
- Empty query validation
- Long query validation
- num_results clamping
- Default parameters
- Custom language/country parameters

All tests pass.

* refactor(tools): add multi-provider support to web_search tool

BREAKING CHANGE: None - backward compatible. Brave remains default.

- Add Google Custom Search as alternative provider in web_search
- Add 'provider' parameter: 'auto' (default), 'google', 'brave'
- Auto mode tries Brave first for backward compatibility
- Remove separate google_search_tool (consolidated into web_search)
- Update tests to cover multi-provider functionality (13 tests)
- Update README documentation

Users with BRAVE_SEARCH_API_KEY: No changes needed
Users with GOOGLE_API_KEY + GOOGLE_CSE_ID: Can use provider='google'
Users with both: Brave preferred by default, use provider='google' to force

Closes aden-hive#793

* feat(tools): fixed readme

---------

Co-authored-by: Mustafa Abdat <[email protected]>

Author: vrijmetse

tools/README.md

@@ -25,7 +25,11 @@ cp .env.example .env
 | Variable               | Required For                  | Get Key                                                 |
 | ---------------------- | ----------------------------- | ------------------------------------------------------- |
 | `ANTHROPIC_API_KEY`    | MCP server startup, LLM nodes | [console.anthropic.com](https://console.anthropic.com/) |
-| `BRAVE_SEARCH_API_KEY` | `web_search` tool             | [brave.com/search/api](https://brave.com/search/api/)   |
+| `BRAVE_SEARCH_API_KEY` | `web_search` tool (Brave)     | [brave.com/search/api](https://brave.com/search/api/)   |
+| `GOOGLE_API_KEY`       | `web_search` tool (Google)    | [console.cloud.google.com](https://console.cloud.google.com/) |
+| `GOOGLE_CSE_ID`        | `web_search` tool (Google)    | [programmablesearchengine.google.com](https://programmablesearchengine.google.com/) |
+
+> **Note:** `web_search` supports multiple providers. Set either Brave OR Google credentials. Brave is preferred for backward compatibility.
 
 Alternatively, export as environment variables:
 
@@ -68,7 +72,7 @@ python mcp_server.py
 | `apply_patch`          | Apply unified patches to files                 |
 | `grep_search`          | Search file contents with regex                |
 | `execute_command_tool` | Execute shell commands                         |
-| `web_search`           | Search the web using Brave Search API          |
+| `web_search`           | Search the web (Google or Brave, auto-detected) |
 | `web_scrape`           | Scrape and extract content from webpages       |
 | `pdf_read`             | Read and extract text from PDF files           |
 

tools/src/aden_tools/credentials/search.py

@@ -3,6 +3,7 @@
 
 Contains credentials for search providers like Brave Search, Google, Bing, etc.
 """
+
 from .base import CredentialSpec
 
 SEARCH_CREDENTIALS = {
@@ -15,14 +16,22 @@
         help_url="https://brave.com/search/api/",
         description="API key for Brave Search",
     ),
-    # Future search providers:
-    # "google_search": CredentialSpec(
-    #     env_var="GOOGLE_SEARCH_API_KEY",
-    #     tools=["google_search"],
-    #     node_types=[],
-    #     required=True,
-    #     startup_required=False,
-    #     help_url="https://developers.google.com/custom-search/v1/overview",
-    #     description="API key for Google Custom Search",
-    # ),
+    "google_search": CredentialSpec(
+        env_var="GOOGLE_API_KEY",
+        tools=["google_search"],
+        node_types=[],
+        required=True,
+        startup_required=False,
+        help_url="https://console.cloud.google.com/",
+        description="API key for Google Custom Search",
+    ),
+    "google_cse": CredentialSpec(
+        env_var="GOOGLE_CSE_ID",
+        tools=["google_search"],
+        node_types=[],
+        required=True,
+        startup_required=False,
+        help_url="https://programmablesearchengine.google.com/",
+        description="Google Custom Search Engine ID",
+    ),
 }

tools/src/aden_tools/tools/__init__.py

@@ -10,6 +10,7 @@
     credentials = CredentialManager()
     register_all_tools(mcp, credentials=credentials)
 """
+
 from typing import List, Optional, TYPE_CHECKING
 
 from fastmcp import FastMCP
@@ -27,11 +28,15 @@
 from .file_system_toolkits.view_file import register_tools as register_view_file
 from .file_system_toolkits.write_to_file import register_tools as register_write_to_file
 from .file_system_toolkits.list_dir import register_tools as register_list_dir
-from .file_system_toolkits.replace_file_content import register_tools as register_replace_file_content
+from .file_system_toolkits.replace_file_content import (
+    register_tools as register_replace_file_content,
+)
 from .file_system_toolkits.apply_diff import register_tools as register_apply_diff
 from .file_system_toolkits.apply_patch import register_tools as register_apply_patch
 from .file_system_toolkits.grep_search import register_tools as register_grep_search
-from .file_system_toolkits.execute_command_tool import register_tools as register_execute_command
+from .file_system_toolkits.execute_command_tool import (
+    register_tools as register_execute_command,
+)
 from .csv_tool import register_tools as register_csv
 
 
@@ -56,9 +61,7 @@ def register_all_tools(
     register_pdf_read(mcp)
 
     # Tools that need credentials (pass credentials if provided)
-    # web_search handles both credential sources internally:
-    # - If credentials provided: uses credentials.get("brave_search")
-    # - If credentials is None: falls back to os.getenv("BRAVE_SEARCH_API_KEY")
+    # web_search supports multiple providers (Google, Brave) with auto-detection
     register_web_search(mcp, credentials=credentials)
 
     # Register file system toolkits

tools/src/aden_tools/tools/web_search_tool/README.md

@@ -1,31 +1,64 @@
 # Web Search Tool
 
-Search the web using the Brave Search API.
+Search the web using multiple providers with automatic detection.
 
 ## Description
 
 Returns titles, URLs, and snippets for search results. Use when you need current information, research topics, or find websites.
 
+Supports multiple search providers:
+- **Brave Search API** (default, for backward compatibility)
+- **Google Custom Search API** (fallback)
+
 ## Arguments
 
 | Argument | Type | Required | Default | Description |
 |----------|------|----------|---------|-------------|
 | `query` | str | Yes | - | The search query (1-500 chars) |
-| `num_results` | int | No | `10` | Number of results to return (1-20) |
-| `country` | str | No | `us` | Country code for localized results (us, uk, de, etc.) |
+| `num_results` | int | No | `10` | Number of results (1-10 for Google, 1-20 for Brave) |
+| `country` | str | No | `us` | Country code for localized results |
+| `language` | str | No | `en` | Language code (Google only) |
+| `provider` | str | No | `auto` | Provider: "auto", "google", or "brave" |
 
 ## Environment Variables
 
+Set credentials for at least one provider:
+
+### Option 1: Google Custom Search
+| Variable | Required | Description |
+|----------|----------|-------------|
+| `GOOGLE_API_KEY` | Yes | API key from [Google Cloud Console](https://console.cloud.google.com/) |
+| `GOOGLE_CSE_ID` | Yes | Search Engine ID from [Programmable Search Engine](https://programmablesearchengine.google.com/) |
+
+### Option 2: Brave Search
 | Variable | Required | Description |
 |----------|----------|-------------|
 | `BRAVE_SEARCH_API_KEY` | Yes | API key from [Brave Search API](https://brave.com/search/api/) |
 
+## Provider Selection
+
+- `provider="auto"` (default): Uses Brave if available, otherwise Google (backward compatible)
+- `provider="brave"`: Force Brave Search
+- `provider="google"`: Force Google Custom Search
+
+## Example Usage
+
+```python
+# Auto-detect provider based on available credentials
+result = web_search(query="climate change effects")
+
+# Force specific provider
+result = web_search(query="python tutorial", provider="google")
+result = web_search(query="local news", provider="brave", country="id")
+```
+
 ## Error Handling
 
 Returns error dicts for common issues:
-- `BRAVE_SEARCH_API_KEY environment variable not set` - Missing API key
+- `No search credentials configured` - No API keys set
+- `Google credentials not configured` - Missing Google keys when provider="google"
+- `Brave credentials not configured` - Missing Brave key when provider="brave"
 - `Query must be 1-500 characters` - Empty or too long query
-- `Invalid API key` - API key rejected (HTTP 401)
-- `Rate limit exceeded. Try again later.` - Too many requests (HTTP 429)
+- `Invalid API key` - API key rejected
+- `Rate limit exceeded` - Too many requests
 - `Search request timed out` - Request exceeded 30s timeout
-- `Network error: <error>` - Connection or DNS issues