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
This commit is contained in:
skamenan7 2025-07-05 09:16:13 -04:00
parent 9e8802533d
commit ddcf74f59c
2 changed files with 6 additions and 0 deletions

View file

@ -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",

View file

@ -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