test: add unit tests for vector stores of Milvus, Qdrant and Weaviate (#3688)

2024-04-24 21:52:42 +08:00
parent 8ca5aa1190
commit 9cec8c1750
15 changed files with 211 additions and 6 deletions
--- a/api/tests/unit_tests/core/rag/datasource/vdb/milvus/init.py
+++ b/api/tests/unit_tests/core/rag/datasource/vdb/milvus/init.py
--- a/api/tests/unit_tests/core/rag/datasource/vdb/milvus/test_milvus.py
+++ b/api/tests/unit_tests/core/rag/datasource/vdb/milvus/test_milvus.py
@@ -18,7 +18,7 @@ def test_default_value():
        with pytest.raises(ValidationError) as e:
            MilvusConfig(**config)
        assert e.value.errors()[1]['msg'] == f'config MILVUS_{key.upper()} is required'
-    
+
    config = MilvusConfig(**valid_config)
    assert config.secure is False
    assert config.database == 'default'