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

refactor: formatting fix for PR

Browse source 
Fixed formatting for files in pull request

Refactor COG-678
This commit is contained in:
Igor Ilic 2024-11-28 20:22:55 +01:00
parent 56367cb0c3
commit b56a1dee59
2 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -188,7 +188,7 @@ class SQLAlchemyAdapter():
result = await session.execute(query) result = await session.execute(query)
# Fetch all rows as a list of dictionaries # Fetch all rows as a list of dictionaries
rows = result.mappings().all() # Use `.mappings()` to get key-value pairs rows = result.mappings().all()
return rows return rows
async def execute_query(self, query): async def execute_query(self, query):

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

@ -6,15 +6,15 @@ from sqlalchemy import text
created=False created=False
async def create_db_and_tables(): async def create_db_and_tables():
vector_config = get_vectordb_config() vector_config = get_vectordb_config()
vector_engine = get_vector_engine() vector_engine = get_vector_engine()
if vector_config.vector_db_provider == "pgvector": if vector_config.vector_db_provider == "pgvector":
global created global created
if not created: if not created:
await vector_engine.create_database() await vector_engine.create_database()
created = True created = True
async with vector_engine.engine.begin() as connection: async with vector_engine.engine.begin() as connection:
await connection.execute(text("CREATE EXTENSION IF NOT EXISTS vector;")) await connection.execute(text("CREATE EXTENSION IF NOT EXISTS vector;"))