From f447f71a8ee017298acaac590e30b9ca5ae2589e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Sun, 14 Jul 2024 17:59:17 -0300 Subject: [PATCH] Update serper_dev_tool.py --- .../tools/serper_dev_tool/serper_dev_tool.py | 23 ++++++++----------- 1 file changed, 10 insertions(+), 13 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 77d008be9..651d9c5a6 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 @@ -27,7 +27,7 @@ class SerperDevTool(BaseTool): location: Optional[str] = None locale: Optional[str] = None n_results: int = Field(default=10, description="Number of search results to return") - save_file: bool = Field(default=False, description="Flag to determine whether to save the results to a file") + save_file: bool = Field(default=False, description="Flag to determine whether to save the results to a file") def _run( self, @@ -35,22 +35,19 @@ class SerperDevTool(BaseTool): ) -> Any: search_query = kwargs.get('search_query') or kwargs.get('query') - save_file = kwargs.get('save_file', self.save_file) - n_results = kwargs.get('n_results', self.n_results) + save_file = kwargs.get('save_file', self.save_file) + n_results = kwargs.get('n_results', self.n_results) - payload = json.dumps( - { - "q": search_query, - "num": n_results, - "gl": self.country, - "location": self.location, - "hl": self.locale, - } - ) + payload = { "q": search_query, "num": n_results } + payload["gl"] = self.country if self.country + payload["location"] = self.country if self.location + payload["hl"] = self.country if self.locale + + payload = json.dumps(payload) headers = { 'X-API-KEY': os.environ['SERPER_API_KEY'], - 'content-type': 'application/json' + 'content-type': 'application/json' } response = requests.request("POST", self.search_url, headers=headers, data=payload) results = response.json()