Pulastya B commited on
Commit
16616fb
·
1 Parent(s): ae7f05f

Fixed Agent Reasoning and Semanttic Layer Integration issues

Browse files
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
- tool_name=tool_name,
3315
- iteration_count=iteration
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
- tool_name=tool_name,
4027
- iteration_count=iteration
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