phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-16 13:52:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 9658581cf7 into 9c223d8593

Browse source
This commit is contained in:
Christian Zaccaria 2025-10-27 16:22:42 +01:00 • committed by GitHub
commit b85f179951
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 187 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/core/routers/vector_io.py
View file

@ -105,6 +105,7 @@ class VectorIORouter(VectorIO):
embedding_model = extra.get("embedding_model")
embedding_dimension = extra.get("embedding_dimension")
provider_id = extra.get("provider_id")
distance_metric = extra.get("distance_metric")
# Use default embedding model if not specified
if (
@ -154,6 +155,7 @@ class VectorIORouter(VectorIO):
provider_id=provider_id,
provider_vector_store_id=vector_store_id,
vector_store_name=params.name,
distance_metric=distance_metric,
)
provider = await self.routing_table.get_provider_impl(registered_vector_store.identifier)
@ -162,6 +164,8 @@ class VectorIORouter(VectorIO):
params.model_extra = {}
params.model_extra["provider_vector_store_id"] = registered_vector_store.provider_resource_id
params.model_extra["provider_id"] = registered_vector_store.provider_id
if distance_metric is not None:
params.model_extra["distance_metric"] = distance_metric
if embedding_model is not None:
params.model_extra["embedding_model"] = embedding_model
if embedding_dimension is not None: