test: improve vector store tests (#3855)

2024-04-26 19:18:42 +08:00
parent 4d66a86579
commit 045827043d
6 changed files with 58 additions and 74 deletions
--- a/api/tests/integration_tests/vdb/weaviate/test_weaviate.py
+++ b/api/tests/integration_tests/vdb/weaviate/test_weaviate.py
@@ -1,5 +1,4 @@
 from core.rag.datasource.vdb.weaviate.weaviate_vector import WeaviateConfig, WeaviateVector
-from models.dataset import Dataset
 from tests.integration_tests.vdb.test_vector_store import (
    AbstractTestVector,
    setup_mock_redis,