fix:datasource run data error

eosphoros-ai · Apr 10, 2024 · 8fdb043 · 8fdb043
1 parent b900b1c
commit 8fdb043
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/dbgpt/app/scene/chat_dashboard/data_loader.py b/dbgpt/app/scene/chat_dashboard/data_loader.py
@@ -40,16 +40,16 @@ def get_chart_values_by_data(self, field_names, datas, chart_sql: str):
                     )
 
             for field_name in field_names[1:]:
-                # if not field_map[field_name]:
-                #     logger.info("More than 2 non-numeric column:" + field_name)
-                # else:
-                for data in datas:
-                    value_item = ValueItem(
-                        name=data[0],
-                        type=field_name,
-                        value=str(data[field_names.index(field_name)]),
-                    )
-                    values.append(value_item)
+                if not field_map[field_name]:
+                    logger.info("More than 2 non-numeric column:" + field_name)
+                else:
+                    for data in datas:
+                        value_item = ValueItem(
+                            name=data[0],
+                            type=field_name,
+                            value=str(data[field_names.index(field_name)]),
+                        )
+                        values.append(value_item)
             return field_names, values
         except Exception as e:
             logger.debug("Prepare Chart Data Failed!" + str(e))