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

solved more nitpick comments

Browse source
This commit is contained in:
Geoff-Robin 2025-09-15 16:30:58 +05:30 • committed by Igor Ilic
parent 93c733e687
commit 1e59f1594c
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -26,7 +26,10 @@ async def ingest_database_schema(
max_sample_rows: Maximum sample rows per table max_sample_rows: Maximum sample rows per table
Returns: Returns:
List of created DataPoint objects Dict with keys:
"database_schema": DatabaseSchema
"schema_tables": List[SchemaTable]
"relationships": List[SchemaRelationship]
""" """
engine = create_relational_engine( engine = create_relational_engine(
db_path=database_config.get("migration_db_path", ""), db_path=database_config.get("migration_db_path", ""),
@ -48,7 +51,6 @@ async def ingest_database_schema(
qi = engine.engine.dialect.identifier_preparer.quote qi = engine.engine.dialect.identifier_preparer.quote
qname = lambda name : ".".join(qi(p) for p in name.split(".")) qname = lambda name : ".".join(qi(p) for p in name.split("."))
tn = qname(table_name) tn = qname(table_name)
tn = qi(table_name)
rows_result = await cursor.execute( rows_result = await cursor.execute(
text(f"SELECT * FROM {tn} LIMIT :limit;"), text(f"SELECT * FROM {tn} LIMIT :limit;"),
{"limit": max_sample_rows} {"limit": max_sample_rows}
@ -61,7 +63,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=f"{schema_name}:{tn}"), 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"],