diff --git a/agent_r1/tool/tools/search_tool.py b/agent_r1/tool/tools/search_tool.py index fa1ad77..2816d27 100644 --- a/agent_r1/tool/tools/search_tool.py +++ b/agent_r1/tool/tools/search_tool.py @@ -77,7 +77,11 @@ def execute(self, args: Dict) -> str: Returns: Formatted search results """ - pass + query = args["query"] + embedding = self.model.encode_queries([query]) + dist, ids = self.index.search(embedding, 5) # ids: 1*5 + results_str = self._format_results(ids[0]) + return results_str def batch_execute(self, args_list: List[Dict]) -> List[str]: queries = [x["query"] for x in args_list]