|
@@ -17,9 +17,15 @@ public class CompletionsParamsDto {
|
|
|
@JsonProperty("max_tokens")
|
|
|
private Integer maxTokens;
|
|
|
@JsonProperty("temperature")
|
|
|
- private Integer temperature;
|
|
|
+ private Integer temperature = 0;
|
|
|
@JsonProperty("stream")
|
|
|
private Boolean stream;
|
|
|
+ @JsonProperty("top_p")
|
|
|
+ private Integer topP = 1;
|
|
|
+ @JsonProperty("frequency_penalty")
|
|
|
+ private Double frequencyPenalty = 0.0;
|
|
|
+ @JsonProperty("presence_penalty")
|
|
|
+ private Double presencePenalty = 0.0;
|
|
|
|
|
|
@JsonProperty("stop")
|
|
|
private List<String> stop = List.of("\n");
|
|
@@ -28,6 +34,5 @@ public class CompletionsParamsDto {
|
|
|
this.prompt = prompt;
|
|
|
this.model = "text-davinci-003";
|
|
|
this.maxTokens = 4000;
|
|
|
- this.temperature = 1;
|
|
|
}
|
|
|
}
|