phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-12 20:12:33 +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 b95f095a54
commit 429f1d2405
No known key found for this signature in database
GPG key ID: C7611ACB4FECAD54
51 changed files with 16149 additions and 83 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/unit/providers/vector_io/conftest.py
View file

@ -20,7 +20,7 @@ from llama_stack.providers.remote.vector_io.pgvector.config import PGVectorVecto
from llama_stack.providers.remote.vector_io.pgvector.pgvector import PGVectorIndex, PGVectorVectorIOAdapter
from llama_stack.providers.utils.kvstore.config import SqliteKVStoreConfig
EMBEDDING_DIMENSION = 384
EMBEDDING_DIMENSION = 768
COLLECTION_PREFIX = "test_collection"

2
tests/unit/providers/vector_io/test_faiss.py
View file

@ -39,7 +39,7 @@ def loop():
@pytest.fixture
def embedding_dimension():
return 384
return 768
@pytest.fixture

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

@ -23,8 +23,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",
)
@ -86,8 +86,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",
)
@ -112,8 +112,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",
)
@ -122,7 +122,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
)
@ -140,8 +140,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}",
)
@ -169,8 +169,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",
)
@ -207,8 +207,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",
)