Merge pull request #51 from smsajjadzaidi/fix/serper_dev_tool

fixed use of arg n_results
This commit is contained in:
João Moura
2024-07-14 13:39:20 -07:00
committed by GitHub

View File

@@ -46,7 +46,7 @@ class SerperDevTool(BaseTool):
response = requests.request("POST", self.search_url, headers=headers, data=payload) response = requests.request("POST", self.search_url, headers=headers, data=payload)
results = response.json() results = response.json()
if 'organic' in results: if 'organic' in results:
results = results['organic'] results = results['organic'][:self.n_results]
string = [] string = []
for result in results: for result in results:
try: try:
@@ -57,7 +57,7 @@ class SerperDevTool(BaseTool):
"---" "---"
])) ]))
except KeyError: except KeyError:
next continue
content = '\n'.join(string) content = '\n'.join(string)
if save_file: if save_file: