|
@@ -2,7 +2,6 @@ package com.storlead.knowledge.api;
|
|
|
|
|
|
|
|
import com.alibaba.fastjson.JSONArray;
|
|
import com.alibaba.fastjson.JSONArray;
|
|
|
import com.alibaba.fastjson.JSONObject;
|
|
import com.alibaba.fastjson.JSONObject;
|
|
|
-import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
import com.fasterxml.jackson.core.JsonProcessingException;
|
|
import com.fasterxml.jackson.core.JsonProcessingException;
|
|
|
import com.fasterxml.jackson.core.type.TypeReference;
|
|
import com.fasterxml.jackson.core.type.TypeReference;
|
|
@@ -14,28 +13,20 @@ import com.storlead.knowledge.pojo.dto.KnowledgeDTO;
|
|
|
import com.storlead.knowledge.utils.HttpService;
|
|
import com.storlead.knowledge.utils.HttpService;
|
|
|
import com.storlead.knowledge.utils.JacksonHolder;
|
|
import com.storlead.knowledge.utils.JacksonHolder;
|
|
|
import com.storlead.knowledge.pojo.dto.QueryPageDTO;
|
|
import com.storlead.knowledge.pojo.dto.QueryPageDTO;
|
|
|
-import com.storlead.knowledge.pojo.dto.AiTrainingDTO;
|
|
|
|
|
-import com.storlead.knowledge.pojo.dto.AiDatasetsFileDTO;
|
|
|
|
|
import com.storlead.knowledge.pojo.vo.AiTrainingVO;
|
|
import com.storlead.knowledge.pojo.vo.AiTrainingVO;
|
|
|
-import com.storlead.knowledge.pojo.vo.AiDatasetsFileVO;
|
|
|
|
|
import com.storlead.knowledge.pojo.vo.KnowledgeStatisticsVO;
|
|
import com.storlead.knowledge.pojo.vo.KnowledgeStatisticsVO;
|
|
|
import com.storlead.knowledge.service.AiTrainingService;
|
|
import com.storlead.knowledge.service.AiTrainingService;
|
|
|
import com.storlead.knowledge.service.AiDatasetsFileService;
|
|
import com.storlead.knowledge.service.AiDatasetsFileService;
|
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
|
|
import io.swagger.annotations.Api;
|
|
import io.swagger.annotations.Api;
|
|
|
import io.swagger.annotations.ApiOperation;
|
|
import io.swagger.annotations.ApiOperation;
|
|
|
-import org.springframework.beans.factory.annotation.Autowired;
|
|
|
|
|
-import org.springframework.data.domain.PageImpl;
|
|
|
|
|
-import org.springframework.data.domain.PageRequest;
|
|
|
|
|
import org.springframework.web.bind.annotation.*;
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
import javax.annotation.Resource;
|
|
|
-import javax.validation.Valid;
|
|
|
|
|
import java.time.*;
|
|
import java.time.*;
|
|
|
import java.time.temporal.ChronoUnit;
|
|
import java.time.temporal.ChronoUnit;
|
|
|
import java.util.ArrayList;
|
|
import java.util.ArrayList;
|
|
|
-import java.util.Date;
|
|
|
|
|
import java.util.List;
|
|
import java.util.List;
|
|
|
import java.util.Map;
|
|
import java.util.Map;
|
|
|
|
|
|
|
@@ -197,6 +188,7 @@ public class KnowledgeController {
|
|
|
|
|
|
|
|
@PatchMapping("datasets/{dataset_id}")
|
|
@PatchMapping("datasets/{dataset_id}")
|
|
|
@ApiOperation("更新指定ID的知识库")
|
|
@ApiOperation("更新指定ID的知识库")
|
|
|
|
|
+
|
|
|
public Result<Object> updateDataset(@RequestBody KnowledgeDTO knowledgeDTO, @PathVariable String dataset_id) throws JsonProcessingException {
|
|
public Result<Object> updateDataset(@RequestBody KnowledgeDTO knowledgeDTO, @PathVariable String dataset_id) throws JsonProcessingException {
|
|
|
String url = difyProperties.getBaseUrl() + "datasets/"+dataset_id;
|
|
String url = difyProperties.getBaseUrl() + "datasets/"+dataset_id;
|
|
|
String body = JacksonHolder.OBJECT_MAPPER.writeValueAsString(knowledgeDTO);
|
|
String body = JacksonHolder.OBJECT_MAPPER.writeValueAsString(knowledgeDTO);
|