reranker

Documentation

Index

• Constants
• Variables
• func SplitHtml(text string, separators []string, chunkSize int) []string
• type ReRankInterface
  • func CreateCohereClientFromEnv() ReRankInterface
  • func NewCohereClient(token string) ReRankInterface
• type ReRankRequest
• type ReRankResult

Constants

const CHUNK_OVERLAP = 200

const COHERE_RERANK_MODEL = "rerank-english-v3.0"

Variables

var TOP_N_CHUNKS int = 8

Functions

func SplitHtml

func SplitHtml(text string, separators []string, chunkSize int) []string

Types

type ReRankInterface

type ReRankInterface interface {
	ReRank(request ReRankRequest) (*[]ReRankResult, error)
}

func CreateCohereClientFromEnv

func CreateCohereClientFromEnv() ReRankInterface

func NewCohereClient

func NewCohereClient(token string) ReRankInterface

NewCohereClient creates a new cohere client.

type ReRankRequest

type ReRankRequest struct {
	Query     string
	Documents []string
}

type ReRankResult

type ReRankResult struct {
	Index int
	Score float64
}