openai

Documentation

Index

• Variables
• type Client
  • func NewClient(http *http.Client, token string) *Client
  • func (c *Client) GetCompletion(prompt string, options ...CompletionOptions) (string, error)
• type CompletionOptions
• type CompletionRequest
• type CompletionResponse

Variables

var DefaultCompletionOptions = CompletionOptions{
	Model:       "text-davinci-003",
	MaxTokens:   100,
	Temperature: 0.6,
}

Types

type Client

type Client struct {
	// contains filtered or unexported fields
}

func NewClient

func NewClient(http *http.Client, token string) *Client

func (*Client) GetCompletion

func (c *Client) GetCompletion(prompt string, options ...CompletionOptions) (string, error)

type CompletionOptions

type CompletionOptions struct {
	Model       string
	MaxTokens   int
	Temperature float64
}

type CompletionRequest

type CompletionRequest struct {
	Model       string  `json:"model"`
	Prompt      string  `json:"prompt"`
	MaxTokens   int     `json:"max_tokens"`
	Temperature float64 `json:"temperature"`
}

type CompletionResponse

type CompletionResponse struct {
	Choices []struct {
		Text string `json:"text"`
	} `json:"choices"`
}