logit_bias field added to the completion request (#14)

2022-01-09 23:05:23 -08:00
parent 7ff9fedf93
commit eecd212963
1 changed files with 4 additions and 3 deletions
--- a/completion.go
+++ b/completion.go
@@ -28,6 +28,7 @@ type CompletionRequest struct {
 	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