Skip to content

Commit ad74dad

Browse files
chore: replace debug print() with warning_logger in database_kuzu.py
1 parent 34469e6 commit ad74dad

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/codegraphcontext/core/database_kuzu.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def _initialize_schema(self):
124124
self._conn.execute(f"CREATE NODE TABLE `{table_name}`({schema})")
125125
except Exception as e:
126126
if "already exists" not in str(e).lower():
127-
print(f"Kuzu Schema Node Error ({table_name}): {e}")
127+
warning_logger(f"Kuzu Schema Node Error ({table_name}): {e}")
128128
debug_log(f"Kuzu Schema Node Error ({table_name}): {e}")
129129

130130
for table_name, schema in rel_tables:
@@ -133,7 +133,7 @@ def _initialize_schema(self):
133133
self._conn.execute(f"CREATE REL TABLE `{table_name}`({schema})")
134134
except Exception as e:
135135
if "already exists" not in str(e).lower():
136-
print(f"Kuzu Schema Rel Error ({table_name}): {e}")
136+
warning_logger(f"Kuzu Schema Rel Error ({table_name}): {e}")
137137
debug_log(f"Kuzu Schema Rel Error ({table_name}): {e}")
138138

139139
def close_driver(self):
@@ -315,7 +315,7 @@ def _translate_query(self, query: str, parameters: Dict[str, Any]) -> Tuple[str,
315315
query = query.replace(old_block, new_block)
316316
# print(f"DEBUG: Replaced MERGE block: {old_block} -> {new_block}")
317317
else:
318-
print(f"DEBUG_FAILURE: Could not find old_block in query: '{old_block}'")
318+
warning_logger(f"Kuzu UID injection: could not find props block in query for label '{label}'")
319319

320320
parameters[uid_param] = uid_val
321321
# print(f"DEBUG: Injected UID for {label}: {uid_val}")

0 commit comments

Comments
 (0)