From d30c337370d3afe82dbaf6dde133ee56b8c7a91c Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Tue, 22 Oct 2024 14:23:36 +0200 Subject: [PATCH] refactor: Use SQLAlchemyAdapter create_database Added reusing of sqlalchemy adapters create_database method instead of directly writing code for database creation Refactor #COG-170 --- .../databases/vector/pgvector/create_db_and_tables.py | 3 +-- 1 file changed, 1 insertion(+), 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 dd029de0a..99d53d698 100644 --- a/cognee/infrastructure/databases/vector/pgvector/create_db_and_tables.py +++ b/cognee/infrastructure/databases/vector/pgvector/create_db_and_tables.py @@ -7,9 +7,8 @@ async def create_db_and_tables(): vector_engine = get_vector_engine() if vector_config.vector_db_provider == "pgvector": + vector_engine.create_database() async with vector_engine.engine.begin() as connection: - if len(Base.metadata.tables.keys()) > 0: - await connection.run_sync(Base.metadata.create_all) await connection.execute(text("CREATE EXTENSION IF NOT EXISTS vector;"))