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

fix: Add metadata reflection fix to sqlite as well

Browse source 
Added fix when reflecting metadata to sqlite as well

Fix
This commit is contained in:
Igor Ilic 2024-12-16 11:26:33 +01:00
parent d9e558e885
commit 394a0b2dfb
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -113,10 +113,12 @@ class SQLAlchemyAdapter():
""" """
async with self.engine.begin() as connection: async with self.engine.begin() as connection:
if self.engine.dialect.name == "sqlite": if self.engine.dialect.name == "sqlite":
# Load the schema information into the MetaData object # Create a MetaData instance to load table information
await connection.run_sync(Base.metadata.reflect) metadata = MetaData()
if table_name in Base.metadata.tables: # Load table information from schema into MetaData
return Base.metadata.tables[table_name] await connection.run_sync(metadata.reflect)
if table_name in metadata.tables:
return metadata.tables[table_name]
else: else:
raise EntityNotFoundError(message=f"Table '{table_name}' not found.") raise EntityNotFoundError(message=f"Table '{table_name}' not found.")
else: else:
@ -138,8 +140,11 @@ class SQLAlchemyAdapter():
table_names = [] table_names = []
async with self.engine.begin() as connection: async with self.engine.begin() as connection:
if self.engine.dialect.name == "sqlite": if self.engine.dialect.name == "sqlite":
await connection.run_sync(Base.metadata.reflect) # Create a MetaData instance to load table information
for table in Base.metadata.tables: metadata = MetaData()
# Load table information from schema into MetaData
await connection.run_sync(metadata.reflect)
for table in metadata.tables:
table_names.append(str(table)) table_names.append(str(table))
else: else:
schema_list = await self.get_schema_list() schema_list = await self.get_schema_list()