rag

package

v1.2.3 Latest Latest Go to latest Published: Mar 8, 2025 License: MIT Imports: 10 Imported by: 0

Details

Package rag is RAG implementations

This section is empty.

This section is empty.

This section is empty.

type Option func(*Options)

func WithChunker(chunker embedder.Chunker) Option

func WithContextGenerator(fn func(string, []vectordb.Record) string) Option

func WithEhanceQueryAgent(v agents.TypeableAgent[schema.String, schema.String]) Option

func WithEmbedder(e embedder.Embedder) Option

func WithName(name string) Option

func WithSearchOptions(opts ...vectordb.SearchOption) Option

func WithVectorDB(v vectordb.Engine) Option

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

type RAG[O schema.Schema] struct {
	Options
	// contains filtered or unexported fields
}

func NewRAG[O schema.Schema](agent agents.TypeableAgent[schema.String, O], opts ...Option) *RAG[O]

func (r *RAG[O]) AddDocuments(ctx context.Context, collectionName string, docs ...document.Document) (*components.LLMUsage, error)

func (r *RAG[O]) Name() string

func (r *RAG[O]) Run(ctx context.Context, query *schema.String, output *O, llmResp *components.LLMResponse) error

func (r *RAG[O]) RunAnonymous(ctx context.Context, query any, llmResp *components.LLMResponse) (any, error)

func (r *RAG[O]) Search(ctx context.Context, query string, opts ...vectordb.SearchOption) ([]vectordb.Record, *components.LLMUsage, error)

func (r *RAG[O]) SetSearchOptions(opts ...vectordb.SearchOption)

func (r *RAG[O]) Stream(ctx context.Context, query *schema.String) (<-chan string, agents.MergeResponse, error)

func (r *RAG[O]) StreamAnonymous(ctx context.Context, query any) (<-chan string, agents.MergeResponse, error)