From 5fb9ddfa2a604b0a55c3281b10d30eef30608d40 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Moura?= <joaomdmoura@gmail.com>
Date: Thu, 2 May 2024 02:37:47 -0300
Subject: [PATCH] updating serperdev to allow for query and search_query

---
 src/crewai_tools/tools/serper_dev_tool/serper_dev_tool.py | 7 +++++--
 1 file changed, 5 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 e08c1e983..927c0e3b3 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
@@ -19,10 +19,13 @@ class SerperDevTool(BaseTool):
 
 	def _run(
 		self,
-		search_query: str,
 		**kwargs: Any,
 	) -> Any:
-		payload = json.dumps({"q": search_query, "num": self.n_results})
+		search_query = kwargs.get('search_query')
+		if search_query is None:
+			search_query = kwargs.get('query')
+
+		payload = json.dumps({"q": search_query})
 		headers = {
 				'X-API-KEY': os.environ['SERPER_API_KEY'],
 				'content-type': 'application/json'