From 7bbb1f8217f4b07ecce7bb6466e9cfb1c499dcfd Mon Sep 17 00:00:00 2001 From: Francisco Javier Arceo Date: Fri, 10 Oct 2025 17:01:44 -0400 Subject: [PATCH] updated vector io tests to use extra Signed-off-by: Francisco Javier Arceo --- llama_stack/core/resolver.py | 2 -- tests/integration/vector_io/test_vector_io.py | 5 +++++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/llama_stack/core/resolver.py b/llama_stack/core/resolver.py index c8011307d..6bc7a36f6 100644 --- a/llama_stack/core/resolver.py +++ b/llama_stack/core/resolver.py @@ -54,7 +54,6 @@ from llama_stack.providers.datatypes import ( ScoringFunctionsProtocolPrivate, ShieldsProtocolPrivate, ToolGroupsProtocolPrivate, - VectorDBsProtocolPrivate, ) logger = get_logger(name=__name__, category="core") @@ -123,7 +122,6 @@ def additional_protocols_map() -> dict[Api, Any]: return { Api.inference: (ModelsProtocolPrivate, Models, Api.models), Api.tool_groups: (ToolGroupsProtocolPrivate, ToolGroups, Api.tool_groups), - Api.vector_io: (VectorDBsProtocolPrivate,), Api.safety: (ShieldsProtocolPrivate, Shields, Api.shields), Api.datasetio: (DatasetsProtocolPrivate, Datasets, Api.datasets), Api.scoring: ( diff --git a/tests/integration/vector_io/test_vector_io.py b/tests/integration/vector_io/test_vector_io.py index 284a3bffb..f2205ed0a 100644 --- a/tests/integration/vector_io/test_vector_io.py +++ b/tests/integration/vector_io/test_vector_io.py @@ -53,6 +53,7 @@ def test_vector_db_retrieve(client_with_empty_registry, embedding_model_id, embe extra_body={ "embedding_model": embedding_model_id, "embedding_dimension": embedding_dimension, + "provider_id": "my_provider", }, ) @@ -73,6 +74,7 @@ def test_vector_db_register(client_with_empty_registry, embedding_model_id, embe extra_body={ "embedding_model": embedding_model_id, "embedding_dimension": embedding_dimension, + "provider_id": "my_provider", }, ) @@ -109,6 +111,7 @@ def test_insert_chunks(client_with_empty_registry, embedding_model_id, embedding extra_body={ "embedding_model": embedding_model_id, "embedding_dimension": embedding_dimension, + "provider_id": "my_provider", }, ) @@ -150,6 +153,7 @@ def test_insert_chunks_with_precomputed_embeddings(client_with_empty_registry, e extra_body={ "embedding_model": embedding_model_id, "embedding_dimension": embedding_dimension, + "provider_id": "my_provider", }, ) @@ -199,6 +203,7 @@ def test_query_returns_valid_object_when_identical_to_embedding_in_vdb( extra_body={ "embedding_model": embedding_model_id, "embedding_dimension": embedding_dimension, + "provider_id": "my_provider", }, )