'commit'

AI/Text2Sql/Model/biModel.py

@@ -164,6 +164,10 @@ def generate_columns(data):
     # 4. 只有学段的
     elif "学段" in columns:
         category_columns_str = "学段"
+    elif "学段名" in columns:
+        category_columns_str = "学段名"
+    elif "学段名称" in columns:
+        category_columns_str = "学段名称"
     # 5. 上面都不是的，用第一个字段记录
     else:
         category_columns_str = columns[0] if columns else ""
@@ -171,6 +175,8 @@ def generate_columns(data):
     # 6. Y轴一般的常见词是：课程数量，数量，如果都没有，用最后一个字段记录
     if "课程数量" in columns:
         value_column_str = "课程数量"
+    elif "课程数" in columns:
+        value_column_str = "课程数"
     elif "数量" in columns:
         value_column_str = "数量"
     else: