From 3b77de7b21365672c8704191372d7bf87e134be5 Mon Sep 17 00:00:00 2001 From: Jeroen Vet Date: Sat, 23 Mar 2024 18:36:08 +0100 Subject: [PATCH] Update serper_dev_tool.py consider n_results parameter In original code n_results is always None so you always get only 10 results from Serper. With this change, when you explicitly set the n_results parameter when creating a SerperDevTool object it is taken into account. --- src/crewai_tools/tools/serper_dev_tool/serper_dev_tool.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/crewai_tools/tools/serper_dev_tool/serper_dev_tool.py b/src/crewai_tools/tools/serper_dev_tool/serper_dev_tool.py index 3fbf5ea61..a3fe871f1 100644 --- a/src/crewai_tools/tools/serper_dev_tool/serper_dev_tool.py +++ b/src/crewai_tools/tools/serper_dev_tool/serper_dev_tool.py @@ -15,14 +15,14 @@ class SerperDevTool(BaseTool): description: str = "A tool that can be used to semantic search a query from a txt's content." args_schema: Type[BaseModel] = SerperDevToolSchema search_url: str = "https://google.serper.dev/search" - n_results: int = None + n_results: int = 10 def _run( self, search_query: str, **kwargs: Any, ) -> Any: - payload = json.dumps({"q": search_query}) + payload = json.dumps({"q": search_query, "num": self.n_results}) headers = { 'X-API-KEY': os.environ['SERPER_API_KEY'], 'content-type': 'application/json'