diff --git a/AI/Text2Sql/Model/biModel.py b/AI/Text2Sql/Model/biModel.py index fb591990..79f84abf 100644 --- a/AI/Text2Sql/Model/biModel.py +++ b/AI/Text2Sql/Model/biModel.py @@ -159,6 +159,8 @@ def generate_columns(data): # 3. 只有行政区划名的 elif "行政区划名" in columns: category_columns_str = "行政区划名" + elif "行政区划" in columns: + category_columns_str = "行政区划" # 4. 只有学段的 elif "学段" in columns: category_columns_str = "学段"