diff --git a/AI/Text2Sql/Model/__pycache__/biModel.cpython-310.pyc b/AI/Text2Sql/Model/__pycache__/biModel.cpython-310.pyc
index 7ba4c46c..0d7ece5c 100644
Binary files a/AI/Text2Sql/Model/__pycache__/biModel.cpython-310.pyc and b/AI/Text2Sql/Model/__pycache__/biModel.cpython-310.pyc differ
diff --git a/AI/Text2Sql/Model/biModel.py b/AI/Text2Sql/Model/biModel.py
index 79f84abf..bb124c3a 100644
--- a/AI/Text2Sql/Model/biModel.py
+++ b/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: