hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-24 18:23:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: add tests covering all methods of vector store (#3849)

Browse Source
This commit is contained in:
Bowen Liang
2024-04-25 22:27:30 +08:00
committed by GitHub
parent 8bca908f15
commit 45dd1683fd
5 changed files with 48 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/tests/integration_tests/vdb/milvus/test_milvus.py
View File

@@ -24,6 +24,13 @@ class TestMilvusVector(AbstractTestVector):
hits_by_full_text = self.vector.search_by_full_text(query=get_sample_text())
assert len(hits_by_full_text) == 0
def delete_document_by_id(self):
self.vector.delete_by_document_id(self.dataset_id)
def get_ids_by_metadata_field(self):
ids = self.vector.get_ids_by_metadata_field('document_id', self.dataset_id)
assert len(ids) >= 1
def test_milvus_vector(setup_mock_redis):
TestMilvusVector().run_all_test()
TestMilvusVector().run_all_tests()