SIENTIAPDE-1222

Enhance MLFlow debug logging to limit output size

- Updated debug logging statements in the MLFlow activity to include a maximum of 5 items and a depth of 5 for the sample dictionary, improving readability and reducing log clutter for raw and transformed response data.
This commit is contained in:
vitor-aignosi
2025-09-16 10:36:43 -03:00
parent 3752170c3a
commit 39aa089384

View File

@@ -162,7 +162,7 @@ class MLFlow(BaseActivity):
)
self.debug(
f"Transform raw response data: \n {create_sample_dict(response_data)}", metadata)
f"Transform raw response data: \n {create_sample_dict(response_data, max_items=5, max_depth=5)}", metadata)
if response_data['success']:
@@ -194,7 +194,7 @@ class MLFlow(BaseActivity):
response_data['content'] = response_dataframe.to_dict()
self.debug(
f"Transform response data: \n {create_sample_dict(response_data)}", metadata)
f"Transform response data: \n {create_sample_dict(response_data, max_items=5, max_depth=5)}", metadata)
self.info("Data transformed successfully", metadata)
@@ -247,7 +247,7 @@ class MLFlow(BaseActivity):
)
self.debug(
f"Prediction response data: \n {create_sample_dict(response_data)}", metadata)
f"Prediction response data: \n {create_sample_dict(response_data, max_items=5, max_depth=5)}", metadata)
self.info("Data predicted successfully", metadata)