VaalaCat/go-openai
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add logprob support

Browse Source
This commit is contained in:
lucasjones
2020-09-02 10:58:40 +01:00
parent eb1cc604f2
commit 5410948aed
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
completion.go
View File

@@ -17,19 +17,27 @@ type CompletionRequest struct {
N int `json:"n,omitempty"` N int `json:"n,omitempty"`
LogProbs int `json:"logobs,omitempty"` LogProbs int `json:"logprobs,omitempty"`
Echo bool `json:"echo,omitempty"` Echo bool `json:"echo,omitempty"`
Stop string `json:"stop,omitempty"` Stop []string `json:"stop,omitempty"`
PresencePenalty float32 `json:"presence_penalty,omitempty"` PresencePenalty float32 `json:"presence_penalty,omitempty"`
FrequencyPenalty float32 `json:"frequency_penalty,omitempty"` FrequencyPenalty float32 `json:"frequency_penalty,omitempty"`
} }
type Choice struct { type Choice struct {
Text string `json:"text"` Text string `json:"text"`
Index int `json:"index"` Index int `json:"index"`
FinishReason string `json:"finish_reason"` 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 { type CompletionResponse struct {