Ver código fonte

知识设置新增table切分

S0025136190 7 meses atrás
pai
commit
78cc44858c

+ 2 - 0
takai-ai/src/main/java/com/takai/ai/domain/entity/TakaiDocumentSettings.java

@@ -21,4 +21,6 @@ public class TakaiDocumentSettings extends BaseEntity {
 
     private String hidInfo;
 
+    private String setTable;
+
 }

+ 5 - 12
takai-ai/src/main/java/com/takai/ai/service/impl/TakaiAiServiceImpl.java

@@ -495,6 +495,7 @@ public class TakaiAiServiceImpl implements ITakaiAiService {
                         dSettings.setDocumentId(vo.getDocument_id());
                         dSettings.setSetSlice("0");   // 默认 按标题段落切片
                         dSettings.setSetAnalyze("1"); // 默认 图片转换成标识符
+                        dSettings.setSetTable("0");   // 默认 ttable转图片
                         takaiDocumentSettingsMapper.insertDocumentSettings(dSettings);
                     }
                 }
@@ -682,24 +683,14 @@ public class TakaiAiServiceImpl implements ITakaiAiService {
 
     @Override
     public int updateDocumentSetting(TakaiDocumentSettings documentSettings, String documentId) {
-//        if (StringUtils.isNotEmpty(documentSettings.getSetSlice()) && !"3".equals(documentSettings.getSetSlice())) {
-//            documentSettings.setDocumentId(documentId);
-//            int i = takaiDocumentSettingsMapper.updateDocumentSettings(documentSettings);
-//            return i;
-//        }
-//        List<TakaiSysOss> list = new ArrayList<>();
-//        TakaiSysOss ossVo = new TakaiSysOss();
         TakaiDocumentSettings settings = new TakaiDocumentSettings();
         TakaiDocumentSettings settingsVo = takaiDocumentSettingsMapper.selectById(documentId);
-//        settings.setDocumentId(documentId);
         settings.setKnowledgeId(settingsVo.getKnowledgeId());
         settings.setSetSlice(documentSettings.getSetSlice());
         settings.setSliceValue(documentSettings.getSliceValue());
         settings.setSetAnalyze(documentSettings.getSetAnalyze());
+        settings.setSetTable(documentSettings.getSetTable());
         TakaiDocument document = takaiDocumentMapper.selectTargetDocument(TakaiDocument.builder().documentId(documentId).build());
-//        ossVo.setUrl(document.getUrl());
-//        ossVo.setOriginalName(document.getName());
-//        list.add(ossVo);
         List<UploadDocumentParams> listParams = new ArrayList<>();
         if (document != null) {
             UploadDocumentParams params = new UploadDocumentParams();
@@ -835,7 +826,7 @@ public class TakaiAiServiceImpl implements ITakaiAiService {
     }
 
     private boolean analysisFile(List<UploadDocumentParams> result, TakaiDocumentSettings settings, String flag) {
-        TakaiKnowledge knowledge = takaiKnowledgeMapper.selectTargetKnowledge(TakaiKnowledge.builder().knowledgeId(settings.getKnowledgeId()).build());
+//        TakaiKnowledge knowledge = takaiKnowledgeMapper.selectTargetKnowledge(TakaiKnowledge.builder().knowledgeId(settings.getKnowledgeId()).build());
         String url = deepseekConfig.getBaseurl() + deepseekConfig.getUploadKnowledge();
         JSONObject json = new JSONObject();
         json.put("knowledge_id", settings.getKnowledgeId());
@@ -848,11 +839,13 @@ public class TakaiAiServiceImpl implements ITakaiAiService {
             json.put("set_slice", "0");
             json.put("slice_value", null);
             json.put("set_analyze", "1");
+            json.put("set_table", "0");
             json.put("flag", "upload");
         } else {
             json.put("set_slice", settings.getSetSlice());
             json.put("slice_value", settings.getSliceValue());
             json.put("set_analyze", settings.getSetAnalyze());
+            json.put("set_table", settings.getSetTable());
             json.put("flag", "update");
         }
 

+ 5 - 1
takai-ai/src/main/resources/mapper/takaiai/TakaiDocumentSettingsMapper.xml

@@ -12,10 +12,11 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 		<result property="setAnalyze"   column="set_analyze" />
 		<result property="hidInfo"   column="hid_info"   />
 		<result property="sliceValue"   column="slice_value" />
+		<result property="setTable"   column="set_table" />
 	</resultMap>
 
 	<sql id="Base_Column">
-		id, knowledge_id, document_id, set_slice, set_analyze, hid_info, slice_value
+		id, knowledge_id, document_id, set_slice, set_analyze, hid_info, slice_value, set_table
 	</sql>
 
 	<select id="selectById" resultMap="DocumentSettingsResult">
@@ -33,6 +34,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 		<if test="setSlice != null and setSlice != ''">set_slice,</if>
 		<if test="setAnalyze != null and setAnalyze != ''">set_analyze,</if>
 		<if test="hidInfo != null and hidInfo != ''">hid_info,</if>
+		<if test="setTable != null and setTable != ''">set_table,</if>
 		create_time
 		)values(
 		<if test="id != null  and id != ''">#{id},</if>
@@ -41,6 +43,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 		<if test="setSlice != null and setSlice != ''">#{setSlice},</if>
 		<if test="setAnalyze != null and setAnalyze != ''">#{setAnalyze},</if>
 		<if test="hidInfo != null and hidInfo != ''">#{hidInfo},</if>
+		<if test="setTable != null and setTable != ''">#{setTable},</if>
 		sysdate()
 		)
 	</insert>
@@ -54,6 +57,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 			<if test="setSlice != null and setSlice != ''">set_slice= #{setSlice},</if>
 			<if test="setAnalyze != null and setAnalyze != ''">set_analyze= #{setAnalyze},</if>
 			<if test="hidInfo != null and hidInfo != ''">hid_info= #{hidInfo},</if>
+			<if test="setTable != null and setTable != ''">set_table= #{setTable},</if>
 			slice_value= #{sliceValue},
 			update_time = sysdate()
 		</set>