diff --git a/llama_stack/templates/ollama/ollama.py b/llama_stack/templates/ollama/ollama.py index cba25296b..4d1bba18d 100644 --- a/llama_stack/templates/ollama/ollama.py +++ b/llama_stack/templates/ollama/ollama.py @@ -83,6 +83,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", diff --git a/llama_stack/templates/ollama/run-with-safety.yaml b/llama_stack/templates/ollama/run-with-safety.yaml index 98db5fc98..3edceff3c 100644 --- a/llama_stack/templates/ollama/run-with-safety.yaml +++ b/llama_stack/templates/ollama/run-with-safety.yaml @@ -150,6 +150,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 - toolgroup_id: builtin::wolfram_alpha