internal

Documentation

Index

• type Client
  • func NewClient(opts ...Option) (*Client, error)
  • func (c *Client) CreateEmbedding(ctx context.Context, input []string) (*EmbeddingResponse, error)
• type EmbeddingRequest
• type EmbeddingResponse
• type Option
  • func WithSK(sk string) Option
  • func WithUrl(url string) Option

Types

type Client

type Client struct {
	Url   string
	SK    string
	Model string
}

func NewClient

func NewClient(opts ...Option) (*Client, error)

func (*Client) CreateEmbedding

func (c *Client) CreateEmbedding(ctx context.Context, input []string) (*EmbeddingResponse, error)

type EmbeddingRequest

type EmbeddingRequest struct {
	Input []string `json:"input"`
	Model string   `json:"model"`
}

type EmbeddingResponse

type EmbeddingResponse struct {
	Data []struct {
		Embedding []float32 `json:"embedding"`
	} `json:"data"`
	Model  string `json:"model"`
	Object string `json:"object"`
	Usage  struct {
		PromptTokens int64 `json:"prompt_tokens"`
		TotalTokens  int64 `json:"total_tokens"`
	} `json:"usage"`
}

type Option

type Option func(c *Client)

func WithSK

func WithSK(sk string) Option

func WithUrl

func WithUrl(url string) Option