|
|
@@ -4,6 +4,8 @@ import com.fasterxml.jackson.annotation.JsonProperty;
|
|
|
import lombok.Data;
|
|
|
import lombok.NoArgsConstructor;
|
|
|
|
|
|
+import java.util.List;
|
|
|
+
|
|
|
@NoArgsConstructor
|
|
|
@Data
|
|
|
public class CompletionsParamsDto {
|
|
|
@@ -19,6 +21,9 @@ public class CompletionsParamsDto {
|
|
|
@JsonProperty("stream")
|
|
|
private Boolean stream;
|
|
|
|
|
|
+ @JsonProperty("stop")
|
|
|
+ private List<String> stop = List.of("我:", "机器人:");
|
|
|
+
|
|
|
public CompletionsParamsDto(String prompt) {
|
|
|
this.prompt = prompt;
|
|
|
this.model = "text-davinci-003";
|
