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

fix integration tests and add tests for multiple providers

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-10-13 12:13:43 -04:00
parent 1bb3434562
commit 9166669b83
3 changed files with 80 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llama_stack/providers/utils/memory/openai_vector_store_mixin.py
View file

@ -353,14 +353,12 @@ class OpenAIVectorStoreMixin(ABC):
provider_vector_db_id = extra.get("provider_vector_db_id")
embedding_model = extra.get("embedding_model")
embedding_dimension = extra.get("embedding_dimension", 384)
provider_id = extra.get("provider_id")
# use provider_id from router or default to this provider's own ID (need for --stack-config)
provider_id = extra.get("provider_id") or getattr(self, "__provider_id__", None)
# Derive the canonical vector_db_id (allow override, else generate)
vector_db_id = provider_vector_db_id or generate_object_id("vector_store", lambda: f"vs_{uuid.uuid4()}")
if provider_id is None:
raise ValueError("Provider ID is required")
if embedding_model is None:
raise ValueError("Embedding model is required")
@ -369,6 +367,9 @@ class OpenAIVectorStoreMixin(ABC):
raise ValueError("Embedding dimension is required")
# Register the VectorDB backing this vector store
if provider_id is None:
raise ValueError("Provider ID is required but was not provided")
vector_db = VectorDB(
identifier=vector_db_id,
embedding_dimension=embedding_dimension,