openai

Documentation

Index

• type Client
  • func NewClient(apiKey string, db *badger.DB, logger *logging.Logger) *Client
  • func (c *Client) CreateEmbedding(input string, model oai.EmbeddingModel) (*oai.Embedding, error)
• type EmbeddingRequest
• type EmbeddingResult

Types

type Client

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

func NewClient

func NewClient(apiKey string, db *badger.DB, logger *logging.Logger) *Client

func (*Client) CreateEmbedding

func (c *Client) CreateEmbedding(input string, model oai.EmbeddingModel) (*oai.Embedding, error)

type EmbeddingRequest

type EmbeddingRequest struct {
	Input          string `json:"input"`
	Model          string `json:"model"`
	EncodingFormat string `json:"encoding_format,omitempty"`
	User           string `json:"user,omitempty"`
}

type EmbeddingResult

type EmbeddingResult struct {
	Vector []float64 `json:"embedding"`
	Index  int       `json:"index"`
	Object string    `json:"object"`
}