retrieval

package
v0.80.6 Latest Latest
Warning

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

Go to latest
Published: Feb 1, 2026 License: MIT Imports: 10 Imported by: 0

Documentation

Index

Constants

View Source
const (
	// A value of 60 is commonly used in information retrieval.
	RRFK = 60
)

RRF constants.

Variables

This section is empty.

Functions

This section is empty.

Types

type AdaptiveRetriever

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

根据查询复杂度和结果质量动态调整检索策略.

func NewAdaptiveRetriever

func NewAdaptiveRetriever(
	st *store.Store,
	embeddingService ai.EmbeddingService,
	rerankerService ai.RerankerService,
) *AdaptiveRetriever

NewAdaptiveRetriever 创建自适应检索器.

func (*AdaptiveRetriever) Retrieve

func (r *AdaptiveRetriever) Retrieve(ctx context.Context, opts *RetrievalOptions) ([]*SearchResult, error)

Retrieve 自适应检索主入口.

type QualityLevel

type QualityLevel int

QualityLevel 结果质量等级.

const (
	LowQuality QualityLevel = iota
	MediumQuality
	HighQuality
)

func (QualityLevel) String

func (q QualityLevel) String() string

String 返回质量等级的字符串表示.

type RetrievalOptions

type RetrievalOptions struct {
	TimeRange         *queryengine.TimeRange
	Logger            *slog.Logger
	Query             string
	Strategy          string
	RequestID         string
	Limit             int
	UserID            int32
	MinScore          float32
	ScheduleQueryMode queryengine.ScheduleQueryMode
}

RetrievalOptions 检索选项.

type SearchResult

type SearchResult struct {
	Memo     *store.Memo
	Schedule *store.Schedule
	Type     string
	Content  string
	ID       int64
	Score    float32
}

SearchResult 检索结果.

Jump to

Keyboard shortcuts

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