/app/tool: A new tool for Baidu search

app/tool/baidu_search.py

@@ -0,0 +1,57 @@
+import asyncio
+from typing import List
+from baidusearch.baidusearch import search
+# from app.tool.base import BaseTool
+from 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.
+        """
+        print('baidu search query:', query)
+        # Run the search in a thread pool to prevent blocking
+        loop = asyncio.get_event_loop()
+        response = await loop.run_in_executor(
+            None, lambda: search(query, num_results=num_results)
+        )
+        links = []
+        for item in response:
+            url = item['url']
+            if url.startswith('http'):
+                links.append(url)
+        # print('baidu search query links:', links)
+        return links
+
+if __name__ == '__main__':
+    links = asyncio.run(BaiduSearch().execute("Manux", 5))
+    print('baidu search query links:', links)
+