From ec2ce3ef9d642a9771fb103de7b74c7d20bc4943 Mon Sep 17 00:00:00 2001 From: skamenan7 Date: Sat, 5 Jul 2025 09:36:05 -0400 Subject: [PATCH] feat(starter): add Brave search to tool groups - Add builtin::websearch_brave to default_tool_groups in starter.py - Regenerate run.yaml with distro codegen - Fixes #2606: Brave search was configured but not registered Following template architecture pattern: - Modified template source (starter.py) - Used distro codegen to regenerate config files - Never edited generated files directly --- llama_stack/templates/starter/run.yaml | 2 ++ llama_stack/templates/starter/starter.py | 4 ++++ 2 files changed, 6 insertions(+) diff --git a/llama_stack/templates/starter/run.yaml b/llama_stack/templates/starter/run.yaml index 888a2c3bf..b061ae760 100644 --- a/llama_stack/templates/starter/run.yaml +++ b/llama_stack/templates/starter/run.yaml @@ -1196,6 +1196,8 @@ benchmarks: [] tool_groups: - toolgroup_id: builtin::websearch provider_id: tavily-search +- toolgroup_id: builtin::websearch_brave + provider_id: brave-search - toolgroup_id: builtin::rag provider_id: rag-runtime server: diff --git a/llama_stack/templates/starter/starter.py b/llama_stack/templates/starter/starter.py index 6b8aa8974..e27a00dc5 100644 --- a/llama_stack/templates/starter/starter.py +++ b/llama_stack/templates/starter/starter.py @@ -347,6 +347,10 @@ def get_distribution_template() -> DistributionTemplate: toolgroup_id="builtin::websearch", provider_id="tavily-search", ), + ToolGroupInput( + toolgroup_id="builtin::websearch_brave", + provider_id="brave-search", + ), ToolGroupInput( toolgroup_id="builtin::rag", provider_id="rag-runtime",