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

updated vector io tests to use extra

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-10-10 17:01:44 -04:00
parent c730078b27
commit 7bbb1f8217
2 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/core/resolver.py
View file

@ -54,7 +54,6 @@ from llama_stack.providers.datatypes import (
ScoringFunctionsProtocolPrivate, ScoringFunctionsProtocolPrivate,
ShieldsProtocolPrivate, ShieldsProtocolPrivate,
ToolGroupsProtocolPrivate, ToolGroupsProtocolPrivate,
VectorDBsProtocolPrivate,
) )
logger = get_logger(name=__name__, category="core") logger = get_logger(name=__name__, category="core")
@ -123,7 +122,6 @@ def additional_protocols_map() -> dict[Api, Any]:
return { return {
Api.inference: (ModelsProtocolPrivate, Models, Api.models), Api.inference: (ModelsProtocolPrivate, Models, Api.models),
Api.tool_groups: (ToolGroupsProtocolPrivate, ToolGroups, Api.tool_groups), Api.tool_groups: (ToolGroupsProtocolPrivate, ToolGroups, Api.tool_groups),
Api.vector_io: (VectorDBsProtocolPrivate,),
Api.safety: (ShieldsProtocolPrivate, Shields, Api.shields), Api.safety: (ShieldsProtocolPrivate, Shields, Api.shields),
Api.datasetio: (DatasetsProtocolPrivate, Datasets, Api.datasets), Api.datasetio: (DatasetsProtocolPrivate, Datasets, Api.datasets),
Api.scoring: ( Api.scoring: (

5
tests/integration/vector_io/test_vector_io.py
View file

@ -53,6 +53,7 @@ def test_vector_db_retrieve(client_with_empty_registry, embedding_model_id, embe
extra_body={ extra_body={
"embedding_model": embedding_model_id, "embedding_model": embedding_model_id,
"embedding_dimension": embedding_dimension, "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={ extra_body={
"embedding_model": embedding_model_id, "embedding_model": embedding_model_id,
"embedding_dimension": embedding_dimension, "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={ extra_body={
"embedding_model": embedding_model_id, "embedding_model": embedding_model_id,
"embedding_dimension": embedding_dimension, "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={ extra_body={
"embedding_model": embedding_model_id, "embedding_model": embedding_model_id,
"embedding_dimension": embedding_dimension, "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={ extra_body={
"embedding_model": embedding_model_id, "embedding_model": embedding_model_id,
"embedding_dimension": embedding_dimension, "embedding_dimension": embedding_dimension,
"provider_id": "my_provider",
}, },
) )