Refactor test_pgvector.py to use AbstractVectorTest instead of Abstra…

langgenius · Apr 29, 2024 · 76dc6a4 · 76dc6a4
1 parent 20848cb
commit 76dc6a4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/tests/integration_tests/vdb/pgvector/test_pgvector.py b/api/tests/integration_tests/vdb/pgvector/test_pgvector.py
@@ -1,13 +1,13 @@
 from core.rag.datasource.vdb.pgvector.pgvector import PGVector, PGVectorConfig
 from core.rag.models.document import Document
 from tests.integration_tests.vdb.test_vector_store import (
-    AbstractTestVector,
+    AbstractVectorTest,
     get_example_text,
     setup_mock_redis,
 )
 
 
-class TestPGVector(AbstractTestVector):
+class TestPGVector(AbstractVectorTest):
     def __init__(self):
         super().__init__()
         self.vector = PGVector(