增加与ai大模型对话

huimv-employment/fe-api/src/main/java/com/huimv/employment/controller/mp/ConversationController.java

@@ -3,22 +3,30 @@ package com.huimv.employment.controller.mp;
 import com.huimv.employment.common.exception.BizException;
 import com.huimv.employment.common.exception.ErrorCode;
 import com.huimv.employment.common.web.R;
+import com.huimv.employment.integration.kb.SseStreamSupport;
 import com.huimv.employment.security.LoginUser;
 import com.huimv.employment.security.LoginUserHolder;
+import com.huimv.employment.service.conversation.ConversationChatService;
 import com.huimv.employment.service.conversation.ConversationService;
+import com.huimv.employment.service.conversation.dto.ConversationChatRequest;
 import com.huimv.employment.service.conversation.dto.ConversationCreateRequest;
 import com.huimv.employment.service.conversation.dto.ConversationSummaryResponse;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.security.SecurityRequirement;
 import io.swagger.v3.oas.annotations.tags.Tag;
+import org.springframework.http.MediaType;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestHeader;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
 import java.util.List;
 
 /**
@@ -28,8 +36,8 @@ import java.util.List;
  * 从 JWT 解析 {@code userId}，仅返回当前用户自己的会话，不支持跨用户查询。
  * </p>
  * <p>
- * 典型用法：点击「新对话」调用 {@code POST /conversations} 获取 {@code conversationNo}，
- * 再作为 {@code session_id} 调用 {@code POST /ai/console/chat} 进行多轮对话。
+ * 推荐流程：{@code POST /conversations} 创建会话 → {@code POST /conversations/{id}/chat} 多轮对话（服务端落库）。
+ * 如需直连上游网关，仍可使用 {@code POST /ai/console/chat} 纯透传接口。
  * </p>
  */
 @RestController
