| from core.rag.datasource.vdb.milvus.milvus_vector import MilvusConfig, MilvusVector | from core.rag.datasource.vdb.milvus.milvus_vector import MilvusConfig, MilvusVector | ||||
| from tests.integration_tests.vdb.test_vector_store import ( | from tests.integration_tests.vdb.test_vector_store import ( | ||||
| AbstractTestVector, | |||||
| AbstractVectorTest, | |||||
| get_example_text, | get_example_text, | ||||
| setup_mock_redis, | setup_mock_redis, | ||||
| ) | ) | ||||
| class TestMilvusVector(AbstractTestVector): | |||||
| class MilvusVectorTest(AbstractVectorTest): | |||||
| def __init__(self): | def __init__(self): | ||||
| super().__init__() | super().__init__() | ||||
| self.vector = MilvusVector( | self.vector = MilvusVector( | ||||
| def test_milvus_vector(setup_mock_redis): | def test_milvus_vector(setup_mock_redis): | ||||
| TestMilvusVector().run_all_tests() | |||||
| MilvusVectorTest().run_all_tests() |
| from core.rag.datasource.vdb.qdrant.qdrant_vector import QdrantConfig, QdrantVector | from core.rag.datasource.vdb.qdrant.qdrant_vector import QdrantConfig, QdrantVector | ||||
| from tests.integration_tests.vdb.test_vector_store import ( | from tests.integration_tests.vdb.test_vector_store import ( | ||||
| AbstractTestVector, | |||||
| AbstractVectorTest, | |||||
| setup_mock_redis, | setup_mock_redis, | ||||
| ) | ) | ||||
| class TestQdrantVector(AbstractTestVector): | |||||
| class QdrantVectorTest(AbstractVectorTest): | |||||
| def __init__(self): | def __init__(self): | ||||
| super().__init__() | super().__init__() | ||||
| self.attributes = ['doc_id', 'dataset_id', 'document_id', 'doc_hash'] | self.attributes = ['doc_id', 'dataset_id', 'document_id', 'doc_hash'] | ||||
| def test_qdrant_vector(setup_mock_redis): | def test_qdrant_vector(setup_mock_redis): | ||||
| TestQdrantVector().run_all_tests() | |||||
| QdrantVectorTest().run_all_tests() |
| ext_redis.redis_client.lock = mock_redis_lock | ext_redis.redis_client.lock = mock_redis_lock | ||||
| class AbstractTestVector: | |||||
| class AbstractVectorTest: | |||||
| def __init__(self): | def __init__(self): | ||||
| self.vector = None | self.vector = None | ||||
| self.dataset_id = str(uuid.uuid4()) | self.dataset_id = str(uuid.uuid4()) |
| from core.rag.datasource.vdb.weaviate.weaviate_vector import WeaviateConfig, WeaviateVector | from core.rag.datasource.vdb.weaviate.weaviate_vector import WeaviateConfig, WeaviateVector | ||||
| from tests.integration_tests.vdb.test_vector_store import ( | from tests.integration_tests.vdb.test_vector_store import ( | ||||
| AbstractTestVector, | |||||
| AbstractVectorTest, | |||||
| setup_mock_redis, | setup_mock_redis, | ||||
| ) | ) | ||||
| class TestWeaviateVector(AbstractTestVector): | |||||
| class WeaviateVectorTest(AbstractVectorTest): | |||||
| def __init__(self): | def __init__(self): | ||||
| super().__init__() | super().__init__() | ||||
| self.attributes = ['doc_id', 'dataset_id', 'document_id', 'doc_hash'] | self.attributes = ['doc_id', 'dataset_id', 'document_id', 'doc_hash'] | ||||
| def test_weaviate_vector(setup_mock_redis): | def test_weaviate_vector(setup_mock_redis): | ||||
| TestWeaviateVector().run_all_tests() | |||||
| WeaviateVectorTest().run_all_tests() |