cohere

Documentation

Index

• type Config
• type Provider
  • func New(config Config) (*Provider, error)
  • func (provider *Provider) Descriptor() llmux.ProviderDescriptor
  • func (provider *Provider) EmbeddingModel(modelID string) (llmux.EmbeddingModel, error)
  • func (provider *Provider) LanguageModel(modelID string) (llmux.LanguageModel, error)
  • func (provider *Provider) ListModels(ctx context.Context) ([]llmux.ModelInfo, error)
  • func (provider *Provider) Name() string
  • func (provider *Provider) RerankingModel(modelID string) (llmux.RerankingModel, error)

Types

type Config

type Config struct {
	APIKey  string
	BaseURL string
	Headers http.Header
	Client  *http.Client
	Retry   llmux.RetryPolicy
}

type Provider

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

func New

func New(config Config) (*Provider, error)

func (*Provider) Descriptor

func (provider *Provider) Descriptor() llmux.ProviderDescriptor

func (*Provider) EmbeddingModel

func (provider *Provider) EmbeddingModel(modelID string) (llmux.EmbeddingModel, error)

func (*Provider) LanguageModel

func (provider *Provider) LanguageModel(modelID string) (llmux.LanguageModel, error)

func (*Provider) ListModels

func (provider *Provider) ListModels(ctx context.Context) ([]llmux.ModelInfo, error)

ListModels implements llmux.ModelLister using Cohere's Models API: GET /v1/models (paginated). Chat uses /v2; discovery remains on /v1.

func (*Provider) Name

func (provider *Provider) Name() string

func (*Provider) RerankingModel

func (provider *Provider) RerankingModel(modelID string) (llmux.RerankingModel, error)