Spaces:
Running
Running
Pulastya B commited on
Commit ·
16616fb
1
Parent(s): ae7f05f
Fixed Agent Reasoning and Semanttic Layer Integration issues
Browse files- src/orchestrator.py +4 -4
- src/utils/semantic_layer.py +3 -3
src/orchestrator.py
CHANGED
|
@@ -3311,8 +3311,8 @@ You receive quality reports from EDA agent and deliver clean data to modeling ag
|
|
| 3311 |
'task_description': task_description,
|
| 3312 |
'target_col': target_col
|
| 3313 |
},
|
| 3314 |
-
|
| 3315 |
-
|
| 3316 |
)
|
| 3317 |
|
| 3318 |
# Track in workflow
|
|
@@ -4023,8 +4023,8 @@ You receive quality reports from EDA agent and deliver clean data to modeling ag
|
|
| 4023 |
'task_description': task_description,
|
| 4024 |
'target_col': target_col
|
| 4025 |
},
|
| 4026 |
-
|
| 4027 |
-
|
| 4028 |
)
|
| 4029 |
|
| 4030 |
# Check for errors and display them prominently
|
|
|
|
| 3311 |
'task_description': task_description,
|
| 3312 |
'target_col': target_col
|
| 3313 |
},
|
| 3314 |
+
last_tool=tool_name,
|
| 3315 |
+
iteration=iteration
|
| 3316 |
)
|
| 3317 |
|
| 3318 |
# Track in workflow
|
|
|
|
| 4023 |
'task_description': task_description,
|
| 4024 |
'target_col': target_col
|
| 4025 |
},
|
| 4026 |
+
last_tool=tool_name,
|
| 4027 |
+
iteration=iteration
|
| 4028 |
)
|
| 4029 |
|
| 4030 |
# Check for errors and display them prominently
|
src/utils/semantic_layer.py
CHANGED
|
@@ -109,11 +109,11 @@ class SemanticLayer:
|
|
| 109 |
|
| 110 |
# Add statistics
|
| 111 |
if stats:
|
| 112 |
-
if 'mean' in stats:
|
| 113 |
description_parts.append(f"Mean: {stats['mean']:.2f}")
|
| 114 |
-
if 'unique_count' in stats:
|
| 115 |
description_parts.append(f"Unique values: {stats['unique_count']}")
|
| 116 |
-
if 'null_percentage' in stats:
|
| 117 |
description_parts.append(f"Missing: {stats['null_percentage']:.1f}%")
|
| 118 |
|
| 119 |
# Combine into single text
|
|
|
|
| 109 |
|
| 110 |
# Add statistics
|
| 111 |
if stats:
|
| 112 |
+
if 'mean' in stats and stats['mean'] is not None:
|
| 113 |
description_parts.append(f"Mean: {stats['mean']:.2f}")
|
| 114 |
+
if 'unique_count' in stats and stats['unique_count'] is not None:
|
| 115 |
description_parts.append(f"Unique values: {stats['unique_count']}")
|
| 116 |
+
if 'null_percentage' in stats and stats['null_percentage'] is not None:
|
| 117 |
description_parts.append(f"Missing: {stats['null_percentage']:.1f}%")
|
| 118 |
|
| 119 |
# Combine into single text
|