rag

package

v0.1.0 Latest Latest Go to latest Published: Feb 5, 2024 License: MIT Imports: 11 Imported by: 1

Details

This section is empty.

This section is empty.

This section is empty.

type Fusion struct {
	RAG
	// contains filtered or unexported fields
}

func NewFusion(index *index.Index, llm LLM) *Fusion

func (r *Fusion) Retrieve(ctx context.Context, query string) ([]string, error)

type LLM interface {
	Generate(context.Context, *thread.Thread) error
}

type Loader interface {
	LoadFromSource(context.Context, string) ([]document.Document, error)
}

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

func New(index *index.Index) *RAG

func (r *RAG) AddDocuments(ctx context.Context, documents ...document.Document) error

func (r *RAG) AddSources(ctx context.Context, sources ...string) error

func (r *RAG) Retrieve(ctx context.Context, query string) ([]string, error)

func (r *RAG) WithChunkOverlap(chunkOverlap uint) *RAG

func (r *RAG) WithChunkSize(chunkSize uint) *RAG

func (r *RAG) WithLoader(sourceRegexp *regexp.Regexp, loader Loader) *RAG

func (r *RAG) WithTopK(topK uint) *RAG