From c7ed46ddaab4c91bb7dd8e9f360ea6397416860d Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Tue, 22 Oct 2024 12:57:17 +0200 Subject: [PATCH] fix: Change to new syntax for vector_engine_provider Syntax changed for vector_engine_provider to vector_db_provider Fix #COG-170 --- .../databases/vector/pgvector/create_db_and_tables.py | 2 +- cognee/tests/test_pgvector.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/infrastructure/databases/vector/pgvector/create_db_and_tables.py b/cognee/infrastructure/databases/vector/pgvector/create_db_and_tables.py index 1b09e873c..dd029de0a 100644 --- a/cognee/infrastructure/databases/vector/pgvector/create_db_and_tables.py +++ b/cognee/infrastructure/databases/vector/pgvector/create_db_and_tables.py @@ -6,7 +6,7 @@ async def create_db_and_tables(): vector_config = get_vectordb_config() vector_engine = get_vector_engine() - if vector_config.vector_engine_provider == "pgvector": + if vector_config.vector_db_provider == "pgvector": async with vector_engine.engine.begin() as connection: if len(Base.metadata.tables.keys()) > 0: await connection.run_sync(Base.metadata.create_all) diff --git a/cognee/tests/test_pgvector.py b/cognee/tests/test_pgvector.py index 9105bd8e5..88195bbc1 100644 --- a/cognee/tests/test_pgvector.py +++ b/cognee/tests/test_pgvector.py @@ -11,7 +11,7 @@ logging.basicConfig(level=logging.DEBUG) async def main(): cognee.config.set_vector_db_config({ "vector_db_url": "", "vector_db_key": "", - "vector_engine_provider": "pgvector" + "vector_db_provider": "pgvector" } ) cognee.config.set_relational_db_config({"db_path": "",