phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: add Brave search tool group in ollama.py template

Browse source 
- 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
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/templates/ollama/ollama.py
View file

@ -83,6 +83,10 @@ def get_distribution_template() -> DistributionTemplate:
toolgroup_id="builtin::websearch", toolgroup_id="builtin::websearch",
provider_id="tavily-search", provider_id="tavily-search",
), ),
ToolGroupInput(
toolgroup_id="builtin::websearch_brave",
provider_id="brave-search",
),
ToolGroupInput( ToolGroupInput(
toolgroup_id="builtin::rag", toolgroup_id="builtin::rag",
provider_id="rag-runtime", provider_id="rag-runtime",

2
llama_stack/templates/ollama/run-with-safety.yaml
View file

@ -150,6 +150,8 @@ benchmarks: []
tool_groups: tool_groups:
- toolgroup_id: builtin::websearch - toolgroup_id: builtin::websearch
provider_id: tavily-search provider_id: tavily-search
- toolgroup_id: builtin::websearch_brave
provider_id: brave-search
- toolgroup_id: builtin::rag - toolgroup_id: builtin::rag
provider_id: rag-runtime provider_id: rag-runtime
- toolgroup_id: builtin::wolfram_alpha - toolgroup_id: builtin::wolfram_alpha