1818
1919def test_create_lancedb_vector_store ():
2020 kwargs = {
21- "collection_name" : "test_collection" ,
2221 "db_uri" : "/tmp/lancedb" ,
2322 }
2423 vector_store = VectorStoreFactory .create_vector_store (
2524 vector_store_type = VectorStoreType .LanceDB .value ,
26- vector_store_schema_config = VectorStoreSchemaConfig (),
25+ vector_store_schema_config = VectorStoreSchemaConfig (index_name = "test_collection" ),
2726 kwargs = kwargs ,
2827 )
2928 assert isinstance (vector_store , LanceDBVectorStore )
@@ -33,13 +32,12 @@ def test_create_lancedb_vector_store():
3332@pytest .mark .skip (reason = "Azure AI Search requires credentials and setup" )
3433def test_create_azure_ai_search_vector_store ():
3534 kwargs = {
36- "collection_name" : "test_collection" ,
3735 "url" : "https://test.search.windows.net" ,
3836 "api_key" : "test_key" ,
3937 }
4038 vector_store = VectorStoreFactory .create_vector_store (
4139 vector_store_type = VectorStoreType .AzureAISearch .value ,
42- vector_store_schema_config = VectorStoreSchemaConfig (),
40+ vector_store_schema_config = VectorStoreSchemaConfig (index_name = "test_collection" ),
4341 kwargs = kwargs ,
4442 )
4543 assert isinstance (vector_store , AzureAISearchVectorStore )
@@ -48,14 +46,13 @@ def test_create_azure_ai_search_vector_store():
4846@pytest .mark .skip (reason = "CosmosDB requires credentials and setup" )
4947def test_create_cosmosdb_vector_store ():
5048 kwargs = {
51- "collection_name" : "test_collection" ,
5249 "connection_string" : "AccountEndpoint=https://test.documents.azure.com:443/;AccountKey=test_key==" ,
5350 "database_name" : "test_db" ,
5451 }
5552
5653 vector_store = VectorStoreFactory .create_vector_store (
5754 vector_store_type = VectorStoreType .CosmosDB .value ,
58- vector_store_schema_config = VectorStoreSchemaConfig (),
55+ vector_store_schema_config = VectorStoreSchemaConfig (index_name = "test_collection" ),
5956 kwargs = kwargs ,
6057 )
6158
0 commit comments