mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-25 16:48:04 +00:00
reverting to original call order for a simpler change
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
parent
56811a1483
commit
d361154102
6 changed files with 25 additions and 17 deletions
|
|
@ -53,6 +53,7 @@ def test_vector_db_retrieve(client_with_empty_registry, embedding_model_id, embe
|
|||
vector_db_id=vector_db_id,
|
||||
embedding_model=embedding_model_id,
|
||||
embedding_dimension=embedding_dimension,
|
||||
provider_vector_db_id=vector_db_id,
|
||||
)
|
||||
|
||||
# Retrieve the memory bank and validate its properties
|
||||
|
|
@ -69,6 +70,7 @@ def test_vector_db_register(client_with_empty_registry, embedding_model_id, embe
|
|||
vector_db_id=vector_db_id,
|
||||
embedding_model=embedding_model_id,
|
||||
embedding_dimension=embedding_dimension,
|
||||
provider_vector_db_id=vector_db_id,
|
||||
)
|
||||
|
||||
vector_dbs_after_register = [vector_db.identifier for vector_db in client_with_empty_registry.vector_dbs.list()]
|
||||
|
|
@ -96,6 +98,7 @@ def test_insert_chunks(client_with_empty_registry, embedding_model_id, embedding
|
|||
vector_db_id=vector_db_id,
|
||||
embedding_model=embedding_model_id,
|
||||
embedding_dimension=embedding_dimension,
|
||||
provider_vector_db_id=vector_db_id,
|
||||
)
|
||||
|
||||
client_with_empty_registry.vector_io.insert(
|
||||
|
|
@ -131,6 +134,7 @@ def test_insert_chunks_with_precomputed_embeddings(client_with_empty_registry, e
|
|||
vector_db_id=vector_db_id,
|
||||
embedding_model=embedding_model_id,
|
||||
embedding_dimension=embedding_dimension,
|
||||
provider_vector_db_id=vector_db_id,
|
||||
)
|
||||
|
||||
chunks_with_embeddings = [
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue