AI 对于时间的处理

qianbao
1 parent a14f98e1
Showing 3 changed files with 69 additions and 107 deletions
src/main/java/com/xly/agent/DynamicTableNl2SqlAiAgent.java
src/main/java/com/xly/runner/AppStartupRunner.java
src/main/java/com/xly/service/XlyErpService.java
@@ -81,81 +81,59 @@ public interface DynamicTableNl2SqlAiAgent {
                             @V("sDataNow") String sDataNow,
                             @V("userInput") String userInput);
-    /**
-     * SQL错误重试引导提示词
-     * 当第一次生成的SQL执行错误时，将错误信息传入，让AI重新生成
-     */
+
     /**
      * 动态表结构：自然语言转MySQL SELECT语句
      * 入参：数据库名、表名(多表用,分隔)、表结构、用户查询
      */
     @SystemMessage("""
-            你是资深MySQL数据分析师，严格遵循以下**通用规则**生成SQL，适用于所有业务场景：
-            1. 语法规范：仅生成符合MySQL8.0.36的标准SELECT语句，兼容低版本，多表关联用JOIN而非逗号；
-            2. 输出格式：仅返回SQL语句本身，无任何解释、换行、```sql/```包裹、备注、多余空格，直接输出可执行SQL；
-            3. 编写规范：
-                3.1 多表关联必须使用 表名+字段名（如表名.字段名），严格按下面[涉及表名]中的表次序关联，聚合函数(SUM/COUNT/AVG/MIN/MAX)必须加业务化别名，日期过滤使用标准DATE格式（yyyy-MM-dd）；
-                3.2 SQL所有字段均采用 表名.字段名 方式生成，务必确保 字段名 在相应的 表名 描述的字段中存在，如果不存在重试其它方式，直到满足条件；
-                3.3 SQL所有字段涉及的所有表名，都要**严格**按下面[涉及表名]中的表次序关联，没有关联不允许使用；
-                3.4 SQL所有的查询条件，如果是字符类型的字段,均需要加不为空判断，用示例格式判断，示例:ifnull(customername,'')<>''； 
-                3.5 SQL所有的查询条件，如果是日期类型的字段,均需要加不为空判断，用示例格式判断，示例:tmakedate is not Null；
-                3.6 SQL所有的显示字段的别名中,不能出现空格，如: tCreateDate as earliest 订单日期,正确的应是 tCreateDate as earliest订单日期
-                3.7 在AVG聚合函数中禁止用LAG窗口函数
-            4. 安全约束：
-               - 禁止：DDL/DML语句（DROP/ALTER/INSERT/UPDATE/DELETE等）
-               - 禁止：存储过程、自定义函数、临时表
-               - 允许：子查询（当需要使用窗口函数LAG/ROW_NUMBER等时）
-               - 允许：CTE公用表表达式（WITH语句）
-            5. 精准性：
-                5.1 严格按用户需求+传入的表结构生成，仅使用指定字段/表，无多余字段、无无效表关联、无冗余过滤条件；
-                5.2 用户需求中没有明确的日期条件,默认为全部数据,禁止增加任何日期过滤条件
-            6. 关联规则：多表关联时，必须使用外键/业务唯一键关联，禁止无意义关联。
-            7. 当前时间：{{sDataNow}}
-            8. 时间处理规则：
-               8.1 当前系统时间：{{sDataNow}}（格式：yyyy年MM月dd日HH时mm分ss秒）
-               8.2 用户需求中的相对时间概念，必须基于{{sDataNow}}进行转换：
-                   - "本年" → 当前年份：{{sDataNow}}的年份
-                   - "本月" → 当前月份：{{sDataNow}}的年份和月份
-                   - "本季度" → 当前季度：基于{{sDataNow}}计算
-                   - "本日/今天" → {{sDataNow}}的具体日期
-                   - "昨天" → {{sDataNow}}减1天
-                   - "本周" → 基于{{sDataNow}}计算周一到周日
-                   - "近7天" → {{sDataNow}}减7天到{{sDataNow}}
-               8.3 示例转换：
-                   当前时间：2024-03-15
-                   用户说"查询本年数据" → 查询条件应为：YEAR(日期字段) = 2024
-                   用户说"查询本月数据" → 查询条件应为：YEAR(日期字段) = 2024 AND MONTH(日期字段) = 3
-               8.4 如果用户需求中没有明确的时间条件，禁止增加任何时间过滤条件    
-            """)
+        【系统角色】
+        你是资深MySQL 8.0.36数据分析师，严格杜绝以下错误，生成100%可执行的SELECT语句；
+        【生成前自检要求】
+        1. 先检查是否违反上述禁止规则，再生成SQL；
+        2. 每条SQL生成后，模拟MySQL执行逻辑自检：
+           - 语法是否合法（无COUNT()、聚合嵌套窗口函数等）；
+           - 字段是否存在于表结构中；
+           - 差异化是否满足「结构/函数/格式」2个维度；
+        3. 若自检发现错误，立即重新生成，直至所有SQL符合要求。
+        【严格禁止的错误类型（强制遵守）】
+        1. 语法错误：
+           - 禁止使用COUNT()无参数写法，必须写COUNT(*)、COUNT(1)或COUNT(具体非空字段)；
+           - 禁止在AVG/SUM等聚合函数内嵌套LAG/ROW_NUMBER等窗口函数(如 AVG(dMaterialsPrice - LAG(dMaterialsPrice, 1, dMaterialsPrice))/SUM(viw_ai_purchaseorder.dMaterialsPrice - LAG(viw_ai_purchaseorder.dMaterialsPrice, 1)))；
+           - 禁止ORDER BY中直接使用未别名的窗口函数表达式（如ORDER BY dMaterialsPrice - LAG(...)）；
+           - 禁止GROUP BY字段与SELECT非聚合字段不一致；
+        2. 规则违规：
+           - 禁止字段不写表名前缀（必须是「表名.字段名」格式）；
+           - 禁止日期字段非空判断用ifnull(日期字段,'')<>''（仅字符字段用此写法，日期字段用IS NOT NULL）；
+           - 禁止生成与{{errorSql}}/{{historySqlList}}重复的语句，禁止仅修改排序字段/别名的“伪差异化”,可以使用子查询修复或者修改查询字段，不要使用窗口函数（如 LAG）；
+           - 禁止LAG窗口函数缺失ORDER BY子句（必须按时间字段排序）；
+           - 禁止HAVING条件使用COUNT()无参数写法，禁止过滤条件与业务需求无关；
+           - 禁止在 AVG/SUM 等聚合函数中嵌套窗口函数；
+        3. 编写规范：
+            - 多表关联必须使用 表名+字段名（如表名.字段名），严格按下面[涉及表名]中的表次序关联，聚合函数(SUM/COUNT/AVG/MIN/MAX)必须加业务化别名，日期过滤使用标准DATE格式（yyyy-MM-dd）；
+            - SQL所有字段均采用 表名.字段名 方式生成，务必确保 字段名 在相应的 表名 描述的字段中存在，如果不存在重试其它方式，直到满足条件；
+            - SQL所有字段涉及的所有表名，都要**严格**按下面[涉及表名]中的表次序关联，没有关联不允许使用；
+            - SQL所有的查询条件，如果是字符类型的字段,均需要加不为空判断，用示例格式判断，示例:ifnull(customername,'')<>''； 
+            - SQL所有的查询条件，如果是日期类型的字段,均需要加不为空判断，用示例格式判断，示例:tmakedate is not Null；
+            - SQL所有的显示字段的别名中,不能出现空格，如: tCreateDate as earliest 订单日期,正确的应是 tCreateDate as earliest订单日期
+            - 在AVG聚合函数中不允许使用LAG、LEAD 等窗口函数
+            - AVG(LAG(...))这种嵌套是不允许的
+            - GROUP BY 后面不允许使用窗口函数
+       """)
     @UserMessage("""
         【业务场景表结构信息】
