孙永军 1 年之前
父节点
当前提交
0cfea52074

+ 1 - 2
src/main/java/com/sxtvs/open/api/chat/entity/KimiChat.java

@@ -1,6 +1,5 @@
 package com.sxtvs.open.api.chat.entity;
 
-import com.baomidou.mybatisplus.annotation.IdType;
 import com.baomidou.mybatisplus.annotation.TableId;
 import com.baomidou.mybatisplus.annotation.TableName;
 import java.io.Serializable;
@@ -28,7 +27,7 @@ public class KimiChat implements Serializable {
 
     private Long groupId;
 
-    private Integer index;
+    private Integer ind;
 
     private String role;
 

+ 1 - 15
src/main/java/com/sxtvs/open/api/chat/service/impl/KimiChatServiceImpl.java

@@ -3,40 +3,26 @@ package com.sxtvs.open.api.chat.service.impl;
 import cn.hutool.core.util.IdUtil;
 import com.alibaba.fastjson.JSON;
 import com.sxtvs.open.api.chat.dto.KimiRequest;
-import com.sxtvs.open.api.chat.dto.KimiResponse;
 import com.sxtvs.open.api.chat.dto.KimiStreamResponse;
 import com.sxtvs.open.api.chat.dto.Message;
 import com.sxtvs.open.api.chat.entity.KimiChat;
-import com.sxtvs.open.api.chat.entity.KimiChatGroup;
 import com.sxtvs.open.api.chat.mapper.KimiChatMapper;
 import com.sxtvs.open.api.chat.service.IKimiChatService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.sxtvs.open.api.review.dto.AuthReq;
 import com.sxtvs.open.api.review.service.impl.SSEService;
-import com.sxtvs.open.core.auth.AESUtil;
-import com.sxtvs.open.core.auth.HttpContextUtil;
 import jakarta.annotation.Resource;
 import lombok.extern.slf4j.Slf4j;
-import org.apache.http.util.TextUtils;
 import org.springframework.http.*;
 import org.springframework.stereotype.Service;
 import org.springframework.web.client.RestTemplate;
 
-import java.io.BufferedInputStream;
-import java.io.IOException;
-import java.io.InputStream;
 import java.net.URI;
 import java.net.http.HttpClient;
 import java.net.http.HttpRequest;
 import java.net.http.HttpResponse;
-import java.net.http.WebSocket;
 import java.time.LocalDateTime;
 import java.util.ArrayList;
-import java.util.Collections;
 import java.util.List;
-import java.util.Optional;
-import java.util.concurrent.CompletionStage;
-import java.util.concurrent.Flow;
 
 /**
  * <p>
@@ -106,7 +92,7 @@ public class KimiChatServiceImpl extends ServiceImpl<KimiChatMapper, KimiChat> i
             kimiChat = new KimiChat();
             kimiChat.setId(kimiResponse.getId());
             kimiChat.setGroupId(groupId);
-            kimiChat.setIndex(kimiResponse.getChoices().get(0).getIndex());
+            kimiChat.setInd(kimiResponse.getChoices().get(0).getIndex());
             kimiChat.setRole("system");
             kimiChat.setContent(kimiResponse.getChoices().stream().map(x -> x.getDelta().getContent( )).reduce((a, b) -> a + b).get());
             kimiChat.setCreateTime(LocalDateTime.now());

+ 1 - 1
src/main/java/com/sxtvs/open/api/review/controller/SseController.java

@@ -36,7 +36,7 @@ public class SseController {
     @Resource
     private KimiChatServiceImpl kimiChatService;
 
-    @PostMapping(path = "/chat", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    @PostMapping(path = "/chat"/*, produces = MediaType.TEXT_EVENT_STREAM_VALUE*/)
     @NoAPIResponse
     public SseEmitter handleSsePost(@RequestBody KimiChat kimiChat) throws IOException {
         SseEmitter sseEmitter = new SseEmitter(0L); // 设置超时时间