@@ -39,11 +47,15 @@ import java.util.List;
 public class ConversationController {
 
     private static final String USER_TYPE_ENTERPRISE = "enterprise";
+    private static final String HEADER_AGENT_ID = "X-Agent-Id";
 
     private final ConversationService conversationService;
+    private final ConversationChatService conversationChatService;
 
-    public ConversationController(ConversationService conversationService) {
+    public ConversationController(ConversationService conversationService,
+                                  ConversationChatService conversationChatService) {
         this.conversationService = conversationService;
+        this.conversationChatService = conversationChatService;
     }
 
     /**
@@ -63,7 +75,7 @@ public class ConversationController {
 
     /**
      * 新建 AI 对话会话。
-     * <p>返回的 {@code conversationNo} 即后续 AI 对话的 {@code session_id}。</p>
+     * <p>返回的 {@code conversationNo} 即上游 AI 网关的 {@code session_id}。</p>
      */
     @PostMapping
     @Operation(summary = "新建 AI 对话会话",
@@ -74,6 +86,25 @@ public class ConversationController {
         return R.ok(conversationService.create(loginUser.getUserId(), request));
     }
 
+    /**
+     * 在已有会话内发起 AI 对话（SSE 流式，流结束后落库）。
+     */
+    @PostMapping(value = "/{id}/chat", consumes = MediaType.APPLICATION_JSON_VALUE)
+    @Operation(summary = "会话内 AI 对话（SSE 流式 + 落库）",
+            description = "需先 POST /conversations 创建会话。请求体仅需 message 字段；"
+                    + "服务端组装上游 console/chat 请求，流式开始前写入 user 消息，"
+                    + "流式结束后写入 assistant 消息。可选请求头 X-Agent-Id。")
+    public void chat(@PathVariable("id") Long conversationId,
+                     @Validated @RequestBody ConversationChatRequest request,
+                     HttpServletResponse response,
+                     @RequestHeader(value = HEADER_AGENT_ID, required = false) String agentId) throws IOException {
+        LoginUser loginUser = requireEnterprise();
+        SseStreamSupport.prepareSseResponse(response);
+        conversationChatService.chat(
+                loginUser.getUserId(), conversationId, request, agentId, response.getOutputStream());
+        response.flushBuffer();
+    }
+
     private static LoginUser requireEnterprise() {
         LoginUser loginUser = LoginUserHolder.require();
         if (!USER_TYPE_ENTERPRISE.equals(loginUser.getUserType())) {

huimv-employment/fe-dao/src/main/java/com/huimv/employment/dao/entity/FeConversationMessage.java

@@ -0,0 +1,127 @@
+package com.huimv.employment.dao.entity;
+
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+
+import java.time.LocalDateTime;
+
+/**
+ * AI 对话消息 {@code fe_conversation_message} 实体。
+ */
+@TableName("fe_conversation_message")
+public class FeConversationMessage {
+
+    @TableId(type = IdType.AUTO)
+    private Long id;
+
+    private Long conversationId;
+
+    /** user / assistant / system */
+    private String role;
+
+    private String content;
+
+    /** text / card_draft / card_cost / card_order */
+    private String contentType;
+
+    private String cardPayload;
+
+    private String agentModule;
+
+    private String nextAction;
+
+    private Long relatedDraftId;
+
+    private String tokenUsage;
+
+    private LocalDateTime createTime;
+
+    public Long getId() {
+        return id;
+    }
+
+    public void setId(Long id) {
+        this.id = id;
+    }
+
+    public Long getConversationId() {
+        return conversationId;
+    }
+
+    public void setConversationId(Long conversationId) {
+        this.conversationId = conversationId;
+    }
+
+    public String getRole() {
+        return role;
+    }
+
+    public void setRole(String role) {
+        this.role = role;
+    }
+
+    public String getContent() {
+        return content;
+    }
+
+    public void setContent(String content) {
+        this.content = content;
+    }
+
+    public String getContentType() {
+        return contentType;
+    }
+
+    public void setContentType(String contentType) {
+        this.contentType = contentType;
+    }
+
+    public String getCardPayload() {
+        return cardPayload;
+    }
+
+    public void setCardPayload(String cardPayload) {
+        this.cardPayload = cardPayload;
+    }
+
+    public String getAgentModule() {
+        return agentModule;
+    }
+
+    public void setAgentModule(String agentModule) {
+        this.agentModule = agentModule;
+    }
+
+    public String getNextAction() {
+        return nextAction;
+    }
+
+    public void setNextAction(String nextAction) {
+        this.nextAction = nextAction;
+    }
+
+    public Long getRelatedDraftId() {
+        return relatedDraftId;
+    }
+
+    public void setRelatedDraftId(Long relatedDraftId) {
+        this.relatedDraftId = relatedDraftId;
+    }
+
+    public String getTokenUsage() {
+        return tokenUsage;
+    }
+
+    public void setTokenUsage(String tokenUsage) {
+        this.tokenUsage = tokenUsage;
+    }
+
+    public LocalDateTime getCreateTime() {
+        return createTime;
+    }
+
+    public void setCreateTime(LocalDateTime createTime) {
+        this.createTime = createTime;
+    }
+}

huimv-employment/fe-dao/src/main/java/com/huimv/employment/dao/mapper/FeConversationMessageMapper.java

@@ -0,0 +1,10 @@
+package com.huimv.employment.dao.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import com.huimv.employment.dao.entity.FeConversationMessage;
+import org.apache.ibatis.annotations.Mapper;
+
+/** {@code fe_conversation_message} MyBatis-Plus Mapper。 */
+@Mapper
+public interface FeConversationMessageMapper extends BaseMapper<FeConversationMessage> {
+}

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

@@ -0,0 +1,91 @@
+package com.huimv.employment.integration.kb;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+import java.nio.charset.StandardCharsets;
+
+/**
+ * 从 {@code /api/console/chat} SSE 字节流中增量提取 assistant 文本。
+ */
+public class ConsoleChatSseAggregator {
+
+    private final ObjectMapper objectMapper;
+    private final StringBuilder lineBuffer = new StringBuilder();
+    private final StringBuilder assistantText = new StringBuilder();
+
+    public ConsoleChatSseAggregator(ObjectMapper objectMapper) {
+        this.objectMapper = objectMapper;
+    }
+
+    public void appendBytes(byte[] buffer, int offset, int length) {
+        if (buffer == null || length <= 0) {
+            return;
+        }
+        lineBuffer.append(new String(buffer, offset, length, StandardCharsets.UTF_8));
+        drainCompleteLines();
+    }
+
+    public String getAssistantText() {
+        return assistantText.toString();
+    }
+
+    private void drainCompleteLines() {
+        int newlineIndex;
+        while ((newlineIndex = indexOfLineBreak(lineBuffer)) >= 0) {
+            String line = lineBuffer.substring(0, newlineIndex).trim();
+            removeThroughLineBreak(lineBuffer, newlineIndex);
+            consumeSseLine(line);
+        }
+    }
+
+    private void consumeSseLine(String line) {
+        if (!line.startsWith("data:")) {
+            return;
+        }
+        String payload = line.substring(5).trim();
+        if (payload.isEmpty() || "[DONE]".equals(payload)) {
+            return;
+        }
+        try {
+            JsonNode data = objectMapper.readTree(payload);
+            JsonNode output = data.get("output");
+            if (output == null || !output.isArray()) {
+                return;
+            }
+            for (JsonNode item : output) {
+                if (!"assistant".equals(item.path("role").asText())) {
+                    continue;
+                }
+                JsonNode contents = item.get("content");
+                if (contents == null || !contents.isArray()) {
+                    continue;
+                }
+                for (JsonNode content : contents) {
+                    if ("text".equals(content.path("type").asText()) && content.has("text")) {
+                        assistantText.append(content.get("text").asText(""));
+                    }
+                }
+            }
+        } catch (Exception ignored) {
+            // 跳过无法解析的 SSE 片段，不影响流式转发
+        }
+    }
+
+    private static int indexOfLineBreak(StringBuilder sb) {
+        for (int i = 0; i < sb.length(); i++) {
+            if (sb.charAt(i) == '\n') {
+                return i;
+            }
+        }
+        return -1;
+    }
+
+    private static void removeThroughLineBreak(StringBuilder sb, int newlineIndex) {
+        int removeEnd = newlineIndex + 1;
+        if (removeEnd < sb.length() && sb.charAt(removeEnd) == '\r') {
+            removeEnd++;
+        }
+        sb.delete(0, removeEnd);
+    }
+}

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

@@ -1,5 +1,7 @@
 package com.huimv.employment.integration.kb;
 
+import com.fasterxml.jackson.databind.ObjectMapper;
+
 import org.springframework.http.ResponseEntity;
 
 import java.io.IOException;
@@ -13,4 +15,10 @@ public interface KbOpenAiProxyService {
     ResponseEntity<byte[]> getVersion();
 
     void streamConsoleChat(byte[] requestBody, String agentId, OutputStream output) throws IOException;
+
+    /**
+     * 流式转发 SSE 并聚合 assistant 文本（供会话落库场景使用，不影响纯透传接口）。
+     */
+    String streamConsoleChatCollecting(byte[] requestBody, String agentId, OutputStream output,
+                                       ObjectMapper objectMapper) throws IOException;
 }

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

@@ -1,5 +1,6 @@
 package com.huimv.employment.integration.kb;
 
+import com.fasterxml.jackson.databind.ObjectMapper;
 import com.huimv.employment.common.exception.BizException;
 import com.huimv.employment.common.exception.ErrorCode;
 import org.springframework.beans.factory.annotation.Qualifier;
@@ -57,6 +58,19 @@ public class KbOpenAiProxyServiceImpl implements KbOpenAiProxyService {
 
     @Override
     public void streamConsoleChat(byte[] requestBody, String agentId, OutputStream output) throws IOException {
+        streamConsoleChatInternal(requestBody, agentId, output, null);
+    }
+
+    @Override
+    public String streamConsoleChatCollecting(byte[] requestBody, String agentId, OutputStream output,
+                                              ObjectMapper objectMapper) throws IOException {
+        ConsoleChatSseAggregator aggregator = new ConsoleChatSseAggregator(objectMapper);
+        streamConsoleChatInternal(requestBody, agentId, output, aggregator);
+        return aggregator.getAssistantText();
+    }
+
+    private void streamConsoleChatInternal(byte[] requestBody, String agentId, OutputStream output,
+                                           ConsoleChatSseAggregator aggregator) throws IOException {
         ensureEnabled();
         validateRequestBody(requestBody);
         HttpURLConnection conn = null;
@@ -76,7 +90,7 @@ public class KbOpenAiProxyServiceImpl implements KbOpenAiProxyService {
                 if (status >= 400) {
                     writeUpstreamError(output, input);
                 } else {
-                    SseStreamSupport.pipeWithFlush(input, output);
+                    SseStreamSupport.pipeWithFlush(input, output, aggregator);
                 }
             }
         } finally {

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

@@ -31,6 +31,14 @@ public final class SseStreamSupport {
     }
 
     public static void pipeWithFlush(InputStream input, OutputStream output) throws IOException {
+        pipeWithFlush(input, output, null);
+    }
+
+    /**
+     * 转发 SSE 并在转发过程中收集字节（供 assistant 文本解析）。
+     */
+    public static void pipeWithFlush(InputStream input, OutputStream output, ConsoleChatSseAggregator aggregator)
+            throws IOException {
         if (input == null) {
             return;
         }
@@ -39,6 +47,9 @@ public final class SseStreamSupport {
         while ((read = input.read(buffer)) != -1) {
             output.write(buffer, 0, read);
             output.flush();
+            if (aggregator != null) {
+                aggregator.appendBytes(buffer, 0, read);
+            }
         }
     }
 

huimv-employment/fe-service/src/main/java/com/huimv/employment/service/conversation/ConversationChatService.java

@@ -0,0 +1,68 @@
+package com.huimv.employment.service.conversation;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ArrayNode;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import com.huimv.employment.dao.entity.FeConversation;
+import com.huimv.employment.integration.kb.KbOpenAiProxyService;
+import com.huimv.employment.service.conversation.dto.ConversationChatRequest;
+import org.springframework.stereotype.Service;
+
+import java.io.IOException;
+import java.io.OutputStream;
+
+/**
+ * 会话内 AI 对话：组装上游请求、流式转发并在结束后落库。
+ */
+@Service
+public class ConversationChatService {
+
+    private final ConversationService conversationService;
+    private final ConversationMessageService conversationMessageService;
+    private final KbOpenAiProxyService kbOpenAiProxyService;
+    private final ObjectMapper objectMapper;
+
+    public ConversationChatService(ConversationService conversationService,
+                                   ConversationMessageService conversationMessageService,
+                                   KbOpenAiProxyService kbOpenAiProxyService,
+                                   ObjectMapper objectMapper) {
+        this.conversationService = conversationService;
+        this.conversationMessageService = conversationMessageService;
+        this.kbOpenAiProxyService = kbOpenAiProxyService;
+        this.objectMapper = objectMapper;
+    }
+
+    /**
+     * 在指定会话内发起一轮 AI 对话，SSE 流结束后写入 user/assistant 消息。
+     */
+    public void chat(Long userId, Long conversationId, ConversationChatRequest request, String agentId,
+                     OutputStream output) throws IOException {
+        FeConversation conversation = conversationService.requireOwnedActive(userId, conversationId);
+        String userText = request.getMessage().trim();
+        byte[] upstreamBody = buildUpstreamBody(conversation, userId, userText);
+
+        conversationMessageService.saveUserMessage(conversation.getId(), userText);
+        String assistantText = kbOpenAiProxyService.streamConsoleChatCollecting(
+                upstreamBody, agentId, output, objectMapper);
+        conversationMessageService.saveAssistantMessage(conversation.getId(), assistantText);
+    }
+
+    private byte[] buildUpstreamBody(FeConversation conversation, Long userId, String userText) throws IOException {
+        ObjectNode root = objectMapper.createObjectNode();
+        ArrayNode input = objectMapper.createArrayNode();
+        ObjectNode userTurn = objectMapper.createObjectNode();
+        userTurn.put("role", "user");
+        ArrayNode content = objectMapper.createArrayNode();
+        ObjectNode textPart = objectMapper.createObjectNode();
+        textPart.put("type", "text");
+        textPart.put("text", userText);
+        content.add(textPart);
+        userTurn.set("content", content);
+        input.add(userTurn);
+        root.set("input", input);
+        root.put("session_id", conversation.getConversationNo());
+        root.put("user_id", String.valueOf(userId));
+        root.put("channel", "console");
+        return objectMapper.writeValueAsBytes(root);
+    }
+}

huimv-employment/fe-service/src/main/java/com/huimv/employment/service/conversation/ConversationMessageService.java

@@ -0,0 +1,57 @@
+package com.huimv.employment.service.conversation;
+
+import com.huimv.employment.dao.entity.FeConversationMessage;
+import com.huimv.employment.dao.mapper.FeConversationMessageMapper;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.StringUtils;
+
+import java.time.LocalDateTime;
+
+/**
+ * AI 对话消息落库：user 消息在流式开始前写入，assistant 消息在流式结束后写入。
+ */
+@Service
+public class ConversationMessageService {
+
+    private static final String ROLE_USER = "user";
+    private static final String ROLE_ASSISTANT = "assistant";
+    private static final String CONTENT_TYPE_TEXT = "text";
+
+    private final ConversationService conversationService;
+    private final FeConversationMessageMapper feConversationMessageMapper;
+
+    public ConversationMessageService(ConversationService conversationService,
+                                      FeConversationMessageMapper feConversationMessageMapper) {
+        this.conversationService = conversationService;
+        this.feConversationMessageMapper = feConversationMessageMapper;
+    }
+
+    @Transactional(rollbackFor = Exception.class)
+    public void saveUserMessage(Long conversationId, String userText) {
+        if (conversationId == null || !StringUtils.hasText(userText)) {
+            return;
+        }
+        insertMessage(conversationId, ROLE_USER, userText.trim());
+        conversationService.refreshAfterMessage(conversationId, userText);
+    }
+
+    @Transactional(rollbackFor = Exception.class)
+    public void saveAssistantMessage(Long conversationId, String assistantText) {
+        if (conversationId == null || !StringUtils.hasText(assistantText)) {
+            return;
+        }
+        insertMessage(conversationId, ROLE_ASSISTANT, assistantText.trim());
+        conversationService.refreshAfterMessage(conversationId, null);
+    }
+
+    private void insertMessage(Long conversationId, String role, String content) {
+        FeConversationMessage message = new FeConversationMessage();
+        message.setConversationId(conversationId);
+        message.setRole(role);
+        message.setContent(content);
+        message.setContentType(CONTENT_TYPE_TEXT);
+        message.setCreateTime(LocalDateTime.now());
+        feConversationMessageMapper.insert(message);
+    }
+}

huimv-employment/fe-service/src/main/java/com/huimv/employment/service/conversation/ConversationService.java

@@ -115,6 +115,45 @@ public class ConversationService {
         return toSummary(created);
     }
 
+    /**
+     * 校验会话归属与状态，供落库聊天接口使用。
+     */
+    public FeConversation requireOwnedActive(Long userId, Long conversationId) {
+        if (userId == null || conversationId == null) {
+            throw new BizException(ErrorCode.NOT_FOUND, "会话不存在");
+        }
+        FeConversation conversation = feConversationMapper.selectById(conversationId);
+        if (conversation == null || !userId.equals(conversation.getUserId())) {
+            throw new BizException(ErrorCode.NOT_FOUND, "会话不存在");
+        }
+        if (!STATUS_ACTIVE.equals(conversation.getStatus())) {
+            throw new BizException(ErrorCode.BAD_REQUEST, "会话已归档，无法继续对话");
+        }
+        return conversation;
+    }
+
+    /**
+     * 消息入库后刷新会话 {@code last_message_at}；首条 user 消息可自动生成标题。
+     */
+    @Transactional(rollbackFor = Exception.class)
+    public void refreshAfterMessage(Long conversationId, String userTextForTitle) {
+        if (conversationId == null) {
+            return;
+        }
+        LocalDateTime now = LocalDateTime.now();
+        FeConversation update = new FeConversation();
+        update.setId(conversationId);
+        update.setLastMessageAt(now);
+        update.setUpdateTime(now);
+        if (StringUtils.hasText(userTextForTitle)) {
+            FeConversation current = feConversationMapper.selectById(conversationId);
+            if (current != null && !StringUtils.hasText(current.getTitle())) {
+                update.setTitle(buildTitleFromUserText(userTextForTitle));
+            }
+        }
+        feConversationMapper.updateById(update);
+    }
+
     /**
      * 用户发起 AI 对话时刷新会话记录。
      * <p>
@@ -244,4 +283,12 @@ public class ConversationService {
         }
         return sessionId.substring(0, CONVERSATION_NO_MAX_LEN);
     }
+
+    private static String buildTitleFromUserText(String userText) {
+        if (!StringUtils.hasText(userText)) {
+            return null;
+        }
+        String trimmed = userText.trim();
+        return trimmed.length() <= 200 ? trimmed : trimmed.substring(0, 200);
+    }
 }

huimv-employment/fe-service/src/main/java/com/huimv/employment/service/conversation/dto/ConversationChatRequest.java

@@ -0,0 +1,26 @@
+package com.huimv.employment.service.conversation.dto;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+
+import javax.validation.constraints.NotBlank;
+import javax.validation.constraints.Size;
+
+/**
+ * 会话内 AI 对话请求（封装上游 console/chat 所需用户输入）。
+ */
+@Schema(description = "会话内 AI 对话请求")
+public class ConversationChatRequest {
+
+    @Schema(description = "用户本轮输入文本", example = "帮我招 10 个仓库临时工", requiredMode = Schema.RequiredMode.REQUIRED)
+    @NotBlank(message = "消息内容不能为空")
+    @Size(max = 8000, message = "消息内容不能超过 8000 字")
+    private String message;
+
+    public String getMessage() {
+        return message;
+    }
+
+    public void setMessage(String message) {
+        this.message = message;
+    }
+}