retrievers

package

v0.1.8 Latest Latest Go to latest Published: Jul 3, 2024 License: Apache-2.0 Imports: 9 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type BasicRetriever struct {
	TopK int
}

func (r *BasicRetriever) Retrieve(ctx context.Context, store store.Store, query string, datasetID string) ([]vs.Document, error)

type Retriever interface {
	Retrieve(ctx context.Context, store store.Store, query string, datasetID string) ([]vs.Document, error)
}

func GetDefaultRetriever() Retriever

func GetRetriever(name string) (Retriever, error)

type RoutingRetriever struct {
	Model             llm.LLMConfig
	AvailableDatasets []string
	TopK              int
}

func (r *RoutingRetriever) Retrieve(ctx context.Context, store store.Store, query string, datasetID string) ([]vs.Document, error)

type SubqueryRetriever struct {
	Model llm.LLMConfig
	Limit int
	TopK  int
}

func (s SubqueryRetriever) Retrieve(ctx context.Context, store store.Store, query string, datasetID string) ([]vs.Document, error)