zyx преди 2 години
родител
ревизия
1d572fe687

+ 0 - 2
src/main/java/com/sxtvs/api/chatgpt/controller/ChatGptController.java

@@ -27,8 +27,6 @@ public class ChatGptController {
     @Autowired
     private AliyunLogger logger;
 
-    private HttpClient httpClient = HttpClientBuilder.create().build();
-
     @SneakyThrows
     @RequestMapping("completions")
     public CompletionsResponseDto completions(@RequestBody CompletionsRequestDto dto) {

+ 0 - 4
src/main/java/com/sxtvs/api/chatgpt/dto/CompletionsParamsDto.java

@@ -4,8 +4,6 @@ import com.fasterxml.jackson.annotation.JsonProperty;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 
-import java.util.List;
-
 @NoArgsConstructor
 @Data
 public class CompletionsParamsDto {
@@ -27,8 +25,6 @@ public class CompletionsParamsDto {
     @JsonProperty("presence_penalty")
     private Double presencePenalty = 0.0;
 
-    @JsonProperty("stop")
-    private List<String> stop = List.of("我:", "机器人:");
 
     public CompletionsParamsDto(String prompt) {
         this.prompt = prompt;

+ 15 - 1
src/main/java/com/sxtvs/api/chatgpt/dto/CompletionsRequestDto.java

@@ -2,7 +2,21 @@ package com.sxtvs.api.chatgpt.dto;
 
 import lombok.Data;
 
+import java.util.List;
+
 @Data
 public class CompletionsRequestDto {
-    private String prompt;
+    private List<Prompt> prompt;
+
+    @Data
+    public static class Prompt {
+        private String text;
+        private ItemType type;
+
+    }
+
+    public enum ItemType {
+        user,
+        robot
+    }
 }