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

refactor: switch to the new default nomic-embed-text-v1.5 embedding model in LS

Browse source
This commit is contained in:
r3v5 2025-08-18 10:34:46 +01:00
parent f1748e2f92
commit 1d0f0a0d8e
No known key found for this signature in database
GPG key ID: C7611ACB4FECAD54
63 changed files with 16170 additions and 186 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/unit/registry/test_registry.py
View file

@ -22,8 +22,8 @@ from llama_stack.providers.utils.kvstore.config import SqliteKVStoreConfig
def sample_vector_db():
return VectorDB(
identifier="test_vector_db",
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
embedding_model="nomic-embed-text-v1.5",
embedding_dimension=768,
provider_resource_id="test_vector_db",
provider_id="test-provider",
)
@ -85,8 +85,8 @@ async def test_cached_registry_initialization(sqlite_kvstore, sample_vector_db,
async def test_cached_registry_updates(cached_disk_dist_registry):
new_vector_db = VectorDB(
identifier="test_vector_db_2",
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
embedding_model="nomic-embed-text-v1.5",
embedding_dimension=768,
provider_resource_id="test_vector_db_2",
provider_id="baz",
)
@ -111,8 +111,8 @@ async def test_cached_registry_updates(cached_disk_dist_registry):
async def test_duplicate_provider_registration(cached_disk_dist_registry):
original_vector_db = VectorDB(
identifier="test_vector_db_2",
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
embedding_model="nomic-embed-text-v1.5",
embedding_dimension=768,
provider_resource_id="test_vector_db_2",
provider_id="baz",
)
@ -121,7 +121,7 @@ async def test_duplicate_provider_registration(cached_disk_dist_registry):
duplicate_vector_db = VectorDB(
identifier="test_vector_db_2",
embedding_model="different-model",
embedding_dimension=384,
embedding_dimension=768,
provider_resource_id="test_vector_db_2",
provider_id="baz", # Same provider_id
)
@ -138,8 +138,8 @@ async def test_get_all_objects(cached_disk_dist_registry):
test_vector_dbs = [
VectorDB(
identifier=f"test_vector_db_{i}",
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
embedding_model="nomic-embed-text-v1.5",
embedding_dimension=768,
provider_resource_id=f"test_vector_db_{i}",
provider_id=f"provider_{i}",
)
@ -167,8 +167,8 @@ async def test_get_all_objects(cached_disk_dist_registry):
async def test_parse_registry_values_error_handling(sqlite_kvstore):
valid_db = VectorDB(
identifier="valid_vector_db",
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
embedding_model="nomic-embed-text-v1.5",
embedding_dimension=768,
provider_resource_id="valid_vector_db",
provider_id="test-provider",
)
@ -205,8 +205,8 @@ async def test_parse_registry_values_error_handling(sqlite_kvstore):
async def test_cached_registry_error_handling(sqlite_kvstore):
valid_db = VectorDB(
identifier="valid_cached_db",
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
embedding_model="nomic-embed-text-v1.5",
embedding_dimension=768,
provider_resource_id="valid_cached_db",
provider_id="test-provider",
)