From eecd21296372e9fe06c8f7b470295ff8962921be Mon Sep 17 00:00:00 2001
From: Viacheslav Kovalevskyi <viacheslav@kovalevskyi.com>
Date: Sun, 9 Jan 2022 23:05:23 -0800
Subject: [PATCH] logit_bias field added to the completion request (#14)

---
 completion.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/completion.go b/completion.go
index a8a7de1..1712b17 100644
--- 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