From ddcf74f59c4177219d82605b3d468a6a524f2780 Mon Sep 17 00:00:00 2001 From: skamenan7 Date: Sat, 5 Jul 2025 09:16:13 -0400 Subject: [PATCH] fix: add Brave search tool group in ollama.py template - Add ToolGroupInput for builtin::websearch_brave in ollama.py - Auto-generate run-with-safety.yaml using distro codegen - Follow proper template pattern instead of manual YAML editing - Resolves precommit issues and reviewer feedback Fixes #2606 --- llama_stack/templates/ollama/ollama.py | 4 ++++ llama_stack/templates/ollama/run-with-safety.yaml | 2 ++ 2 files changed, 6 insertions(+) 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