gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Change to new syntax for vector_engine_provider

Browse source 
Syntax changed for vector_engine_provider to vector_db_provider

Fix #COG-170
This commit is contained in:
Igor Ilic 2024-10-22 12:57:17 +02:00
parent dbc86e284a
commit c7ed46ddaa
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/infrastructure/databases/vector/pgvector/create_db_and_tables.py
View file

@ -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)

2
cognee/tests/test_pgvector.py
View file

@ -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": "",