添加向量库

qianbao
1 parent 17a230bb
Showing 2 changed files with 4 additions and 4 deletions
src/main/java/com/xly/service/XlyErpService.java
src/main/java/com/xly/thread/AiUserAgentQuestionThread.java
@@ -377,11 +377,11 @@ public class XlyErpService {
                     //查询是否走向量库 还是数据库查询
 //                    isAggregation = aiAgent.routeQuery(session.getUserId(), input);
                     session.setDbCach("D");
-                    bHasCach = false;
+//                    bHasCach = false;
                 }else{
                     isAggregation = "MYSQL".equals(cachMap.get("cachType"));
                     session.setDbCach("H");
-                    bHasCach = true;
+//                    bHasCach = true;
                     sCleanSql = ObjectUtil.isNotEmpty(cachMap.get("sSqlContent"))?cachMap.get("sSqlContent").toString() : StrUtil.EMPTY;
                 }
 //                if(!isAggregation){
@@ -58,8 +58,8 @@ public class AiUserAgentQuestionThread implements Runnable {
         String searchText = String.format("场景：%s 方法：%s 客户问题：%s", sSceneId, sMethodId, sQuestion);
                 // sSceneId+"_"+sMethodId +"_"+sQuestion;
 //                SqlValidateUtil.getsKey( sSceneId, sMethodId, SqlValidateUtil.getsQuestion(session.getSUserQuestionList()));
-        //存入向量库 存在SQL语句并且没有where  并且执行成功
-        if(!SqlWhereUtil.hasValidConditionAfterClean(sSqlContent) && ObjectUtil.isNotEmpty(sSqlContent) && bSucess){
+        //存入向量库 存在SQL语句并且没有where  并且执行成功 并且来源不是缓存
+        if(!SqlWhereUtil.hasValidConditionAfterClean(sSqlContent) && ObjectUtil.isNotEmpty(sSqlContent) && bSucess && !"H".equals(session.getDbCach())){
             aiGlobalAgentQuestionSqlEmitterService.addAiGlobalAgentQuestionSqlEmitter(searchText,data,sQuestion,sSqlContent,cachType,"ai_global_agent_question_sql",false);
         }
         //调用数据库插入数据库