feat(tool): 添加百度和必应搜索引擎支持

- 新增 BaiduSearch 和 BingSearch 工具类
- 更新配置文件，增加搜索引擎配置选项
- 添加必要的依赖包

Author: tjlzw

app/tool/baidu_search.py

@@ -0,0 +1,47 @@
+import asyncio
+from typing import List
+
+from baidusearch.baidusearch import search 
+from app.tool.base import BaseTool
+
+
+class BaiduSearch(BaseTool):
+    name: str = "baidu_search"
+    description: str = """Perform a Baidu search and return a list of relevant links.
+Use this tool when you need to find information on the web, get up-to-date data, or research specific topics.
+The tool returns a list of URLs that match the search query.
+"""
+    parameters: dict = {
+        "type": "object",
+        "properties": {
+            "query": {
+                "type": "string",
+                "description": "(required) The search query to submit to Baidu.",
+            },
+            "num_results": {
+                "type": "integer",
+                "description": "(optional) The number of search results to return. Default is 10.",
+                "default": 10,
+            },
+        },
+        "required": ["query"],
+    }
+
+    async def execute(self, query: str, num_results: int = 10) -> List[str]:
+        """
+        Execute a Baidu search and return a list of URLs.
+
+        Args:
+            query (str): The search query to submit to Baidu.
+            num_results (int, optional): The number of search results to return. Default is 10.
+
+        Returns:
+            List[str]: A list of URLs matching the search query.
+        """
+        # Run the search in a thread pool to prevent blocking
+        loop = asyncio.get_event_loop()
+        links = await loop.run_in_executor(
+            None, lambda: [result['url'] for result in search(query, num_results=num_results)]
+        )
+
+        return links

app/tool/bing_search.py

@@ -0,0 +1,66 @@
+import aiohttp
+from bs4 import BeautifulSoup
+from typing import List
+
+from app.tool.base import BaseTool
+
+
+class BingSearch(BaseTool):
+    name: str = "bing_search"
+    description: str = """使用必应搜索返回相关链接列表。
+    当需要查找网络信息、获取最新数据或研究特定主题时使用此工具。
+    该工具返回与搜索查询匹配的URL列表。
+    """
+    parameters: dict = {
+        "type": "object",
+        "properties": {
+            "query": {
+                "type": "string",
+                "description": "(必填) 提交给必应的搜索查询。",
+            },
+            "num_results": {
+                "type": "integer",
+                "description": "(可选) 要返回的搜索结果数量。默认为10。",
+                "default": 10,
+            },
+        },
+        "required": ["query"],
+    }
+
+    async def execute(self, query: str, num_results: int = 10) -> List[str]:
+        """
+        执行必应搜索并返回URL列表。
+
+        参数:
+            query (str): 要提交给必应的搜索查询。
+            num_results (int, optional): 要返回的搜索结果数量。默认为10。
+
+        返回:
+            List[str]: 与搜索查询匹配的URL列表。
+        """
+        headers = {
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
+        }
+        search_url = f"https://www.bing.com/search?q={query}"
+
+        async with aiohttp.ClientSession() as session:
+            try:
+                async with session.get(search_url, headers=headers) as response:
+                    response.raise_for_status()
+                    html = await response.text()
+            except Exception as e:
+                raise RuntimeError(f"必应搜索请求失败: {str(e)}")
+
+        soup = BeautifulSoup(html, 'html.parser')
+        links = []
+
+        # 必应搜索结果链接通常在类名为"b_algo"的div内，具体选择器可能需要根据实际页面结构调整
+        for result in soup.select('.b_algo'):
+            a_tag = result.select_one('a')
+            if a_tag and 'href' in a_tag.attrs:
+                link = a_tag['href']
+                links.append(link)
+                if len(links) >= num_results:
+                    break
+
+        return links[:num_results]

config/config.example.toml

@@ -25,3 +25,7 @@ api_key = "sk-..."
 [server]
 host = "localhost"
 port = 5172
+
+# Search engine configuration
+[search]
+default_engine = "google"  # Available options: google, baidu, bing

requirements.txt

@@ -15,6 +15,8 @@ uvicorn~=0.34.0
 unidiff~=0.7.5
 browser-use~=0.1.40
 googlesearch-python~=1.3.0
+python-baidu-api~=0.1.1
+py-bing-search~=0.2.7
 
 aiofiles~=24.1.0
 pydantic_core~=2.27.2