cohereclient

Documentation

Index

• Variables
• type Client
  • func New(token string, baseURL string, model string, opts ...Option) (*Client, error)
  • func (c *Client) CreateGeneration(ctx context.Context, r *GenerationRequest) (*Generation, error)
  • func (c *Client) GetNumTokens(text string) int
• type Doer
• type Generation
• type GenerationRequest
• type Option
  • func WithHTTPClient(client Doer) Option

Variables

var (
	ErrEmptyResponse = errors.New("empty response")
	ErrModelNotFound = errors.New("model not found")
)

Types

type Client

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

func New

func New(token string, baseURL string, model string, opts ...Option) (*Client, error)

func (*Client) CreateGeneration

func (c *Client) CreateGeneration(ctx context.Context, r *GenerationRequest) (*Generation, error)

func (*Client) GetNumTokens

func (c *Client) GetNumTokens(text string) int

type Doer

type Doer interface {
	Do(req *http.Request) (*http.Response, error)
}

Doer performs a HTTP request.

type Generation

type Generation struct {
	Text string `json:"text"`
}

type GenerationRequest

type GenerationRequest struct {
	Prompt string `json:"prompt"`
}

type Option

type Option func(*Client) error

func WithHTTPClient

func WithHTTPClient(client Doer) Option

WithHTTPClient allows setting a custom HTTP client.