rerank

package

v0.1.1 Latest Latest Go to latest Published: Sep 11, 2026 License: MIT Imports: 14 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/codeus-morbid/contextmaxxer

Links

Open Source Insights

Documentation ¶

Index ¶

Constants
func New(ctx context.Context, cfg Config) (retrieve.Reranker, io.Closer, error)
type Config
type OnnxReranker
- func NewOnnxReranker(_ context.Context, cfg Config) (*OnnxReranker, error)

Constants ¶

View Source

const (
	NoneName                       = "none"
	JinaRerankerV2BaseMultilingual = "jina-reranker-v2-base-multilingual"
	JinaRerankerV1TinyEN           = "jina-reranker-v1-tiny-en"
	MxbaiRerankXsmallV1            = "mxbai-rerank-xsmall-v1"
)

View Source

const JinaTinyFT = "jina-tiny-ft"

JinaTinyFT is the locally fine-tuned jina-reranker-v1-tiny-en (CORE-Bench issue->edit pairs; no download URLs — produced by the local ft pipeline).

Variables ¶

This section is empty.

Functions ¶

func New ¶

func New(ctx context.Context, cfg Config) (retrieve.Reranker, io.Closer, error)

Types ¶

type Config ¶

type Config struct {
	ModelName      string
	CacheDir       string
	BatchSize      int
	MaxTokens      int
	MaxQueryTokens int
	Log            *slog.Logger
}

type OnnxReranker ¶

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

func NewOnnxReranker ¶

func NewOnnxReranker(_ context.Context, cfg Config) (*OnnxReranker, error)

func (*OnnxReranker) Close ¶

func (r *OnnxReranker) Close() error

func (*OnnxReranker) Rerank ¶

func (r *OnnxReranker) Rerank(ctx context.Context, query string, candidates []retrieve.ScoredResult) ([]retrieve.ScoredResult, error)

func (*OnnxReranker) ScoreTexts ¶

func (r *OnnxReranker) ScoreTexts(ctx context.Context, query string, docs []string) ([]float32, error)

ScoreTexts implements retrieve.TextScorer so evidence-window selection can reuse the loaded cross-encoder on arbitrary spans of a body.

Source Files ¶

View all Source files

onnx.go

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL