From 5410948aedfa977475f2c602756d24d74036ee68 Mon Sep 17 00:00:00 2001
From: lucasjones <lucasjonesdev@hotmail.co.uk>
Date: Wed, 2 Sep 2020 10:58:40 +0100
Subject: [PATCH] Add logprob support

---
 completion.go | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/completion.go b/completion.go
index e0074a9..aba3249 100644
--- a/completion.go
+++ b/completion.go
@@ -17,19 +17,27 @@ type CompletionRequest struct {
 
 	N int `json:"n,omitempty"`
 
-	LogProbs int `json:"logobs,omitempty"`
+	LogProbs int `json:"logprobs,omitempty"`
 
-	Echo bool   `json:"echo,omitempty"`
-	Stop string `json:"stop,omitempty"`
+	Echo bool     `json:"echo,omitempty"`
+	Stop []string `json:"stop,omitempty"`
 
 	PresencePenalty  float32 `json:"presence_penalty,omitempty"`
 	FrequencyPenalty float32 `json:"frequency_penalty,omitempty"`
 }
 
 type Choice struct {
-	Text         string `json:"text"`
-	Index        int    `json:"index"`
-	FinishReason string `json:"finish_reason"`
+	Text         string        `json:"text"`
+	Index        int           `json:"index"`
+	FinishReason string        `json:"finish_reason"`
+	LogProbs     LogprobResult `json:"logprobs"`
+}
+
+type LogprobResult struct {
+	Tokens        []string             `json:"tokens"`
+	TokenLogprobs []float32            `json:"token_logprobs"`
+	TopLogprobs   []map[string]float32 `json:"top_logprobs"`
+	TextOffset    []int                `json:"text_offset"`
 }
 
 type CompletionResponse struct {