zyx před 2 roky
rodič
revize
e283ea5e93

+ 0 - 1
build.sh

@@ -3,6 +3,5 @@ git reset HEAD --hard
 git pull
 pm2 stop chatgpt
 gradle build -Dorg.gradle.java.home=/opt/jdk-17.0.4.1
-
 pm2 start chatgpt
 # pm2 start 'java -jar /home/chatgpt/build/libs/chatgpt-0.0.1-SNAPSHOT.jar' --name chatgpt

+ 3 - 0
src/main/java/com/sxtvs/api/chatgpt/dto/CompletionsRequestDto.java

@@ -9,6 +9,9 @@ import java.util.stream.Collectors;
 public class CompletionsRequestDto {
     private List<Prompt> prompt;
 
+    private String userId;
+    private String userName;
+
     @Data
     public static class Prompt {
         private String text;

+ 7 - 3
src/main/java/com/sxtvs/api/chatgpt/service/ChatGptService.java

@@ -5,8 +5,8 @@ import com.sxtvs.api.chatgpt.dto.CompletionsParamsDto;
 import com.sxtvs.api.chatgpt.dto.CompletionsRequestDto;
 import com.sxtvs.api.chatgpt.dto.CompletionsResponseDto;
 import com.sxtvs.api.chatgpt.dto.GptResponse;
-import com.sxtvs.core.sls.AliyunLogger;
 import com.sxtvs.core.advice.BizException;
+import com.sxtvs.core.sls.AliyunLogger;
 import lombok.Cleanup;
 import lombok.SneakyThrows;
 import org.apache.commons.io.IOUtils;
@@ -22,7 +22,6 @@ import org.springframework.web.bind.annotation.RequestBody;
 
 import java.nio.charset.StandardCharsets;
 import java.util.concurrent.ArrayBlockingQueue;
-import java.util.stream.Collectors;
 
 @Service
 public class ChatGptService {
@@ -82,7 +81,12 @@ public class ChatGptService {
     public CompletionsResponseDto completions(@RequestBody CompletionsRequestDto dto) {
         var paramsDto = new CompletionsParamsDto(dto.toText());
         var params = objectMapper.writeValueAsString(paramsDto);
-        logger.info("key", "completions", "request", params, "dto", objectMapper.writeValueAsString(dto));
+        logger.info("key", "completions",
+                "request", params,
+                "dto", objectMapper.writeValueAsString(dto),
+                "userId", dto.getUserId(),
+                "userName", dto.getUserName()
+        );
 
         var result = request(params);