添加向量库

qianbao
1 parent 57dbbfae
Showing 2 changed files with 5 additions and 5 deletions
src/main/java/com/xly/milvus/service/impl/AiGlobalAgentQuestionSqlEmitterServiceImpl.java
src/main/java/com/xly/thread/AiUserAgentQuestionThread.java
@@ -83,11 +83,10 @@ public class AiGlobalAgentQuestionSqlEmitterServiceImpl implements AiGlobalAgent
                 .build();
         InsertResp insertResp = milvusClient.insert(insertReq);
-        if(insertResp.getInsertCnt()>0){
-            //调用数据库插入数据库
-            Map<String, Object> searMap = dynamicExeDbService.getDoProMap(sProName, data);
-            dynamicExeDbService.getCallPro(searMap, sProName);
-        }
+        //调用数据库插入数据库
+        Map<String, Object> searMap = dynamicExeDbService.getDoProMap(sProName, data);
+        dynamicExeDbService.getCallPro(searMap, sProName);
+
         System.out.println("成功插入 " + insertResp.getInsertCnt() + " 条数据");
         System.out.println("   - 数据预览:");
     }
@@ -47,6 +47,7 @@ public class AiUserAgentQuestionThread implements Runnable {
         data.put("sQuestion",SqlValidateUtil.getsQuestion(session.getSUserQuestionList()));
         data.put("sId",new UUIDGenerator().next());
         data.put("cachType",cachType);
+        data.put("userInput",sQuestion);
         //插入Redis缓存
         if(bRedis==1 && ObjectUtil.isNotEmpty(sSqlContent)){
             String sReidKey = SqlValidateUtil.getsKey( sSceneId, sMethodId, sQuestionGroupNo);