documentcompressor

Documentation

Index

• type CohereClient
• type CohereRerank
  • func NewCohereRank(client CohereClient, optFns ...func(o *CohereRerankOptions)) *CohereRerank
  • func (c *CohereRerank) Compress(ctx context.Context, docs []schema.Document, query string) ([]schema.Document, error)
• type CohereRerankOptions

Types

type CohereClient

type CohereClient interface {
	Rerank(ctx context.Context, request *cohere.RerankRequest) (*cohere.RerankResponse, error)
}

CohereClient is an interface for interacting with the Cohere API.

type CohereRerank

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

CohereRerank is a struct representing the Cohere Rerank compression functionality.

func NewCohereRank

func NewCohereRank(client CohereClient, optFns ...func(o *CohereRerankOptions)) *CohereRerank

NewCohereRank creates a new instance of CohereRerank with the provided client and options.

func (*CohereRerank) Compress

func (c *CohereRerank) Compress(ctx context.Context, docs []schema.Document, query string) ([]schema.Document, error)

Compress compresses the input documents using Cohere Rerank.

type CohereRerankOptions

type CohereRerankOptions struct {
	ModelName string
	TopN      int
}

CohereRerankOptions contains options for Cohere Rerank compression.