llama-stack-mirror/llama_stack/templates
Ashwin Bharambe 11697f85c5
fix: pull ollama embedding model if necessary (#1209)
Embedding models are tiny and can be pulled on-demand. Let's do that so
the user doesn't have to do "yet another thing" to get themselves set
up.

Thanks @hardikjshah for the suggestion.

Also fixed a build dependency miss (TODO: distro_codegen needs to
actually check that the build template contains all providers mentioned
for the run.yaml file)

## Test Plan 

First run `ollama rm all-minilm:latest`. 

Run `llama stack build --template ollama && llama stack run ollama --env
INFERENCE_MODEL=llama3.2:3b-instruct-fp16`. See that it outputs a
"Pulling embedding model `all-minilm:latest`" output and the stack
starts up correctly. Verify that `ollama list` shows the model is
correctly downloaded.
2025-02-21 10:35:56 -08:00
..
bedrock ModelAlias -> ProviderModelEntry 2025-02-20 14:02:36 -08:00
cerebras ModelAlias -> ProviderModelEntry 2025-02-20 14:02:36 -08:00
dell fix!: update eval-tasks -> benchmarks (#1032) 2025-02-13 16:40:58 -08:00
experimental-post-training fix!: update eval-tasks -> benchmarks (#1032) 2025-02-13 16:40:58 -08:00
fireworks Update embedding dimension singular 2025-02-20 16:14:46 -08:00
hf-endpoint fix!: update eval-tasks -> benchmarks (#1032) 2025-02-13 16:40:58 -08:00
hf-serverless fix!: update eval-tasks -> benchmarks (#1032) 2025-02-13 16:40:58 -08:00
meta-reference-gpu fix!: update eval-tasks -> benchmarks (#1032) 2025-02-13 16:40:58 -08:00
meta-reference-quantized-gpu fix!: update eval-tasks -> benchmarks (#1032) 2025-02-13 16:40:58 -08:00
nvidia test(client-sdk): Update embedding test types to use latest imports (#1203) 2025-02-21 08:09:17 -08:00
ollama fix: pull ollama embedding model if necessary (#1209) 2025-02-21 10:35:56 -08:00
passthrough feat: inference passthrough provider (#1166) 2025-02-19 21:47:00 -08:00
remote-vllm fix!: update eval-tasks -> benchmarks (#1032) 2025-02-13 16:40:58 -08:00
sambanova ModelAlias -> ProviderModelEntry 2025-02-20 14:02:36 -08:00
tgi fix!: update eval-tasks -> benchmarks (#1032) 2025-02-13 16:40:58 -08:00
together Update embedding dimension singular 2025-02-20 16:14:46 -08:00
vllm-gpu fix!: update eval-tasks -> benchmarks (#1032) 2025-02-13 16:40:58 -08:00
__init__.py Auto-generate distro yamls + docs (#468) 2024-11-18 14:57:06 -08:00
template.py build: format codebase imports using ruff linter (#1028) 2025-02-13 10:06:21 -08:00