From 60016a6b09a76083ce01baeb38b1559b0acd130a Mon Sep 17 00:00:00 2001 From: Geoff-Robin Date: Mon, 15 Sep 2025 12:55:38 +0530 Subject: [PATCH] more nitpick comments solved --- cognee/tasks/schema/ingest_database_schema.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/cognee/tasks/schema/ingest_database_schema.py b/cognee/tasks/schema/ingest_database_schema.py index c4c13449d..f93314a3d 100644 --- a/cognee/tasks/schema/ingest_database_schema.py +++ b/cognee/tasks/schema/ingest_database_schema.py @@ -47,8 +47,11 @@ async def ingest_database_schema( async with engine.engine.begin() as cursor: for table_name, details in schema.items(): + qi = engine.engine.dialect.identifier_preparer.quote + tn = qi(table_name) rows_result = await cursor.execute( - text(f"SELECT * FROM {table_name} LIMIT {max_sample_rows}") + text(f"SELECT * FROM {tn} LIMIT :limit;"), + {"limit": max_sample_rows} ) rows = [ dict(zip([col["name"] for col in details["columns"]], row)) @@ -58,7 +61,7 @@ async def ingest_database_schema( row_count_estimate = count_result.scalar() schema_table = SchemaTable( - id=uuid5(NAMESPACE_OID, name=table_name), + id=uuid5(NAMESPACE_OID, name=f"{schema_name}:{table_name}"), table_name=table_name, schema_name=schema_name, columns=details["columns"],