andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

updating serperdev to allow for query and search_query

Browse Source
This commit is contained in:
João Moura
2024-05-02 02:37:47 -03:00
parent b992238a5d
commit 5fb9ddfa2a
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/crewai_tools/tools/serper_dev_tool/serper_dev_tool.py
View File

@@ -19,10 +19,13 @@ class SerperDevTool(BaseTool):
def _run( def _run(
self, self,
search_query: str,
**kwargs: Any, **kwargs: Any,
) -> 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 = { headers = {
'X-API-KEY': os.environ['SERPER_API_KEY'], 'X-API-KEY': os.environ['SERPER_API_KEY'],
'content-type': 'application/json' 'content-type': 'application/json'