From 7cf4a0daeb260318bb2f9db9c32b69c2187f1db5 Mon Sep 17 00:00:00 2001 From: Geoff-Robin Date: Mon, 15 Sep 2025 11:12:43 +0530 Subject: [PATCH] id mismatch risk negated --- cognee/tasks/ingestion/migrate_relational_database.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/cognee/tasks/ingestion/migrate_relational_database.py b/cognee/tasks/ingestion/migrate_relational_database.py index 62a8a0eac..e857ab34d 100644 --- a/cognee/tasks/ingestion/migrate_relational_database.py +++ b/cognee/tasks/ingestion/migrate_relational_database.py @@ -66,9 +66,10 @@ async def migrate_relational_database( ), ) ) + table_name_to_id = {t.table_name: t.id for t in schema_tables} for rel in schema_relationships: - source_table_id = uuid5(NAMESPACE_OID, name=rel.source_table) - target_table_id = uuid5(NAMESPACE_OID, name=rel.target_table) + source_table_id = table_name_to_id.get(rel.source_table) + target_table_id = table_name_to_id.get(rel.target_table) relationship_id = rel.id