diff --git a/completion.go b/completion.go index aba3249..4e1cac5 100644 --- a/completion.go +++ b/completion.go @@ -8,6 +8,7 @@ import ( "net/http" ) +// CompletionRequest represents a request structure for competion API type CompletionRequest struct { Prompt string `json:"prompt,omitempty"` MaxTokens int `json:"max_tokens,omitempty"` @@ -26,6 +27,7 @@ type CompletionRequest struct { FrequencyPenalty float32 `json:"frequency_penalty,omitempty"` } +// Choice represents one of possible completions type Choice struct { Text string `json:"text"` Index int `json:"index"` @@ -33,6 +35,7 @@ type Choice struct { LogProbs LogprobResult `json:"logprobs"` } +// LogprobResult represents logprob result of Choice type LogprobResult struct { Tokens []string `json:"tokens"` TokenLogprobs []float32 `json:"token_logprobs"` @@ -40,6 +43,7 @@ type LogprobResult struct { TextOffset []int `json:"text_offset"` } +// CompletionResponse represents a response structure for competion API type CompletionResponse struct { ID string `json:"id"` Object string `json:"object"` diff --git a/engines.go b/engines.go index 4af32d1..bcebe65 100644 --- a/engines.go +++ b/engines.go @@ -6,6 +6,7 @@ import ( "net/http" ) +// Engine struct represents engine from OpenAPI API type Engine struct { ID string `json:"id"` Object string `json:"object"` @@ -13,6 +14,7 @@ type Engine struct { Ready bool `json:"ready"` } +// EnginesList is a list of engines type EnginesList struct { Engines []Engine `json:"data"` } diff --git a/search.go b/search.go index 1ee4cb2..3eccc17 100644 --- a/search.go +++ b/search.go @@ -8,16 +8,19 @@ import ( "net/http" ) +// SearchRequest represents a request structure for search API type SearchRequest struct { Documents []string `json:"documents"` Query string `json:"query"` } +// SearchResult represents single result from search API type SearchResult struct { Document int `json:"document"` Score float32 `json:"score"` } +// SearchResponse represents a response structure for search API type SearchResponse struct { SearchResults []SearchResult `json:"data"` }