From b6c3aabce244cd2dde84a046ef5f7cfa9e9f0175 Mon Sep 17 00:00:00 2001 From: AlfredLong Date: Sat, 3 May 2025 20:28:07 +0000 Subject: [PATCH 1/3] Fix model_merger.sh by adding .py --- scripts/model_merge.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/model_merge.sh b/scripts/model_merge.sh index 6a5b8cb..0132c53 100644 --- a/scripts/model_merge.sh +++ b/scripts/model_merge.sh @@ -1,3 +1,3 @@ export CHECKPOINT_DIR= -python3 verl/scripts/model_merger --local_dir $CHECKPOINT_DIR \ No newline at end of file +python3 verl/scripts/model_merger.py --local_dir $CHECKPOINT_DIR \ No newline at end of file From ab38b34a96078adfde092f02123b80e80d862e88 Mon Sep 17 00:00:00 2001 From: AlfredLong Date: Sat, 3 May 2025 20:31:38 +0000 Subject: [PATCH 2/3] Fix execute meethod in search_tool by replacing pass --- agent_r1/tool/tools/search_tool.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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] From 6fb4b88821e4c3c6499f438b0d1ed9c5c3bc0fa0 Mon Sep 17 00:00:00 2001 From: AlfredLong Date: Sat, 3 May 2025 20:31:38 +0000 Subject: [PATCH 3/3] Fix execute meethod in search_tool by replacing pass --- scripts/model_merge.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/model_merge.sh b/scripts/model_merge.sh index 0132c53..6a5b8cb 100644 --- a/scripts/model_merge.sh +++ b/scripts/model_merge.sh @@ -1,3 +1,3 @@ export CHECKPOINT_DIR= -python3 verl/scripts/model_merger.py --local_dir $CHECKPOINT_DIR \ No newline at end of file +python3 verl/scripts/model_merger --local_dir $CHECKPOINT_DIR \ No newline at end of file