zyx 2 jaren geleden
bovenliggende
commit
494196ebb6
1 gewijzigde bestanden met toevoegingen van 7 en 2 verwijderingen
  1. 7 2
      src/main/java/com/sxtvs/api/chatgpt/dto/CompletionsParamsDto.java

+ 7 - 2
src/main/java/com/sxtvs/api/chatgpt/dto/CompletionsParamsDto.java

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