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