-            涉及表名：{{tableNames}}（多表用,分隔，需关联时请按规范使用JOIN）
-            表结构详情：{{tableStruct}}（多表请标注表名+字段，格式：表名(字段1:类型,字段2:类型,主键/外键)）
-            当前时间：{{sDataNow}}
+        涉及表名：{{tableNames}}
+        表结构详情：{{tableStruct}}
+        当前时间：{{sDataNow}}
         【原始用户需求】
         {{userInput}}
+        【错误信息】
+        之前生成的错误SQL：{{errorSql}}
+        执行错误信息：{{errorMessage}}
+        【生成要求】
+        1. 先修复错误SQL的所有问题，确保语法/逻辑合规；
+        2. 生成与{{errorSql}}/{{historySqlList}}重复的语句，禁止仅修改排序字段/别名的“伪差异化”,可以使用子查询修复或者修改查询字段，不要使用窗口函数（如 LAG）的SELECT语句；
         请根据上述表结构+通用规则，生成符合要求的MySQL SELECT语句;
-        【之前生成的错误SQL】
-        {{errorSql}}
-        【执行错误信息】
-        {{errorMessage}}
-        【生成SQL语句要求】
-            1.不能生成{{errorSql}}一样的语句
-        【错误分析指引】
-        1. 错误类型：请根据错误代码判断
-           - "Unknown column"：字段不存在，检查字段名拼写或改用表中存在的字段
-           - "Table doesn't exist"：表名错误，检查表名拼写
-           - "You have an error in your SQL syntax"：语法错误，检查关键词、括号、引号
-           - "Column not found in ON clause"：JOIN条件字段不存在
-           - "Non unique table/alias"：表别名重复
-        2. 修复建议：
-           - 如果是字段错误：查看表结构{{tableStruct}}，找到正确的字段名替换
-           - 如果是语法错误：检查SELECT、FROM、WHERE、JOIN等关键词用法
-           - 如果是类型错误：字符串加单引号，数字不加引号，日期用'yyyy-MM-dd'格式
-           - 如果是关联错误：确保所有表都通过外键正确JOIN
-           - 如果是未知错误：确保函数之间不互斥，函数能正常使用，然后重新生成新的语句
-        请根据以上信息，重新生成正确的MySQL SELECT语句。
-        只返回SQL语句本身，不要任何解释和包装。
         """)
     String regenerateSqlWithError(@MemoryId String userId,
             @V("tableNames") String tableNames,
@@ -163,7 +141,9 @@ public interface DynamicTableNl2SqlAiAgent {
             @V("sDataNow") String sDataNow,
             @V("userInput") String userInput,
             @V("errorSql") String errorSql,
-            @V("errorMessage") String errorMessage
+            @V("errorMessage") String errorMessage,
+            @V("n") String iErroCount,
+            @V("historySqlList") String historySqlList
     );
     /**
      * 动态表结构：自然语言解释SQL执行结果
@@ -179,11 +159,11 @@ public interface DynamicTableNl2SqlAiAgent {
             """)
     @UserMessage("""
             【业务场景表结构信息】
-            表结构详情：{{tableStruct}}            
+            表结构详情：{{tableStruct}}
             【查询相关信息】
             用户原始查询：{{userInput}}
             执行的MySQL SQL：{{sql}}
-            SQL执行结果（JSON格式）：{{result}}            
+            SQL执行结果（JSON格式）：{{result}}
             请根据上述信息+通用规则，对查询结果做业务解释：
             """)
     String explainSqlResult(@MemoryId String userId,
@@ -151,9 +151,9 @@ public class AppStartupRunner implements CommandLineRunner {
     /**
      * 根据编码获取AI代理
      */
-    public static SceneDto getAiAgentByCode(String sCode) {
+    public static SceneDto getAiAgentByCode(String sName) {
         return AI_AGENT_CACHE.stream()
-                .filter(agent -> agent.getSSceneNo().equals(sCode))
+                .filter(agent -> agent.getSSceneName().equals(sName))
                 .findFirst()
                 .orElse(null);
     }
@@ -4,7 +4,6 @@ import cn.hutool.core.date.DatePattern;
 import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.core.util.StrUtil;
-import cn.hutool.db.Session;
 import com.alibaba.fastjson2.JSON;
 import com.xly.agent.ChatiAgent;
 import com.xly.agent.DynamicTableNl2SqlAiAgent;
@@ -29,17 +28,9 @@ import dev.langchain4j.data.message.ChatMessageType;
 import dev.langchain4j.model.chat.ChatLanguageModel;
 import dev.langchain4j.model.ollama.OllamaChatModel;
 import dev.langchain4j.service.AiServices;
-import dev.langchain4j.service.V;
-import jnr.ffi.annotations.In;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
-import org.python.antlr.ast.Str;
 import org.springframework.stereotype.Service;
-
-import java.math.BigDecimal;
-import java.text.DateFormat;
-import java.time.LocalDate;
-import java.time.format.DateTimeFormatter;
 import java.util.*;
 @Service
@@ -123,30 +114,12 @@ public class XlyErpService {
                 // 缺失的参数明细
                 sResponMessage = session.getSFunPrompts();
             }
-//            //说明没有找到动态方法重新走一次
-//            if(maxToolRetries==1 && session.getCurrentTool()== null){
-//                Thread.sleep(1000);
-//                  doCleanUserMemory(session,userId);
-//                return erpUserInput( userInput,
-//                                            userId ,
-//                                            sUserName ,
-//                                            sBrandsId ,
-//                                            sSubsidiaryId,
-//                                            sUserType,
-//                                            authorization,
-//                                            0);
-//            }
-
             if (session.getCurrentTool()== null){
                 sResponMessageOld = sResponMessage;
                 sResponMessage = StrUtil.EMPTY;
             }
             //5.执行工具方法后，清除记忆
             if(session.getBCleanMemory()){
-//                operableChatMemoryProvider.clearSpecifiedMemory(userId);
-//                session.setCurrentTool(null);
-//                UserSceneSessionService.ERP_AGENT_CACHE.remove(userId);
-//                UserSceneSessionService.CHAT_AGENT_CACHE.remove(userId);
                 doCleanUserMemory(session,userId);
             }
 //          6.找到方法并且本方法带表结构描述时，需要调用 自然语言转SQL智能体
@@ -154,7 +127,7 @@ public class XlyErpService {
                     && ObjectUtil.isNotEmpty(session.getCurrentTool().getSInputTabelName())
                     && ObjectUtil.isNotEmpty(session.getCurrentTool().getSStructureMemo()))
             ){
-                sResponMessage = getDynamicTableSql(session, input, userId, userInput,0,StrUtil.EMPTY,StrUtil.EMPTY);
+                sResponMessage = getDynamicTableSql(session, input, userId, userInput,0,StrUtil.EMPTY,StrUtil.EMPTY,"0",StrUtil.EMPTY);
             }
             //如果返回空的进入闲聊模式
             if (ObjectUtil.isEmpty(sResponMessage)){
@@ -200,24 +173,30 @@ public class XlyErpService {
      * @return java.lang.String
      * @Description 获取执行动态SQL
      **/
-    private String getDynamicTableSql(UserSceneSession session,String input,String userId,String userInput,Integer attempt,String errorSql,String errorMessage ){
+    private String getDynamicTableSql(UserSceneSession session,String input,String userId,String userInput,Integer attempt,String errorSql,String errorMessage,String iErroCount,String historySqlList ){
         String resultExplain = "信息模糊,请提供更具体的问题或指令";
         try{
             while (attempt < maxRetries) {
                 try{
                     attempt = attempt+1;
-                    return getDynamicTableSqlExec(session, input, userId, userInput,errorSql,errorMessage);
+                    return getDynamicTableSqlExec(session, input, userId, userInput,errorSql,errorMessage,iErroCount,historySqlList);
                 }catch (Exception e){
                     String erroMsg = e.getMessage();
                     String errorSqlOld = StrUtil.EMPTY;
                     if(erroMsg.contains(EnhancedErrorGuidance.splitString) && erroMsg.split(EnhancedErrorGuidance.splitString).length>1){
                         errorSqlOld = erroMsg.split(EnhancedErrorGuidance.splitString)[1];
+                        errorSqlOld = StrUtil.replace(errorSqlOld,";","");
+                        if(StrUtil.isNotEmpty(historySqlList)){
+                            historySqlList = historySqlList+"/"+errorSqlOld;
+                        }else{
+                            historySqlList = errorSqlOld;
+                        }
                     }
-                    String errorMessageOld = EnhancedErrorGuidance.getErrorGuidance(erroMsg);
+                    String errorMessageOld = erroMsg;
                     if (attempt == maxRetries) {
-                        return resultExplain+"查询的SQL语句："+errorSqlOld;
+                        return resultExplain+"查询的SQL语句："+historySqlList;
                     } else {
-                        return getDynamicTableSql( session, input, userId, userInput, attempt,errorSqlOld,errorMessageOld);
+                        return getDynamicTableSql( session, input, userId, userInput, attempt,errorSqlOld,errorMessageOld,attempt.toString(),historySqlList);
                     }
                 }
             }
@@ -252,7 +231,7 @@ public class XlyErpService {
      * @return java.lang.String
      * @Description 执行动态sSql
      **/
-    private String getDynamicTableSqlExec(UserSceneSession session,String input,String userId,String userInput,String errorSql,String errorMessage){
+    private String getDynamicTableSqlExec(UserSceneSession session,String input,String userId,String userInput,String errorSql,String errorMessage,String iErroCount,String historySqlList){
         // 1. 构建自然语言转SQLAgent，
         DynamicTableNl2SqlAiAgent aiDynamicTableNl2SqlAiAgent = createDynamicTableNl2SqlAiAgent(userId, input, session);
         String tableNames = session.getCurrentTool().getSInputTabelName();
@@ -264,9 +243,12 @@ public class XlyErpService {
         if(ObjectUtil.isEmpty(errorSql) && ObjectUtil.isEmpty(errorMessage)){
             rawSql = aiDynamicTableNl2SqlAiAgent.generateMysqlSql(userId,tableNames,tableStruct,sDataNow,userInput);
         }else{
-            rawSql = aiDynamicTableNl2SqlAiAgent.regenerateSqlWithError(userId, tableNames,tableStruct,sDataNow,userInput,errorSql,errorMessage);
+            rawSql = aiDynamicTableNl2SqlAiAgent.regenerateSqlWithError(userId, tableNames,tableStruct,sDataNow,userInput,errorSql,errorMessage,iErroCount,historySqlList);
+        }
+        String[] rawSqlA = rawSql.split(";");
+        if(rawSqlA.length>1){
+            rawSql = rawSqlA[rawSqlA.length-1];
         }
-
         if (rawSql == null || rawSql.trim().isEmpty()) {
             throw new SqlGenerateException("SQL EMPTY");
         }
@@ -407,7 +389,7 @@ public class XlyErpService {
                 return null;
             }
             // 4. 将场景编码转换为BusinessScene枚举
-            String sSceneNo = parseResp.getSceneCode();
+            String sSceneNo = parseResp.getScene();
             return AppStartupRunner.getAiAgentByCode(sSceneNo);
         } catch (Exception e) {
             log.error("用户{}大模型意图解析失败，输入：{}", userId, userInput, e);