Add docs

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"`

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"`
 }

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"`
 }