调用python redis切片数据

takai-admin/src/main/java/com/takai/web/controller/takaiai/TakaiAiController.java

@@ -535,6 +535,12 @@ public class TakaiAiController extends BaseController {
                 .collect(Collectors.toList());
         return getDataTableResultTotal(sliceList, sortedChunkIds.size());
     }
+
+    @GetMapping("/slice/search/{id}")
+    public AjaxResult searchSlice(@PathVariable String id) {
+        return success(takaiAisService.searchSlice(id));
+    }
+
     @GetMapping("/test")
     public AjaxResult test() {
         TakaiMediaReplacement vo = takaiAisService.getTakaiMediaReplacement();

takai-admin/src/main/resources/application.yml

@@ -18,7 +18,7 @@ takai:
 # 开发环境配置
 server:
   # 服务器的HTTP端口，默认为8080
-  port: 8091
+  port: 8401
   servlet:
     # 应用的访问路径
     context-path: /
@@ -54,7 +54,7 @@ spring:
     # 国际化资源文件路径
     basename: i18n/messages
   profiles:
-    active: prd
+    active: dev
   # 文件上传
   servlet:
     multipart:
@@ -164,8 +164,8 @@ bigmodel:
   prompt: 你是总结和提问大师。 """应用的名称：{{应用名称}} 这是用户的历史对话记录：{{历史对话}}""" 你只根据用户的对话记录，推演出用户接下来可能提出的问题，不要杜撰问题。可以参考应用的名称和应用的简介。 你必须遵守以下要求：1. 不要输出用户问过的问题；2. 你需要输出3个问题供用户选择。3. 你只需要输出问题，不需要解释，不需要提问。4. 你的问题可以是空的，但你不能杜撰问题。5. 问题需要站在使用这个应用的人的视角提出，因此你要注意提问的语气和人称代词。不要用您这个字。你一定要按照以下格式输出：{"问题":["xxx","xxx","xxx"]}
 deepseek:
   #base url
-  baseurl: http://10.1.27.4:18079
-#  baseurl: http://xia0miduo.gicp.net:6001
+#  baseurl: http://10.1.27.4:18079
+  baseurl: http://xia0miduo.gicp.net:6001
 #  baseurl: http://192.168.3.209:18079
   #创建知识库
   createKnowledge: /rag/create_collection
@@ -189,5 +189,7 @@ deepseek:
   deleteDoc: /rag/delete_doc
   #新增切片
   addSlice: /rag/insert_slice
+  #获取排序切片列表
+  searchSlice: /rag/slice/search
   #提示词
   prompt: 你是总结和提问大师。你只根据用户的对话记录，推演出用户接下来可能提出的问题，不要杜撰问题。可以参考应用的名称和应用的简介。 你必须遵守以下要求：1. 不要输出用户问过的问题；2. 你需要输出3个问题供用户选择。3. 你只需要输出问题，不需要解释，不需要提问。4. 你的问题可以是空的，但你不能杜撰问题。5. 问题需要站在使用这个应用的人的视角提出，因此你要注意提问的语气和人称代词。不要用您这个字。你一定要按照以下格式输出：{"问题":["xxx","xxx","xxx"]}

takai-ai/src/main/java/com/takai/ai/service/ITakaiAiService.java

@@ -94,4 +94,6 @@ public interface ITakaiAiService
 
     List<SliceInfo> slicePageInfoList(List<String> list);
 
+    JSONObject searchSlice(String id);
+
 }

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

@@ -1008,6 +1008,27 @@ public class TakaiAiServiceImpl implements ITakaiAiService {
         return takaiSliceInfoMapper.slicePageInfoList(list);
     }
 
+    @Override
+    public JSONObject searchSlice(String id) {
+        String url = deepseekConfig.getBaseurl() + deepseekConfig.getSearchSlice() + "/" + id;
+        Request request = buildGetRequest(url);
+        OkHttpClient client = buildOkHttpClient();
+        try {
+            Response response = client.newCall(request).execute();
+            if (response.isSuccessful()) {
+                String body = response.body().string();
+                JSONObject obj = JSON.parseObject(body);
+                Integer code = obj.getInteger("code");
+                if (code == 200) {
+                    return obj;
+                }
+            }
+        }catch (Exception e) {
+            logger.error("切片得分调用python接口失败", e.getMessage());
+        }
+        return null;
+    }
+
     private JSONObject analysisFile(List<UploadDocumentParams> result, TakaiDocumentSettings settings, String flag) {
         String url = deepseekConfig.getBaseurl() + deepseekConfig.getUploadKnowledge();
         JSONObject json = new JSONObject();

takai-common/src/main/java/com/takai/common/config/DeepseekConfig.java

@@ -34,6 +34,8 @@ public class DeepseekConfig {
 
     private String addSlice;
 
+    private String searchSlice;
+
     public String getBaseurl() {
         return baseurl;
     }
@@ -137,4 +139,12 @@ public class DeepseekConfig {
     public void setAddSlice(String addSlice) {
         this.addSlice = addSlice;
     }
+
+    public String getSearchSlice() {
+        return searchSlice;
+    }
+
+    public void setSearchSlice(String searchSlice) {
+        this.searchSlice = searchSlice;
+    }
 }