mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-13 13:42:35 +00:00
fix test
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com> updating structure of default Signed-off-by: Francisco Javier Arceo <farceo@redhat.com> fix model id creation Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
parent
b3addc94d1
commit
7ffd20d112
10 changed files with 119 additions and 62 deletions
|
|
@ -255,4 +255,7 @@ server:
|
|||
telemetry:
|
||||
enabled: true
|
||||
vector_stores:
|
||||
embedding_model_id: sentence-transformers/nomic-ai/nomic-embed-text-v1.5
|
||||
default_provider_id: faiss
|
||||
default_embedding_model:
|
||||
provider_id: sentence-transformers
|
||||
model_id: nomic-ai/nomic-embed-text-v1.5
|
||||
|
|
|
|||
|
|
@ -9,6 +9,7 @@ from typing import Any
|
|||
|
||||
from llama_stack.core.datatypes import (
|
||||
BuildProvider,
|
||||
DefaultEmbeddingModel,
|
||||
Provider,
|
||||
ProviderSpec,
|
||||
ShieldInput,
|
||||
|
|
@ -249,7 +250,11 @@ def get_distribution_template(name: str = "starter") -> DistributionTemplate:
|
|||
default_tool_groups=default_tool_groups,
|
||||
default_shields=default_shields,
|
||||
vector_stores_config=VectorStoresConfig(
|
||||
embedding_model_id="sentence-transformers/nomic-ai/nomic-embed-text-v1.5"
|
||||
default_provider_id="faiss",
|
||||
default_embedding_model=DefaultEmbeddingModel(
|
||||
provider_id="sentence-transformers",
|
||||
model_id="nomic-ai/nomic-embed-text-v1.5",
|
||||
),
|
||||
),
|
||||
),
|
||||
},
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue