From 79fa05f6b47054cb934c716776a391ad2b2af3e9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=B7=E9=9B=A8?= Date: Fri, 26 Sep 2025 15:29:49 +0800 Subject: [PATCH 1/2] =?UTF-8?q?feat:sql=E6=94=B9=E6=88=90m-chema=E6=A0=BC?= =?UTF-8?q?=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- util/load_ddl_doc.py | 2 +- util/train_ddl.py | 865 +++++++++++++++++++++++++++++++++++++------ 2 files changed, 746 insertions(+), 121 deletions(-) diff --git a/util/load_ddl_doc.py b/util/load_ddl_doc.py index 1ac1fed..b11ed78 100644 --- a/util/load_ddl_doc.py +++ b/util/load_ddl_doc.py @@ -2,7 +2,7 @@ from service.cus_vanna_srevice import CustomVanna from util import train_ddl table_ddls = [ - train_ddl.ddl_sql, + train_ddl.ddl_sql,train_ddl.attendance_ddl ] list_documentions = [ train_ddl.train_document, diff --git a/util/train_ddl.py b/util/train_ddl.py index 23adb04..d9c50fd 100644 --- a/util/train_ddl.py +++ b/util/train_ddl.py @@ -1,124 +1,749 @@ -ddl_sql = """ - create table YJOA_APPSERVICE_DB."t_pr3rl2oj_yj_person_database" - ( - "person_status" VARCHAR(108) default 'NULL' comment '人员状态', - "gender" VARCHAR(108) default 'NULL' comment '性别(1:男,2:女)', - "id_card" VARCHAR(600) default 'NULL' comment '身份证号码', - "input_time" DATETIME(39) default NULL comment '录入时间', - "external_unit" VARCHAR(600) default 'NULL' comment '外部单位', - "photo" TEXT(max) default NULL comment '照片(二进制文本)', - "input_user" VARCHAR(108) default 'NULL' comment '录入人', - "is_internal" VARCHAR(108) default 'NULL' comment '是否内部员工(1代表是，0代表否)', - "phone_number" VARCHAR(600) default 'NULL' comment '手机号', - "external_dept" VARCHAR(600) default 'NULL' comment '福建亿榕信息技术有限公司', - "pass_type" VARCHAR(108) default 'NULL' comment '通行证类型', - "expire_time" DATETIME(39) default NULL comment '失效时间', - "internal_dept" VARCHAR(108) default 'NULL' comment '内部部门', - "input_dept" VARCHAR(108) default 'NULL' comment '录入部门', - "name" VARCHAR(600) default 'NULL' comment '姓名', - "internal_unit" VARCHAR(108) default 'NULL' comment '内部单位', - "person_type" VARCHAR(108) default 'NULL' comment '人员类型', - "id" VARCHAR(22) not null comment 'id', - "pubts" DATETIME(39) default CURRENT_TIMESTAMP, - "dr" INT default 0 comment '逻辑删除，1代表已删除', - "code" VARCHAR(600) default 'NULL' comment '编码', - "ytenant_id" VARCHAR(64) not null - constraint T_PR3RL2OJ_YJ_PERSON_DATABASE_I_I_YTENANT_ID - unique comment '租户ID', - "sourcegrand_id" VARCHAR(108) default 'NULL' comment '来源表id', - "first_id" VARCHAR(108) default 'NULL' comment '来源单据主表id', - "firstchild_id" VARCHAR(108) default 'NULL' comment '来源单据子表id', - "firstbusiobj" VARCHAR(108) default 'NULL' comment '来源业务对象', - "firstcode" VARCHAR(600) default 'NULL' comment '来源单据号', - "verifystate" INT default NULL comment '单据状态', - "auditor" VARCHAR(180) default 'NULL' comment '终审审批人', - "audit_time" DATETIME(39) default NULL comment '审批日期', - "auditnote" VARCHAR(600) default 'NULL' comment '当前审批人', - "procinst_id" VARCHAR(108) default 'NULL' comment '流程实例ID', - "bizflow_id" VARCHAR(108) default 'NULL' comment '业务流id', - "bizflowname" VARCHAR(600) default 'NULL' comment '流程名称', - "source_id" VARCHAR(108) default 'NULL' comment '上游单据主表id', - "sourcechild_id" VARCHAR(108) default 'NULL' comment '上游单据子表id', - "bizflowinstance_id" VARCHAR(108) default 'NULL' comment '业务流实例id', - "sourcebusiobj" VARCHAR(108) default 'NULL' comment '上游业务对象', - "sourcecode" VARCHAR(600) default 'NULL' comment '上游单据号', - "bizflow_makebillcode" VARCHAR(600) default 'NULL' comment '单据转换规则编码', - "create_time" DATETIME(39) default NULL comment '创建时间', - "modify_time" DATETIME(39) default NULL comment '修改时间', - "creator" VARCHAR(180) default 'NULL' comment '创建人', - "modifier" VARCHAR(180) default 'NULL' comment '修改人', - "worker_id" VARCHAR(200) default 'NULL' comment '工号', - "to_dept" VARCHAR(600) default 'NULL' comment '所属处室', - "id_card_type" VARCHAR(108) default 'NULL' comment '身份证件类型', - "native_place" VARCHAR(600) default 'NULL' comment '籍贯', - "nation" VARCHAR(600) default 'NULL' comment '民族', - "country" VARCHAR(600) default 'NULL' comment '国籍', - "residence_address" VARCHAR(600) default 'NULL' comment '户籍地址', - "highest_education" VARCHAR(108) default 'NULL' comment '最高学历', - "highest_degree" VARCHAR(108) default 'NULL' comment '最高学位', - "graduate_school" VARCHAR(600) default 'NULL' comment '毕业院校', - "birthday" VARCHAR(30) default 'NULL' comment '出生日期', - "work_date" VARCHAR(30) default 'NULL' comment '参加工作日期', - "political_status" VARCHAR(600) default 'NULL' comment '政治面貌', - "work_unit" VARCHAR(600) default 'NULL' comment '工作单位全称', - "post" VARCHAR(600) default 'NULL' comment '职务', - "engage_post" VARCHAR(600) default 'NULL' comment '现从事岗位', - "work_content" TEXT(max) default NULL comment '工作内容', - "engage_contract_no" VARCHAR(600) default 'NULL' comment '从事项目合同编号', - "engage_contract_name" VARCHAR(600) default 'NULL' comment '从事项目合同名称', - "is_subcontractor" VARCHAR(108) default 'NULL' comment '是否分包商', - "general_contractor_unit" VARCHAR(600) default 'NULL' comment '总包单位全称', - "entry_date" VARCHAR(30) default 'NULL' comment '入场日期', - "expected_departure_date" VARCHAR(30) default 'NULL' comment '预计离场日期', - "office_city" VARCHAR(600) default 'NULL' comment '办公城市', - "office_address" VARCHAR(600) default 'NULL' comment '办公地点', - "is_sign_confidentiality_agreement" VARCHAR(108) default 'NULL' comment '是否签署保密协议(1:是,0:否)', - "email" VARCHAR(600) default 'NULL' comment '电子邮箱', - DHDATASTA INT default 0 comment '推送状态(0新增，1更新)', - "auditor1" VARCHAR(36) comment '处室负责人' - ) - """ +ddl_sql = ''' +{ + "table_name": "t_pr3rl2oj_yj_person_database", + "table_comment": "人员信息库", + "database": "YJOA_APPSERVICE_DB", + "columns": [ + { + "name": "id", + "type": "VARCHAR(22)", + "comment": "id", + "role": "dimension", + "tags": ["主键", "ID"], + "nullable": false, + "example": "10001" + }, + { + "name": "ytenant_id", + "type": "VARCHAR(64)", + "comment": "租户ID", + "role": "dimension", + "tags": ["租户", "唯一"], + "nullable": false, + "example": "tenant_001" + }, + { + "name": "name", + "type": "VARCHAR(600)", + "comment": "姓名", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "张三" + }, + { + "name": "gender", + "type": "VARCHAR(108)", + "comment": "性别(1:男,2:女)", + "role": "dimension", + "tags": ["人员基础信息", "枚举"], + "values": { + "1": "男", + "2": "女" + }, + "example": "1" + }, + { + "name": "id_card", + "type": "VARCHAR(600)", + "comment": "身份证号码", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "110101199001011234" + }, + { + "name": "phone_number", + "type": "VARCHAR(600)", + "comment": "手机号", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "13800138000" + }, + { + "name": "email", + "type": "VARCHAR(600)", + "comment": "电子邮箱", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "zhangsan@example.com" + }, + { + "name": "birthday", + "type": "VARCHAR(30)", + "comment": "出生日期", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "1990-01-01" + }, + { + "name": "native_place", + "type": "VARCHAR(600)", + "comment": "籍贯", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "北京市" + }, + { + "name": "nation", + "type": "VARCHAR(600)", + "comment": "民族", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "汉族" + }, + { + "name": "country", + "type": "VARCHAR(600)", + "comment": "国籍", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "中国" + }, + { + "name": "residence_address", + "type": "VARCHAR(600)", + "comment": "户籍地址", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "北京市朝阳区" + }, + { + "name": "highest_education", + "type": "VARCHAR(108)", + "comment": "最高学历", + "role": "dimension", + "tags": ["人员基础信息", "枚举"], + "values": { + "1": "初中", + "2": "高中", + "3": "中专", + "4": "技校", + "5": "职高", + "6": "大专", + "7": "本科", + "8": "硕士", + "9": "博士" + }, + "example": "7" + }, + { + "name": "highest_degree", + "type": "VARCHAR(108)", + "comment": "最高学位", + "role": "dimension", + "tags": ["人员基础信息", "枚举"], + "values": { + "1": "学士学位", + "2": "硕士学位", + "3": "博士学位", + "4": "无" + }, + "example": "1" + }, + { + "name": "graduate_school", + "type": "VARCHAR(600)", + "comment": "毕业院校", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "北京大学" + }, + { + "name": "political_status", + "type": "VARCHAR(600)", + "comment": "政治面貌", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "中共党员" + }, + { + "name": "work_date", + "type": "VARCHAR(30)", + "comment": "参加工作日期", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "2012-07-01" + }, + { + "name": "worker_id", + "type": "VARCHAR(200)", + "comment": "工号", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "G001" + }, + { + "name": "is_internal", + "type": "VARCHAR(108)", + "comment": "是否内部员工(1代表是，0代表否)", + "role": "dimension", + "tags": ["人员基础信息", "枚举"], + "values": { + "1": "是", + "0": "否" + }, + "example": "1" + }, + { + "name": "person_type", + "type": "VARCHAR(108)", + "comment": "人员类型", + "role": "dimension", + "tags": ["人员基础信息", "枚举"], + "values": { + "YG": "正式员工", + "PQ": "劳务派遣人员", + "QT": "其他柔性引进人员", + "WHZ": "合作单位", + "WLS": "临时访客", + "WQT": "其他外部人员" + }, + "example": "YG" + }, + { + "name": "person_status", + "type": "VARCHAR(108)", + "comment": "人员状态", + "role": "dimension", + "tags": ["人员基础信息", "枚举"], + "values": { + "1": "草稿", + "2": "审批中", + "3": "制卡中", + "4": "已入库", + "5": "停用" + }, + "example": "4" + }, + { + "name": "internal_unit", + "type": "VARCHAR(108)", + "comment": "内部单位", + "role": "dimension", + "tags": ["组织信息"], + "example": "北京总部" + }, + { + "name": "internal_dept", + "type": "VARCHAR(108)", + "comment": "内部部门", + "role": "dimension", + "tags": ["组织信息"], + "example": "研发部" + }, + { + "name": "to_dept", + "type": "VARCHAR(600)", + "comment": "所属处室", + "role": "dimension", + "tags": ["组织信息"], + "example": "技术处" + }, + { + "name": "external_unit", + "type": "VARCHAR(600)", + "comment": "外部单位", + "role": "dimension", + "tags": ["组织信息"], + "example": "外包公司A" + }, + { + "name": "external_dept", + "type": "VARCHAR(600)", + "comment": "外部部门", + "role": "dimension", + "tags": ["组织信息"], + "example": "福建亿榕信息技术有限公司" + }, + { + "name": "post", + "type": "VARCHAR(600)", + "comment": "职务", + "role": "dimension", + "tags": ["工作信息"], + "example": "项目经理" + }, + { + "name": "engage_post", + "type": "VARCHAR(600)", + "comment": "现从事岗位", + "role": "dimension", + "tags": ["工作信息"], + "example": "开发工程师" + }, + { + "name": "work_content", + "type": "TEXT(max)", + "comment": "工作内容", + "role": "dimension", + "tags": ["工作信息"], + "example": "负责系统架构设计与开发" + }, + { + "name": "work_unit", + "type": "VARCHAR(600)", + "comment": "工作单位全称", + "role": "dimension", + "tags": ["工作信息"], + "example": "福建亿榕信息技术有限公司" + }, + { + "name": "entry_date", + "type": "VARCHAR(30)", + "comment": "入场日期", + "role": "dimension", + "tags": ["工作信息"], + "example": "2023-01-15" + }, + { + "name": "expected_departure_date", + "type": "VARCHAR(30)", + "comment": "预计离场日期", + "role": "dimension", + "tags": ["工作信息"], + "example": "2025-01-15" + }, + { + "name": "office_city", + "type": "VARCHAR(600)", + "comment": "办公城市", + "role": "dimension", + "tags": ["工作信息"], + "example": "北京" + }, + { + "name": "office_address", + "type": "VARCHAR(600)", + "comment": "办公地点", + "role": "dimension", + "tags": ["工作信息"], + "example": "北京市海淀区中关村" + }, + { + "name": "is_sign_confidentiality_agreement", + "type": "VARCHAR(108)", + "comment": "是否签署保密协议(1:是,0:否)", + "role": "dimension", + "tags": ["工作信息", "枚举"], + "values": { + "1": "是", + "0": "否" + }, + "example": "1" + }, + { + "name": "is_subcontractor", + "type": "VARCHAR(108)", + "comment": "是否分包商", + "role": "dimension", + "tags": ["工作信息", "枚举"], + "values": { + "1": "是", + "0": "否" + }, + "example": "0" + }, + { + "name": "general_contractor_unit", + "type": "VARCHAR(600)", + "comment": "总包单位全称", + "role": "dimension", + "tags": ["工作信息"], + "example": "中建集团" + }, + { + "name": "engage_contract_no", + "type": "VARCHAR(600)", + "comment": "从事项目合同编号", + "role": "dimension", + "tags": ["工作信息"], + "example": "CON2023001" + }, + { + "name": "engage_contract_name", + "type": "VARCHAR(600)", + "comment": "从事项目合同名称", + "role": "dimension", + "tags": ["工作信息"], + "example": "智慧城市建设项目" + }, + { + "name": "pass_type", + "type": "VARCHAR(108)", + "comment": "通行证类型", + "role": "dimension", + "tags": ["工作信息", "枚举"], + "values": { + "1": "集团公司员工", + "2": "借调人员", + "3": "借用人员", + "4": "外部监管人员", + "5": "外协服务人员", + "6": "工勤人员", + "7": "来访人员" + }, + "example": "1" + }, + { + "name": "expire_time", + "type": "DATETIME(39)", + "comment": "失效时间", + "role": "dimension", + "tags": ["工作信息"], + "example": "2025-01-01 00:00:00" + }, + { + "name": "input_time", + "type": "DATETIME(39)", + "comment": "录入时间", + "role": "dimension", + "tags": ["系统信息"], + "example": "2023-01-01 12:00:00" + }, + { + "name": "input_user", + "type": "VARCHAR(108)", + "comment": "录入人", + "role": "dimension", + "tags": ["系统信息"], + "example": "admin" + }, + { + "name": "input_dept", + "type": "VARCHAR(108)", + "comment": "录入部门", + "role": "dimension", + "tags": ["系统信息"], + "example": "人事部" + }, + { + "name": "create_time", + "type": "DATETIME(39)", + "comment": "创建时间", + "role": "dimension", + "tags": ["系统信息"], + "example": "2023-01-01 12:00:00" + }, + { + "name": "modify_time", + "type": "DATETIME(39)", + "comment": "修改时间", + "role": "dimension", + "tags": ["系统信息"], + "example": "2023-01-02 12:00:00" + }, + { + "name": "creator", + "type": "VARCHAR(180)", + "comment": "创建人", + "role": "dimension", + "tags": ["系统信息"], + "example": "admin" + }, + { + "name": "modifier", + "type": "VARCHAR(180)", + "comment": "修改人", + "role": "dimension", + "tags": ["系统信息"], + "example": "editor" + }, + { + "name": "pubts", + "type": "DATETIME(39)", + "comment": "pubts", + "role": "dimension", + "tags": ["系统信息"], + "example": "2023-01-01 12:00:00" + }, + { + "name": "dr", + "type": "INT", + "comment": "逻辑删除，1代表已删除", + "role": "dimension", + "tags": ["系统信息", "枚举"], + "values": { + "0": "未删除", + "1": "已删除" + }, + "example": "0" + }, + { + "name": "DHDATASTA", + "type": "INT", + "comment": "推送状态(0新增，1更新)", + "role": "dimension", + "tags": ["系统信息", "枚举"], + "values": { + "0": "新增", + "1": "更新" + }, + "example": "0" + }, + { + "name": "code", + "type": "VARCHAR(600)", + "comment": "编码", + "role": "dimension", + "tags": ["系统信息"], + "example": "CODE001" + }, + { + "name": "photo", + "type": "TEXT(max)", + "comment": "照片(二进制文本)", + "role": "dimension", + "tags": ["人员基础信息"], + "example": "[binary data]" + }, + { + "name": "id_card_type", + "type": "VARCHAR(108)", + "comment": "身份证件类型", + "role": "dimension", + "tags": ["人员基础信息", "枚举"], + "values": { + "1": "身份证", + "2": "护照", + "3": "港澳通行证" + }, + "example": "1" + }, + { + "name": "auditor", + "type": "VARCHAR(180)", + "comment": "终审审批人", + "role": "dimension", + "tags": ["审批信息"], + "example": "李四" + }, + { + "name": "audit_time", + "type": "DATETIME(39)", + "comment": "审批日期", + "role": "dimension", + "tags": ["审批信息"], + "example": "2023-01-03 10:00:00" + }, + { + "name": "auditnote", + "type": "VARCHAR(600)", + "comment": "当前审批人", + "role": "dimension", + "tags": ["审批信息"], + "example": "王五" + }, + { + "name": "verifystate", + "type": "INT", + "comment": "单据状态", + "role": "dimension", + "tags": ["审批信息"], + "example": "1" + }, + { + "name": "procinst_id", + "type": "VARCHAR(108)", + "comment": "流程实例ID", + "role": "dimension", + "tags": ["审批信息"], + "example": "PROC001" + }, + { + "name": "bizflow_id", + "type": "VARCHAR(108)", + "comment": "业务流id", + "role": "dimension", + "tags": ["审批信息"], + "example": "FLOW001" + }, + { + "name": "bizflowname", + "type": "VARCHAR(600)", + "comment": "流程名称", + "role": "dimension", + "tags": ["审批信息"], + "example": "入职审批" + }, + { + "name": "source_id", + "type": "VARCHAR(108)", + "comment": "上游单据主表id", + "role": "dimension", + "tags": ["来源信息"], + "example": "SOURCE001" + }, + { + "name": "sourcechild_id", + "type": "VARCHAR(108)", + "comment": "上游单据子表id", + "role": "dimension", + "tags": ["来源信息"], + "example": "CHILD001" + }, + { + "name": "sourcebusiobj", + "type": "VARCHAR(108)", + "comment": "上游业务对象", + "role": "dimension", + "tags": ["来源信息"], + "example": "HR系统" + }, + { + "name": "sourcecode", + "type": "VARCHAR(600)", + "comment": "上游单据号", + "role": "dimension", + "tags": ["来源信息"], + "example": "HR2023001" + }, + { + "name": "bizflowinstance_id", + "type": "VARCHAR(108)", + "comment": "业务流实例id", + "role": "dimension", + "tags": ["来源信息"], + "example": "BIZINST001" + }, + { + "name": "bizflow_makebillcode", + "type": "VARCHAR(600)", + "comment": "单据转换规则编码", + "role": "dimension", + "tags": ["来源信息"], + "example": "RULE001" + }, + { + "name": "sourcegrand_id", + "type": "VARCHAR(108)", + "comment": "来源表id", + "role": "dimension", + "tags": ["来源信息"], + "example": "GRAND001" + }, + { + "name": "first_id", + "type": "VARCHAR(108)", + "comment": "来源单据主表id", + "role": "dimension", + "tags": ["来源信息"], + "example": "FIRST001" + }, + { + "name": "firstchild_id", + "type": "VARCHAR(108)", + "comment": "来源单据子表id", + "role": "dimension", + "tags": ["来源信息"], + "example": "FIRSTCHILD001" + }, + { + "name": "firstbusiobj", + "type": "VARCHAR(108)", + "comment": "来源业务对象", + "role": "dimension", + "tags": ["来源信息"], + "example": "ERP系统" + }, + { + "name": "firstcode", + "type": "VARCHAR(600)", + "comment": "来源单据号", + "role": "dimension", + "tags": ["来源信息"], + "example": "ERP2023001" + }, + { + "name": "auditor1", + "type": "VARCHAR(36)", + "comment": "处室负责人", + "role": "dimension", + "tags": ["审批信息"], + "example": "赵六" + } + ], + "relationships": [] + +} + + +''' train_document=''' 语法为达梦数据库语法; - 查询地址，籍贯，公司，单位时，尽量使用like查询; - 查询人员信息时,由于数据表字段过多。只需要展示人员关键信息字段，id，工号，姓名，单位以及用户问题中需要查询的字段; - 表字段信息以及字段枚举信息: - - YJOA_APPSERVICE_DB."t_pr3rl2oj_yj_person_database"(人员信息表); - - person_status: 人员状态 (1:草稿,2:审批中,3:制卡中,4:已入库,5:停用); - pass_type: 通行证类型 ( - 1：集团公司员工, - 2：借调人员, - 3：借用人员, - 4：外部监管人员, - 5：外协服务人员, - 6：工勤人员, - 7：来访人员 - ); - person_type: 人员类型 ( - YG：正式员工, - PQ：劳务派遣人员, - QT：其他柔性引进人员, - WHZ：合作单位, - WLS：临时访客, - WQT：其他外部人员 - ); - id_card_type: 身份证件类型 (1:身份证,2:护照,3:港澳通行证); - highest_education: 最高学历 ( - 1：初中 - 2：高中 - 3：中专 - 4：技校 - 5：职高 - 6：大专 - 7：本科 - 8：硕士 - 9：博士 - ); - highest_degree: 最高学位 (1：学士学位,2：硕士学位,3：博士学位,4：无); - -
- ''' \ No newline at end of file + 查询地址，籍贯，公司，单位,区域时，尽量使用like查询; + 查询人员信息时,由于数据表字段过多。只需要展示人员关键信息字段:工号,姓名,单位,职务以及用户问题中需要查询的字段; + 注意返回字段有枚举字段时，注意字段的转化。 + ''' + + +attendance_ddl=''' +{ + "table_name": "t_yj_person_attendance_rules", + "table_comment": "考勤规则表", + "database": "YJOA_APPSERVICE_DB", + "columns": [ + { + "name": "id", + "type": "VARCHAR(50)", + "comment": "主键ID", + "role": "dimension", + "tags": ["主键", "ID"], + "nullable": false + }, + { + "name": "morning_check_time", + "type": "VARCHAR(50)", + "comment": "早上考勤打卡时间", + "role": "dimension", + "tags": ["时间配置", "考勤"], + "example": "08:30" + }, + { + "name": "afternoon_check_time", + "type": "VARCHAR(50)", + "comment": "下午考勤打卡时间", + "role": "dimension", + "tags": ["时间配置", "考勤"], + "example": "18:00" + }, + { + "name": "before_lunch_time", + "type": "VARCHAR(50)", + "comment": "午休前时间", + "role": "dimension", + "tags": ["时间配置", "午休"], + "example": "12:00" + }, + { + "name": "after_lunch_time", + "type": "VARCHAR(50)", + "comment": "午休后时间", + "role": "dimension", + "tags": ["时间配置", "午休"], + "example": "13:30" + }, + { + "name": "region", + "type": "VARCHAR(50)", + "comment": "考勤地点", + "role": "dimension", + "tags": ["地点", "考勤", "枚举"], + "values": { + "1": "北京", + "2": "成都", + "3": "秭归", + "4": "林芝市区", + "5": "拉萨", + "6": "米林", + "7": "派镇", + "8": "墨脱" + }, + "example": "1" + } + ], + "relationships": [] + +} + +''' \ No newline at end of file From 92d71063d943e489d86ea12f83532ae6e0eb8a96 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=B7=E9=9B=A8?= Date: Fri, 26 Sep 2025 16:05:07 +0800 Subject: [PATCH 2/2] =?UTF-8?q?feat:=E4=BB=8E=E5=90=91=E9=87=8F=E6=95=B0?= =?UTF-8?q?=E6=8D=AE=E5=BA=93=E7=A7=BB=E9=99=A4=E6=96=87=E6=A1=A3=E8=AF=B4?= =?UTF-8?q?=E6=98=8E=EF=BC=8C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- service/cus_vanna_srevice.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/service/cus_vanna_srevice.py b/service/cus_vanna_srevice.py index 94c4e92..986ede0 100644 --- a/service/cus_vanna_srevice.py +++ b/service/cus_vanna_srevice.py @@ -15,7 +15,7 @@ import json from template.template import get_base_template from datetime import datetime import logging - +from util import train_ddl logger = logging.getLogger(__name__) class OpenAICompatibleLLM(VannaBase): @@ -187,13 +187,13 @@ class OpenAICompatibleLLM(VannaBase): logger.info("Start to generate_sql_2 in cus_vanna_srevice") question_sql_list = self.get_similar_question_sql(question, **kwargs) ddl_list = self.get_related_ddl(question, **kwargs) - doc_list = self.get_related_documentation(question, **kwargs) + #doc_list = self.get_related_documentation(question, **kwargs) template = get_base_template() sql_temp = template['template']['sql'] char_temp = template['template']['chart'] # --------基于提示词，生成sql以及图表类型 sys_temp = sql_temp['system'].format(engine=config("DB_ENGINE", default='mysql'), lang='中文', - schema=ddl_list, documentation=doc_list, + schema=ddl_list, documentation=[train_ddl.train_document], data_training=question_sql_list) logger.info(f"sys_temp:{sys_temp}") user_temp = sql_temp['user'].format(question=question,