Underground-Digital/Workflow-Engine

1from core.rag.datasource.vdb.pgvector.pgvector import PGVector, PGVectorConfig2from tests.integration_tests.vdb.test_vector_store import (3    AbstractVectorTest,4    get_example_text,5    setup_mock_redis,6)7 8 9class PGVectorTest(AbstractVectorTest):10    def __init__(self):11        super().__init__()12        self.vector = PGVector(13            collection_name=self.collection_name,14            config=PGVectorConfig(15                host="localhost",16                port=5433,17                user="postgres",18                password="difyai123456",19                database="dify",20                min_connection=1,21                max_connection=5,22            ),23        )24 25 26def test_pgvector(setup_mock_redis):27    PGVectorTest().run_all_tests()28