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

fix: minor changes from PR comments

Browse source
This commit is contained in:
Andrej Milicevic 2026-01-16 12:59:28 +01:00
parent c6ff0eca3d
commit 8030c949f2
3 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
.github/workflows/vector_db_tests.yml vendored
View file

@ -149,7 +149,7 @@ jobs:
VECTOR_DB_PORT: 5432 VECTOR_DB_PORT: 5432
VECTOR_DB_USERNAME: cognee VECTOR_DB_USERNAME: cognee
VECTOR_DB_PASSWORD: ${{ secrets.POSTGRES_PASSWORD }} VECTOR_DB_PASSWORD: ${{ secrets.POSTGRES_PASSWORD }}
VECTOR_DATASET_DATABASE_HANDLER: pgvector VECTOR_DATASET_DATABASE_HANDLER: pgvector_local
run: uv run python ./cognee/tests/test_permissions.py run: uv run python ./cognee/tests/test_permissions.py
run-lancedb-tests: run-lancedb-tests:

2
cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py
View file

@ -559,7 +559,7 @@ class SQLAlchemyAdapter:
# Connect to maintenance db in order to create new database # Connect to maintenance db in order to create new database
# Make sure to execute CREATE DATABASE outside of transaction block, and set AUTOCOMMIT isolation level # Make sure to execute CREATE DATABASE outside of transaction block, and set AUTOCOMMIT isolation level
connection = await self.maintenance_engine.connect() connection = await self.maintenance_engine.connect()
await connection.execution_options(isolation_level="AUTOCOMMIT") connection = await connection.execution_options(isolation_level="AUTOCOMMIT")
await connection.execute(text(f'CREATE DATABASE "{self.engine.url.database}";')) await connection.execute(text(f'CREATE DATABASE "{self.engine.url.database}";'))
# Clean up resources # Clean up resources

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

@ -33,7 +33,7 @@ class PGVectorDatasetDatabaseHandler(DatasetDatabaseHandlerInterface):
"username": vector_config.vector_db_username, "username": vector_config.vector_db_username,
"password": vector_config.vector_db_password, "password": vector_config.vector_db_password,
}, },
"vector_dataset_database_handler": "pgvector", "vector_dataset_database_handler": "pgvector_local",
} }
from .create_db_and_tables import create_pg_database from .create_db_and_tables import create_pg_database