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

more nitpick comments solved

Browse source
This commit is contained in:
Geoff-Robin 2025-09-15 12:55:38 +05:30 • committed by Igor Ilic
parent 7cf4a0daeb
commit 60016a6b09
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cognee/tasks/schema/ingest_database_schema.py
View file

@ -47,8 +47,11 @@ async def ingest_database_schema(
async with engine.engine.begin() as cursor: async with engine.engine.begin() as cursor:
for table_name, details in schema.items(): for table_name, details in schema.items():
qi = engine.engine.dialect.identifier_preparer.quote
tn = qi(table_name)
rows_result = await cursor.execute( 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 = [ rows = [
dict(zip([col["name"] for col in details["columns"]], row)) 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() row_count_estimate = count_result.scalar()
schema_table = SchemaTable( schema_table = SchemaTable(
id=uuid5(NAMESPACE_OID, name=table_name), id=uuid5(NAMESPACE_OID, name=f"{schema_name}:{table_name}"),
table_name=table_name, table_name=table_name,
schema_name=schema_name, schema_name=schema_name,
columns=details["columns"], columns=details["columns"],