diff --git a/agent/tools/github_find_examples.py b/agent/tools/github_find_examples.py index f5f2ddaa..d2661f8f 100644 --- a/agent/tools/github_find_examples.py +++ b/agent/tools/github_find_examples.py @@ -412,7 +412,7 @@ def find_examples( "Sequence: github_find_examples → github_read_file (study the example) → implement based on what you found.\n\n" "Skip this only for: simple data queries, status checks, non-code tasks.\n\n" "Examples:\n" - " {keyword: 'sft', repo: 'trl'} → finds examples/scripts/sft.py\n" + " {keyword: 'sft', repo: 'trl'} → finds examples/sft_gemma3/sft_gemma3.py\n" " {keyword: 'grpo', repo: 'trl'} → finds GRPO training examples\n" " {repo: 'trl', max_results: 20} → lists all available training method examples" ), diff --git a/agent/tools/research_tool.py b/agent/tools/research_tool.py index 3cc19b2b..afc0e90b 100644 --- a/agent/tools/research_tool.py +++ b/agent/tools/research_tool.py @@ -185,7 +185,7 @@ async def _record_research_llm_call( # 6. Now get working code for the training method github_find_examples({"repo": "trl", "keyword": "sft"}) -github_read_file({"repo": "huggingface/trl", "path": "examples/scripts/sft.py"}) +github_read_file({"repo": "huggingface/trl", "path": "trl/scripts/sft.py"}) explore_hf_docs("trl") ```