reranking

package
v0.0.1 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Mar 16, 2026 License: Apache-2.0 Imports: 1 Imported by: 3

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type Model

type Model interface {
	// Rerank scores pre-rendered document texts based on their relevance to the query.
	// Returns a slice of scores with the same length as prompts.
	// Higher scores indicate higher relevance.
	Rerank(ctx context.Context, query string, prompts []string) ([]float32, error)

	// Close releases any resources held by the model (sessions, connections, etc.)
	Close() error
}

Model represents a reranking model that can score text relevance. This interface allows different model implementations (ONNX, API-based, etc.) to be used interchangeably in the reranking pipeline.

Jump to

Keyboard shortcuts

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