From 1f246fbfe96da0cf9e3445e47fab676f17991409 Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Mon, 17 Mar 2025 16:59:41 +0800
Subject: [PATCH] 'commit'

---
 .../Model/__pycache__/biModel.cpython-310.pyc | Bin 4908 -> 4993 bytes
 AI/Text2Sql/Model/biModel.py                  |   6 ++++++
 2 files changed, 6 insertions(+)

diff --git a/AI/Text2Sql/Model/__pycache__/biModel.cpython-310.pyc b/AI/Text2Sql/Model/__pycache__/biModel.cpython-310.pyc
index 7ba4c46c35375c4f42272ca3ddf10eb2c904d071..0d7ece5c73680e8cef12cb3cedc80bee66f0a3d4 100644
GIT binary patch
delta 310
zcmZ3Z)~L>x&&$ij00bFNuBWdP+sLQO&-h`o4ZlIXV+v~xV;Mt{QVKJaW~*h&GfQCy
z^K?@<pfo2?R6B(U%ri-00`gR#I;~S!fjn8L4yP1$AWtlX3#`&Jg$u~zPvHjh0#dkv
zJfMk1VKs~?jFJp_jBY7BwM;3zVCke3-dg4wrW)onreKB?KCoC$3LioYsIdsBksqWH
zs1L4IQ(!Z@z&=L#Qcj=`pRQf@Y~9wU6M9Q|ka^FSZ<u^kaH5I{P*8$hfSHX^fDwq9
S*cdrL7|4dxjFa<(_5lEQfKR>v

delta 207
zcmZovU!%sC&&$ij00cTuuBY!1-N>iQ&v<OI4ZlIXP6}%cV;Mt{QVKJaW~*h&Q%qq8
z^K?@<pfo2?R4j!F%ri-00`mBwYOGV(fjpqPBF7rW6h=vgJjN8RTBa0kuv*^~?po#=
zrW)onreKB?9<W$=3J*+7CYS-JF&3zi7pf7iSd(wFp};=I$$UZ+88s#^5E5f#ntWJj
F9{?_(Gr<4=

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: