增加与ai大模型对话

huimv-employment/fe-integration/src/main/java/com/huimv/employment/integration/kb/KbApiProperties.java

@@ -4,7 +4,7 @@ import org.springframework.boot.context.properties.ConfigurationProperties;
 import org.springframework.http.HttpHeaders;
 
 /**
- * 大模型网关 HTTP 对接（知识库 {@code /api/v1/kb/**}、对话 {@code /v1/models}、{@code /v1/chat/completions}），绑定 {@code fe.kb.*}。
+ * 大模型网关 HTTP 对接（智能体 {@code /api/console/chat}、RAG {@code /api/rag/**}），绑定 {@code fe.kb.*}。
  */
 @ConfigurationProperties(prefix = "fe.kb")
 public class KbApiProperties {
@@ -16,7 +16,7 @@ public class KbApiProperties {
 
     private String host = "localhost";
 
-    private int port = 9107;
+    private int port = 9188;
 
     /** 访问令牌（暂不需要；外网远程访问时再配置 fe.kb.api-key） */
     private String apiKey = "";

huimv-employment/fe-integration/src/main/java/com/huimv/employment/integration/kb/KnowledgeBaseClient.java

@@ -1,37 +1,67 @@
 package com.huimv.employment.integration.kb;
 
+import com.fasterxml.jackson.databind.JsonNode;
+import com.huimv.employment.integration.kb.rag.RagFileListQuery;
+import com.huimv.employment.integration.kb.rag.RagRetrieveRequest;
+
 import java.io.File;
 
 /**
- * 与大模型知识库后台的 HTTP 交互（上传文档、删除文档等），供各业务模块复用。
+ * RAG 知识库 HTTP 客户端（前缀 {@code /api/rag}），供各业务模块复用。
+ *
+ * @see <a href="doc/灵活用工项目后端与智能体接口文档.md">RAG 接口文档</a>
  */
 public interface KnowledgeBaseClient {
 
     /**
-     * 方式一：multipart/form-data 上传本地文件入库。
+     * {@code GET /api/rag/health} 健康检查。
+     */
+    JsonNode health();
+
+    /**
+     * {@code POST /api/rag/files} 上传文件（异步处理）。
      *
-     * @param file           本地文件
-     * @param uploadFilename multipart 中 file 部分的文件名
-     * @param knowledgeBase  知识库标识，默认 {@code default}
-     * @param category       分类
-     * @return 知识库返回的 file_id
+     * @return track_id，需后续调用 {@link #getFileStatus(String)} 轮询状态
      */
-    String uploadFile(File file, String uploadFilename, String knowledgeBase, String category);
+    String uploadFile(File file, String uploadFilename);
 
     /**
-     * 方式二：multipart/form-data 提交 url、category、description 上传文件入库。
+     * {@code PUT /api/rag/files/{file_id}} 更新文件。
      *
-     * @param fileUrl     知识库可 HTTP 拉取的正文地址
-     * @param category    分类
-     * @param description 描述
-     * @return 知识库返回的 file_id
+     * @return track_id
+     */
+    String updateFile(String fileId, File file, String uploadFilename);
+
+    /**
+     * {@code GET /api/rag/files} 分页列出文档。
+     */
+    JsonNode listFiles(RagFileListQuery query);
+
+    /**
+     * {@code GET /api/rag/files/{file_id}} 查询处理状态（track_id 或 doc_id）。
+     * <p>status：processing / processed / failed</p>
      */
-    String uploadFileByUrl(String fileUrl, String category, String description);
+    JsonNode getFileStatus(String fileId);
 
     /**
-     * 按知识库文档 id 移出条目。
+     * {@code DELETE /api/rag/files/{file_id}} 删除单个文档。
+     */
+    void deleteFile(String fileId);
+
+    /**
+     * {@code POST /api/rag/retrieve} 语义检索。
+     */
+    JsonNode retrieve(RagRetrieveRequest request);
+
+    /**
+     * {@code POST /api/rag/insert-text} 插入单条文本。
      *
-     * @param kbDocId {@code /api/v1/kb/files/{id}} 中的 id
+     * @return track_id
+     */
+    String insertText(String text, String fileSource);
+
+    /**
+     * {@code POST /api/rag/reprocess-failed} 重新处理失败文档。
      */
-    void deleteFile(String kbDocId);
+    JsonNode reprocessFailed();
 }

huimv-employment/fe-integration/src/main/java/com/huimv/employment/integration/kb/KnowledgeBaseClientImpl.java

@@ -4,6 +4,9 @@ import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.huimv.employment.common.exception.BizException;
 import com.huimv.employment.common.exception.ErrorCode;
+import com.huimv.employment.integration.kb.rag.RagFileListQuery;
+import com.huimv.employment.integration.kb.rag.RagInsertTextRequest;
+import com.huimv.employment.integration.kb.rag.RagRetrieveRequest;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Qualifier;
@@ -11,6 +14,7 @@ import org.springframework.core.io.FileSystemResource;
 import org.springframework.http.HttpEntity;
 import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpMethod;
+import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Service;
 import org.springframework.util.LinkedMultiValueMap;
@@ -19,6 +23,7 @@ import org.springframework.util.StringUtils;
 import org.springframework.web.client.HttpStatusCodeException;
 import org.springframework.web.client.ResourceAccessException;
 import org.springframework.web.client.RestTemplate;
+import org.springframework.web.util.UriComponentsBuilder;
 
 import java.io.File;
 import java.io.UnsupportedEncodingException;
@@ -28,14 +33,18 @@ import java.nio.charset.StandardCharsets;
 import java.util.function.Supplier;
 
 /**
- * 调用大模型知识库开放 API（multipart 上传 / 删除文件等）。
+ * RAG 知识库 {@code /api/rag/**} HTTP 客户端实现。
  */
 @Service
 public class KnowledgeBaseClientImpl implements KnowledgeBaseClient {
 
     private static final Logger log = LoggerFactory.getLogger(KnowledgeBaseClientImpl.class);
 
-    private static final String[] FILE_ID_KEYS = { "file_id", "fileId", "kbDocId", "kb_doc_id", "docId", "id" };
+    private static final String RAG_PREFIX = "/api/rag";
+
+    private static final String[] TRACK_ID_KEYS = {
+            "track_id", "trackId", "file_id", "fileId", "kbDocId", "kb_doc_id", "docId", "id"
+    };
 
     private final KbApiProperties properties;
     private final RestTemplate restTemplate;
@@ -50,51 +59,188 @@ public class KnowledgeBaseClientImpl implements KnowledgeBaseClient {
     }
 
     @Override
-    public String uploadFile(File file, String uploadFilename, String knowledgeBase, String category) {
+    public JsonNode health() {
+        ensureReady();
+        return getJson("/health");
+    }
+
+    @Override
+    public String uploadFile(File file, String uploadFilename) {
         ensureReady();
         if (file == null || !file.isFile()) {
             throw new BizException(ErrorCode.BAD_REQUEST, "上传文件无效或不存在");
         }
-        String kb = StringUtils.hasText(knowledgeBase) ? knowledgeBase.trim() : properties.getDefaultKnowledgeBase();
-        if (!StringUtils.hasText(kb)) {
-            kb = "default";
-        }
         String name = StringUtils.hasText(uploadFilename) ? uploadFilename.trim() : file.getName();
+        MultiValueMap<String, Object> body = new LinkedMultiValueMap<>();
+        body.add("file", multipartFileResource(file, name));
+        return parseTrackId(postMultipart("/files", body));
+    }
 
+    @Override
+    public String updateFile(String fileId, File file, String uploadFilename) {
+        ensureReady();
+        requireFileId(fileId);
+        if (file == null || !file.isFile()) {
+            throw new BizException(ErrorCode.BAD_REQUEST, "上传文件无效或不存在");
+        }
+        String name = StringUtils.hasText(uploadFilename) ? uploadFilename.trim() : file.getName();
         MultiValueMap<String, Object> body = new LinkedMultiValueMap<>();
-        body.add("knowledge_base", kb);
-        body.add("category", category);
         body.add("file", multipartFileResource(file, name));
+        return parseTrackId(putMultipart("/files/" + encodePathSegment(fileId), body));
+    }
 
-        return postMultipart(body);
+    @Override
+    public JsonNode listFiles(RagFileListQuery query) {
+        ensureReady();
+        UriComponentsBuilder builder = UriComponentsBuilder.fromHttpUrl(ragUrl("/files"));
+        if (query != null) {
+            if (query.getPage() != null) {
+                builder.queryParam("page", query.getPage());
+            }
+            if (query.getPageSize() != null) {
+                builder.queryParam("page_size", query.getPageSize());
+            }
+            if (StringUtils.hasText(query.getSortField())) {
+                builder.queryParam("sort_field", query.getSortField());
+            }
+            if (StringUtils.hasText(query.getSortDirection())) {
+                builder.queryParam("sort_direction", query.getSortDirection());
+            }
+        }
+        return getJsonUri(builder.build(true).toUri());
     }
 
     @Override
-    public String uploadFileByUrl(String fileUrl, String category, String description) {
+    public JsonNode getFileStatus(String fileId) {
         ensureReady();
-        MultiValueMap<String, Object> body = new LinkedMultiValueMap<>();
-        body.add("url", fileUrl);
-        body.add("category", category);
-        body.add("description", description);
-        return postMultipart(body);
+        requireFileId(fileId);
+        return getJson("/files/" + encodePathSegment(fileId));
     }
 
-    private String postMultipart(MultiValueMap<String, Object> body) {
-        String url = properties.baseUrl() + "/api/v1/kb/files";
-        HttpHeaders headers = new HttpHeaders();
-        properties.applyAuthHeader(headers);
-        HttpEntity<MultiValueMap<String, Object>> entity = new HttpEntity<>(body, headers);
-        String raw = executeWithRetry("知识库上传", () -> {
+    @Override
+    public void deleteFile(String fileId) {
+        ensureReady();
+        requireFileId(fileId);
+        String path = "/files/" + encodePathSegment(fileId);
+        executeWithRetry("RAG 删除文档", () -> {
             try {
-                ResponseEntity<String> resp = restTemplate.exchange(URI.create(url), HttpMethod.POST, entity, String.class);
+                HttpEntity<Void> entity = new HttpEntity<>(authHeaders());
+                ResponseEntity<String> resp = restTemplate.exchange(
+                        URI.create(ragUrl(path)), HttpMethod.DELETE, entity, String.class);
+                if (!resp.getStatusCode().is2xxSuccessful()) {
+                    throw new BizException(ErrorCode.KB_SERVICE_FAILED,
+                            "RAG 删除失败 HTTP " + resp.getStatusCode().value());
+                }
+                return null;
+            } catch (HttpStatusCodeException e) {
+                if (e.getRawStatusCode() == 404) {
+                    return null;
+                }
+                throw httpError("RAG 删除失败", e);
+            }
+        });
+    }
+
+    @Override
+    public JsonNode retrieve(RagRetrieveRequest request) {
+        ensureReady();
+        if (request == null || !StringUtils.hasText(request.getQuery())) {
+            throw new BizException(ErrorCode.BAD_REQUEST, "检索 query 不能为空");
+        }
+        return postJson("/retrieve", request);
+    }
+
+    @Override
+    public String insertText(String text, String fileSource) {
+        ensureReady();
+        if (!StringUtils.hasText(text)) {
+            throw new BizException(ErrorCode.BAD_REQUEST, "text 不能为空");
+        }
+        if (!StringUtils.hasText(fileSource)) {
+            throw new BizException(ErrorCode.BAD_REQUEST, "file_source 不能为空");
+        }
+        RagInsertTextRequest body = new RagInsertTextRequest(text.trim(), fileSource.trim());
+        return parseTrackId(postJsonRaw("/insert-text", body));
+    }
+
+    @Override
+    public JsonNode reprocessFailed() {
+        ensureReady();
+        return postJson("/reprocess-failed", new Object());
+    }
+
+    private JsonNode getJson(String path) {
+        return getJsonUri(URI.create(ragUrl(path)));
+    }
+
+    private JsonNode getJsonUri(URI uri) {
+        String raw = executeWithRetry("RAG GET " + uri.getPath(), () -> {
+            try {
+                HttpEntity<Void> entity = new HttpEntity<>(authHeaders());
+                ResponseEntity<String> resp = restTemplate.exchange(uri, HttpMethod.GET, entity, String.class);
+                return resp.getBody();
+            } catch (HttpStatusCodeException e) {
+                throw httpError("RAG 请求失败", e);
+            }
+        });
+        return parseJsonNode(raw);
+    }
+
+    private JsonNode postJson(String path, Object body) {
+        return parseJsonNode(postJsonRaw(path, body));
+    }
+
+    private String postJsonRaw(String path, Object body) {
+        return executeWithRetry("RAG POST " + path, () -> {
+            try {
+                HttpHeaders headers = authHeaders();
+                headers.setContentType(MediaType.APPLICATION_JSON);
+                HttpEntity<Object> entity = new HttpEntity<>(body, headers);
+                ResponseEntity<String> resp = restTemplate.exchange(
+                        URI.create(ragUrl(path)), HttpMethod.POST, entity, String.class);
                 return resp.getBody();
             } catch (HttpStatusCodeException e) {
-                String b = e.getResponseBodyAsString(StandardCharsets.UTF_8);
-                throw new BizException(ErrorCode.KB_SERVICE_FAILED,
-                        "知识库上传失败 HTTP " + e.getRawStatusCode() + "：" + abbreviate(b));
+                throw httpError("RAG 请求失败", e);
             }
         });
-        return parseUploadResponse(raw);
+    }
+
+    private String postMultipart(String path, MultiValueMap<String, Object> body) {
+        return executeWithRetry("RAG 上传", () -> {
+            try {
+                HttpHeaders headers = authHeaders();
+                HttpEntity<MultiValueMap<String, Object>> entity = new HttpEntity<>(body, headers);
+                ResponseEntity<String> resp = restTemplate.exchange(
+                        URI.create(ragUrl(path)), HttpMethod.POST, entity, String.class);
+                return resp.getBody();
+            } catch (HttpStatusCodeException e) {
+                throw httpError("RAG 上传失败", e);
+            }
+        });
+    }
+
+    private String putMultipart(String path, MultiValueMap<String, Object> body) {
+        return executeWithRetry("RAG 更新文件", () -> {
+            try {
+                HttpHeaders headers = authHeaders();
+                HttpEntity<MultiValueMap<String, Object>> entity = new HttpEntity<>(body, headers);
+                ResponseEntity<String> resp = restTemplate.exchange(
+                        URI.create(ragUrl(path)), HttpMethod.PUT, entity, String.class);
+                return resp.getBody();
+            } catch (HttpStatusCodeException e) {
+                throw httpError("RAG 更新失败", e);
+            }
+        });
+    }
+
+    private HttpHeaders authHeaders() {
+        HttpHeaders headers = new HttpHeaders();
+        properties.applyAuthHeader(headers);
+        return headers;
+    }
+
+    private String ragUrl(String path) {
+        return properties.baseUrl() + RAG_PREFIX + path;
     }
 
     private static FileSystemResource multipartFileResource(File file, String uploadName) {
@@ -106,33 +252,95 @@ public class KnowledgeBaseClientImpl implements KnowledgeBaseClient {
         };
     }
 
-    @Override
-    public void deleteFile(String kbDocId) {
-        ensureReady();
-        if (!StringUtils.hasText(kbDocId)) {
-            throw new BizException(ErrorCode.BAD_REQUEST, "知识库文档 id 为空");
+    private static void requireFileId(String fileId) {
+        if (!StringUtils.hasText(fileId)) {
+            throw new BizException(ErrorCode.BAD_REQUEST, "file_id 不能为空");
         }
-        String url = properties.baseUrl() + "/api/v1/kb/files/" + encodePathSegment(kbDocId);
-        HttpHeaders headers = new HttpHeaders();
-        properties.applyAuthHeader(headers);
-        HttpEntity<Void> entity = new HttpEntity<>(headers);
-        executeWithRetry("知识库删除", () -> {
-            try {
-                ResponseEntity<String> resp = restTemplate.exchange(URI.create(url), HttpMethod.DELETE, entity, String.class);
-                if (!resp.getStatusCode().is2xxSuccessful()) {
-                    throw new BizException(ErrorCode.KB_SERVICE_FAILED,
-                            "知识库删除失败 HTTP " + resp.getStatusCode().value());
+    }
+
+    private JsonNode parseJsonNode(String raw) {
+        if (!StringUtils.hasText(raw)) {
+            throw new BizException(ErrorCode.KB_SERVICE_FAILED, "RAG 返回空响应");
+        }
+        try {
+            JsonNode root = objectMapper.readTree(raw);
+            assertBusinessOk(root);
+            return root;
+        } catch (BizException ex) {
+            throw ex;
+        } catch (Exception ex) {
+            throw new BizException(ErrorCode.KB_SERVICE_FAILED, "RAG 响应非 JSON：" + abbreviate(raw));
+        }
+    }
+
+    private String parseTrackId(String raw) {
+        JsonNode root = parseJsonNode(raw);
+        String trackId = extractTrackId(root);
+        if (!StringUtils.hasText(trackId)) {
+            throw new BizException(ErrorCode.KB_SERVICE_FAILED,
+                    "RAG 成功但未解析到 track_id，响应：" + abbreviate(raw));
+        }
+        return trackId;
+    }
+
+    private void assertBusinessOk(JsonNode root) {
+        if (root == null || !root.has("code")) {
+            return;
+        }
+        JsonNode c = root.get("code");
+        boolean ok = false;
+        if (c.isNumber()) {
+            int v = c.intValue();
+            ok = (v == 0 || v == 200);
+        } else if (c.isTextual()) {
+            String s = c.asText();
+            ok = "0".equals(s) || "200".equals(s) || "success".equalsIgnoreCase(s);
+        }
+        if (!ok) {
+            String msg = root.has("message") ? root.get("message").asText(null) : null;
+            if (!StringUtils.hasText(msg) && root.has("msg")) {
+                msg = root.get("msg").asText(null);
+            }
+            if (!StringUtils.hasText(msg)) {
+                msg = "RAG 接口返回失败";
+            }
+            throw new BizException(ErrorCode.KB_SERVICE_FAILED, msg);
+        }
+    }
+
+    private static String extractTrackId(JsonNode root) {
+        JsonNode data = root.get("data");
+        if (data != null && data.isTextual()) {
+            String t = data.asText(null);
+            if (StringUtils.hasText(t)) {
+                return t;
+            }
+        }
+        if (data != null && data.isObject()) {
+            for (String key : TRACK_ID_KEYS) {
+                if (data.has(key) && !data.get(key).isNull()) {
+                    String v = data.get(key).asText(null);
+                    if (StringUtils.hasText(v)) {
+                        return v;
+                    }
                 }
-                return null;
-            } catch (HttpStatusCodeException e) {
-                if (e.getRawStatusCode() == 404) {
-                    return null;
+            }
+        }
+        for (String key : TRACK_ID_KEYS) {
+            if (root.has(key) && !root.get(key).isNull()) {
+                String v = root.get(key).asText(null);
+                if (StringUtils.hasText(v)) {
+                    return v;
                 }
-                String b = e.getResponseBodyAsString(StandardCharsets.UTF_8);
-                throw new BizException(ErrorCode.KB_SERVICE_FAILED,
-                        "知识库删除失败 HTTP " + e.getRawStatusCode() + "：" + abbreviate(b));
             }
-        });
+        }
+        return null;
+    }
+
+    private BizException httpError(String prefix, HttpStatusCodeException e) {
+        String b = e.getResponseBodyAsString(StandardCharsets.UTF_8);
+        return new BizException(ErrorCode.KB_SERVICE_FAILED,
+                prefix + " HTTP " + e.getRawStatusCode() + "：" + abbreviate(b));
     }
 
     private <T> T executeWithRetry(String operation, Supplier<T> action) {
@@ -236,79 +444,6 @@ public class KnowledgeBaseClientImpl implements KnowledgeBaseClient {
         }
     }
 
-    private String parseUploadResponse(String raw) {
-        if (!StringUtils.hasText(raw)) {
-            throw new BizException(ErrorCode.KB_SERVICE_FAILED, "知识库上传返回空响应");
-        }
-        JsonNode root;
-        try {
-            root = objectMapper.readTree(raw);
-        } catch (Exception e) {
-            throw new BizException(ErrorCode.KB_SERVICE_FAILED, "知识库上传响应非 JSON：" + abbreviate(raw));
-        }
-        assertBusinessOk(root);
-        String fileId = extractFileId(root);
-        if (!StringUtils.hasText(fileId)) {
-            throw new BizException(ErrorCode.KB_SERVICE_FAILED,
-                    "知识库上传成功但未解析到 file_id，响应：" + abbreviate(raw));
-        }
-        return fileId;
-    }
-
-    private void assertBusinessOk(JsonNode root) {
-        if (root == null || !root.has("code")) {
-            return;
-        }
-        JsonNode c = root.get("code");
-        boolean ok = false;
-        if (c.isNumber()) {
-            int v = c.intValue();
-            ok = (v == 0 || v == 200);
-        } else if (c.isTextual()) {
-            String s = c.asText();
-            ok = "0".equals(s) || "200".equals(s) || "success".equalsIgnoreCase(s);
-        }
-        if (!ok) {
-            String msg = root.has("message") ? root.get("message").asText(null) : null;
-            if (!StringUtils.hasText(msg) && root.has("msg")) {
-                msg = root.get("msg").asText(null);
-            }
-            if (!StringUtils.hasText(msg)) {
-                msg = "知识库接口返回失败";
-            }
-            throw new BizException(ErrorCode.KB_SERVICE_FAILED, msg);
-        }
-    }
-
-    private static String extractFileId(JsonNode root) {
-        JsonNode data = root.get("data");
-        if (data != null && data.isTextual()) {
-            String t = data.asText(null);
-            if (StringUtils.hasText(t)) {
-                return t;
-            }
-        }
-        if (data != null && data.isObject()) {
-            for (String key : FILE_ID_KEYS) {
-                if (data.has(key) && !data.get(key).isNull()) {
-                    String v = data.get(key).asText(null);
-                    if (StringUtils.hasText(v)) {
-                        return v;
-                    }
-                }
-            }
-        }
-        for (String key : FILE_ID_KEYS) {
-            if (root.has(key) && !root.get(key).isNull()) {
-                String v = root.get(key).asText(null);
-                if (StringUtils.hasText(v)) {
-                    return v;
-                }
-            }
-        }
-        return null;
-    }
-
     private static String abbreviate(String s) {
         if (s == null) {
             return "";

huimv-employment/fe-integration/src/main/java/com/huimv/employment/integration/kb/rag/RagFileListQuery.java

@@ -0,0 +1,54 @@
+package com.huimv.employment.integration.kb.rag;
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * {@code GET /api/rag/files} 分页查询参数。
+ */
+@JsonInclude(JsonInclude.Include.NON_NULL)
+public class RagFileListQuery {
+
+    private Integer page;
+
+    @JsonProperty("page_size")
+    private Integer pageSize;
+
+    @JsonProperty("sort_field")
+    private String sortField;
+
+    @JsonProperty("sort_direction")
+    private String sortDirection;
+
+    public Integer getPage() {
+        return page;
+    }
+
+    public void setPage(Integer page) {
+        this.page = page;
+    }
+
+    public Integer getPageSize() {
+        return pageSize;
+    }
+
+    public void setPageSize(Integer pageSize) {
+        this.pageSize = pageSize;
+    }
+
+    public String getSortField() {
+        return sortField;
+    }
+
+    public void setSortField(String sortField) {
+        this.sortField = sortField;
+    }
+
+    public String getSortDirection() {
+        return sortDirection;
+    }
+
+    public void setSortDirection(String sortDirection) {
+        this.sortDirection = sortDirection;
+    }
+}

huimv-employment/fe-integration/src/main/java/com/huimv/employment/integration/kb/rag/RagInsertTextRequest.java

@@ -0,0 +1,38 @@
+package com.huimv.employment.integration.kb.rag;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * {@code POST /api/rag/insert-text} 请求体。
+ */
+public class RagInsertTextRequest {
+
+    private String text;
+
+    @JsonProperty("file_source")
+    private String fileSource;
+
+    public RagInsertTextRequest() {
+    }
+
+    public RagInsertTextRequest(String text, String fileSource) {
+        this.text = text;
+        this.fileSource = fileSource;
+    }
+
+    public String getText() {
+        return text;
+    }
+
+    public void setText(String text) {
+        this.text = text;
+    }
+
+    public String getFileSource() {
+        return fileSource;
+    }
+
+    public void setFileSource(String fileSource) {
+        this.fileSource = fileSource;
+    }
+}

huimv-employment/fe-integration/src/main/java/com/huimv/employment/integration/kb/rag/RagRetrieveRequest.java

@@ -0,0 +1,60 @@
+package com.huimv.employment.integration.kb.rag;
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+/**
+ * {@code POST /api/rag/retrieve} 请求体。
+ */
+@JsonInclude(JsonInclude.Include.NON_NULL)
+public class RagRetrieveRequest {
+
+    private String query;
+
+    @JsonProperty("top_k")
+    private Integer topK = 5;
+
+    private String mode = "mix";
+
+    @JsonProperty("include_references")
+    private Boolean includeReferences = true;
+
+    public RagRetrieveRequest() {
+    }
+
+    public RagRetrieveRequest(String query) {
+        this.query = query;
+    }
+
+    public String getQuery() {
+        return query;
+    }
+
+    public void setQuery(String query) {
+        this.query = query;
+    }
+
+    public Integer getTopK() {
+        return topK;
+    }
+
+    public void setTopK(Integer topK) {
+        this.topK = topK;
+    }
+
+    public String getMode() {
+        return mode;
+    }
+
+    public void setMode(String mode) {
+        this.mode = mode;
+    }
+
+    public Boolean getIncludeReferences() {
+        return includeReferences;
+    }
+
+    public void setIncludeReferences(Boolean includeReferences) {
+        this.includeReferences = includeReferences;
+    }
+}