diff --git a/lagent/actions/google_search.py b/lagent/actions/google_search.py index 27b3145..f039860 100644 --- a/lagent/actions/google_search.py +++ b/lagent/actions/google_search.py @@ -196,7 +196,7 @@ async def run(self, query: str, k: int = 10) -> ActionReturn: tool_return.errmsg = response tool_return.state = ActionStatusCode.HTTP_ERROR elif status_code == 200: - parsed_res = self._parse_results(response) + parsed_res = self._parse_results(response, k) tool_return.result = [dict(type='text', content=str(parsed_res))] tool_return.state = ActionStatusCode.SUCCESS else: