phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-17 06:52:36 +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

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={
"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",
},
)