Soumik555 commited on
Commit
2e0bfc4
·
1 Parent(s): e326328

changed prompt

Browse files
Files changed (1) hide show
  1. python_code_executor_service.py +3 -1
python_code_executor_service.py CHANGED
@@ -185,7 +185,9 @@ class PythonExecutor:
185
  def _format_result(self, result: Any) -> str:
186
  """Format the result for display"""
187
  if isinstance(result, (pd.DataFrame, pd.Series)):
188
- return str(result)
 
 
189
  elif isinstance(result, (dict, list)):
190
  return json.dumps(result, indent=2)
191
  return str(result)
 
185
  def _format_result(self, result: Any) -> str:
186
  """Format the result for display"""
187
  if isinstance(result, (pd.DataFrame, pd.Series)):
188
+ # Convert DataFrame to a string, then to a list of dicts (handles NumPy types)
189
+ json_str = result.to_json(orient='records', date_format='iso')
190
+ json.dumps(json.loads(json_str), indent=2) # Re-parse for pretty formatting
191
  elif isinstance(result, (dict, list)):
192
  return json.dumps(result, indent=2)
193
  return str(result)