logit_bias field added to the completion request (sashabaranov#14)

9cat · Jan 10, 2022 · eecd212 · eecd212
1 parent 7ff9fed
commit eecd212
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/completion.go b/completion.go
@@ -25,9 +25,10 @@ type CompletionRequest struct {
 	Echo bool     `json:"echo,omitempty"`
 	Stop []string `json:"stop,omitempty"`
 
-	PresencePenalty  float32 `json:"presence_penalty,omitempty"`
-	FrequencyPenalty float32 `json:"frequency_penalty,omitempty"`
-	BestOf           int     `json:"best_of,omitempty"`
+	PresencePenalty  float32        `json:"presence_penalty,omitempty"`
+	FrequencyPenalty float32        `json:"frequency_penalty,omitempty"`
+	BestOf           int            `json:"best_of,omitempty"`
+	LogitBias        map[string]int `json:"logit_bias,omitempty"`
 }
 
 // Choice represents one of possible completions