Merge 376eb94def into d266c59c2a

tests/unit/providers/vector_io/conftest.py

@@ -29,7 +29,7 @@ from llama_stack.providers.remote.vector_io.qdrant.qdrant import QdrantVectorIOA
 from llama_stack.providers.remote.vector_io.weaviate.config import WeaviateVectorIOConfig
 from llama_stack.providers.remote.vector_io.weaviate.weaviate import WeaviateIndex, WeaviateVectorIOAdapter
 
-EMBEDDING_DIMENSION = 384
+EMBEDDING_DIMENSION = 768
 COLLECTION_PREFIX = "test_collection"
 MILVUS_ALIAS = "test_milvus"
 

tests/unit/providers/vector_io/remote/test_pgvector.py

@@ -23,7 +23,7 @@ def loop():
 @pytest.fixture
 def embedding_dimension():
     """Default embedding dimension for tests."""
-    return 384
+    return 768
 
 
 @pytest.fixture

tests/unit/providers/vector_io/test_faiss.py

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

tests/unit/providers/vector_io/test_qdrant.py

@@ -52,11 +52,9 @@ def mock_vector_db(vector_db_id) -> MagicMock:
     mock_vector_db = MagicMock(spec=VectorDB)
     mock_vector_db.embedding_model = "embedding_model"
     mock_vector_db.identifier = vector_db_id
-    mock_vector_db.embedding_dimension = 384
+    mock_vector_db.embedding_dimension = 768
     mock_vector_db.model_dump_json.return_value = (
-        '{"identifier": "'
-        + vector_db_id
-        + '", "provider_id": "qdrant", "embedding_model": "embedding_model", "embedding_dimension": 384}'
+        '{"identifier": "' + vector_db_id + '", "embedding_model": "embedding_model", "embedding_dimension": 768}'
     )
     return mock_vector_db
 

tests/unit/registry/test_registry.py

@@ -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",
     )