openai

Documentation

Index

• Variables
• 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 ClientV2
  • func NewClientV2(apiKey string, kv KVStore, logger zLogger.ZLogger) *ClientV2
  • func (c *ClientV2) CreateEmbedding(input string, model oai.EmbeddingModel) (*oai.Embedding, error)
• type EmbeddingRequest
• type EmbeddingResult
• type KVBadger
  • func NewKVBadger(db *badger.DB) *KVBadger
  • func (k *KVBadger) Get(key string) (*oai.Embedding, error)
  • func (k *KVBadger) Set(key string, value *oai.Embedding) error
• type KVGCache
  • func NewKVGCache(cache gcache.Cache) *KVGCache
  • func (k *KVGCache) Get(key string) (*oai.Embedding, error)
  • func (k *KVGCache) Set(key string, value *oai.Embedding) error
• type KVStore

Variables

var ErrNotExists = errors.New("key does not exist")

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 ClientV2

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

func NewClientV2

func NewClientV2(apiKey string, kv KVStore, logger zLogger.ZLogger) *ClientV2

func (*ClientV2) CreateEmbedding

func (c *ClientV2) 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"`
}

type KVBadger

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

func NewKVBadger

func NewKVBadger(db *badger.DB) *KVBadger

func (*KVBadger) Get

func (k *KVBadger) Get(key string) (*oai.Embedding, error)

func (*KVBadger) Set

func (k *KVBadger) Set(key string, value *oai.Embedding) error

type KVGCache

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

func NewKVGCache

func NewKVGCache(cache gcache.Cache) *KVGCache

func (*KVGCache) Get

func (k *KVGCache) Get(key string) (*oai.Embedding, error)

func (*KVGCache) Set

func (k *KVGCache) Set(key string, value *oai.Embedding) error

type KVStore

type KVStore interface {
	Get(key string) (*oai.Embedding, error)
	Set(key string, value *oai.Embedding) error
}