minor update

Author: fengju0213

camel/toolkits/search_toolkit.py

@@ -15,7 +15,7 @@
 import warnings
 from typing import Any, Dict, List, Literal, Optional, TypeAlias, Union, cast
 
-
+import requests
 
 from camel.logger import get_logger
 from camel.toolkits.base import BaseToolkit
@@ -54,7 +54,7 @@ def __init__(
         """
         super().__init__(timeout=timeout)
         self.exclude_domains = exclude_domains
-    
+
     @api_keys_required(
         [
             (None, "SERPER_API_KEY"),
@@ -80,8 +80,6 @@ def search_serper(
         """
         import json
 
-        import requests
-
         SERPER_API_KEY = os.getenv("SERPER_API_KEY")
 
         url = "https://google.serper.dev/search"
@@ -427,8 +425,6 @@ def search_brave(
             Dict[str, Any]: A dictionary representing a search result.
         """
 
-        import requests
-
         BRAVE_API_KEY = os.getenv("BRAVE_API_KEY")
 
         url = "https://api.search.brave.com/res/v1/web/search"
@@ -570,8 +566,6 @@ def search_google(
         """
         from urllib.parse import quote
 
-        import requests
-
         # Validate input parameters
         if not isinstance(start_page, int) or start_page < 1:
             raise ValueError("start_page must be a positive integer")
@@ -821,8 +815,6 @@ def search_bocha(
         """
         import json
 
-        import requests
-
         BOCHA_API_KEY = os.getenv("BOCHA_API_KEY")
 
         url = "https://api.bochaai.com/v1/web-search"
@@ -874,8 +866,6 @@ def search_baidu(
         """
         from bs4 import BeautifulSoup
 
-        import requests
-
         try:
             url = "https://www.baidu.com/s"
             headers = {
@@ -960,8 +950,6 @@ def search_bing(
         from typing import Any, Dict, List, cast
         from urllib.parse import urlencode
 
-        import requests
-
         from bs4 import BeautifulSoup, Tag
 
         try:
@@ -1228,7 +1216,6 @@ def search_alibaba_tongxiao(
                 message. Each result contains title, snippet, url and other
                 metadata.
         """
-        import requests
 
         TONGXIAO_API_KEY = os.getenv("TONGXIAO_API_KEY")
 
@@ -1412,7 +1399,7 @@ def get_tools(self) -> List[FunctionTool]:
                 representing the functions in the toolkit.
         """
         return [
-            FunctionTool(self.search_serper), 
+            FunctionTool(self.search_serper),
             FunctionTool(self.search_wiki),
             FunctionTool(self.search_linkup),
             FunctionTool(self.search_google),

examples/toolkits/search_toolkit.py

@@ -421,7 +421,6 @@ class PersonInfo(BaseModel):
 """  # noqa: E501
 
 
-
 # Example using Serper search
 if os.getenv("SERPER_API_KEY"):
     serper_response = SearchToolkit().search_serper(
@@ -480,4 +479,4 @@ class PersonInfo(BaseModel):
     Apple Account or create a new account to start using Apple services
     ...', 'position': 10}], 'credits': 1}
     ===============================================================================
-    """
+    """

test/toolkits/test_search_functions.py

@@ -900,6 +900,7 @@ def test_search_metaso_invalid_json(mock_https_connection, search_toolkit):
     mock_https_connection.assert_called_once_with("metaso.cn")
     mock_conn.request.assert_called_once()
 
+
 @patch('requests.post')
 def test_search_serper_success(mock_post, search_toolkit):
     """Test successful Serper search."""
@@ -913,24 +914,24 @@ def test_search_serper_success(mock_post, search_toolkit):
             "gl": "us",
             "hl": "en",
             "num": 10,
-            "type": "search"
+            "type": "search",
         },
         "organic": [
             {
                 "title": "Apple",
                 "link": "https://www.apple.com/",
                 "snippet": "Discover the innovative world of Apple...",
-                "position": 1
+                "position": 1,
             }
-        ]
+        ],
     }
     mock_post.return_value = mock_response
 
     with patch.dict(os.environ, {'SERPER_API_KEY': 'test_key'}):
         result = search_toolkit.search_serper(query="apple inc")
 
     assert result == mock_response.json.return_value
-    
+
     # Verify request
     mock_post.assert_called_once()
     args, kwargs = mock_post.call_args
@@ -944,7 +945,7 @@ def test_search_serper_success(mock_post, search_toolkit):
     assert json.loads(kwargs['data']) == {
         "q": "apple inc",
         "location": "United States",
-        "page": 1
+        "page": 1,
     }
 
 
@@ -956,5 +957,5 @@ def test_search_serper_error(mock_post, search_toolkit):
     with patch.dict(os.environ, {'SERPER_API_KEY': 'test_key'}):
         with pytest.raises(RuntimeError) as excinfo:
             search_toolkit.search_serper(query="test")
-    
+
     assert "Error making request to Serper" in str(excinfo.value)