transformers

Documentation

Index

• Variables
• func DefaultDocumentTransformers(filetype string) (transformers []types.DocumentTransformer)
• func GetTransformer(name string) (types.DocumentTransformer, error)
• type ExtraMetadata
  • func (e *ExtraMetadata) Transform(_ context.Context, docs []vs.Document) ([]vs.Document, error)
• type FilterMarkdownDocsNoContent
  • func (f *FilterMarkdownDocsNoContent) Transform(_ context.Context, docs []vs.Document) ([]vs.Document, error)
• type GenericTransformer
  • func (g *GenericTransformer) Transform(ctx context.Context, docs []vs.Document) ([]vs.Document, error)
• type KeywordExtractor
  • func NewKeyWordExtractor(numKeywords int, llm llm.LLM) *KeywordExtractor
  • func (k *KeywordExtractor) Transform(ctx context.Context, docs []vs.Document) ([]vs.Document, error)

Variables

var TransformerMap = map[string]types.DocumentTransformer{
	"extra_metadata":                  &ExtraMetadata{},
	"filter_markdown_docs_no_content": &FilterMarkdownDocsNoContent{},
	"keywords":                        &KeywordExtractor{},
}

Functions

func DefaultDocumentTransformers

func DefaultDocumentTransformers(filetype string) (transformers []types.DocumentTransformer)

func GetTransformer

func GetTransformer(name string) (types.DocumentTransformer, error)

Types

type ExtraMetadata

type ExtraMetadata struct {
	Metadata map[string]any
}

func (*ExtraMetadata) Transform

func (e *ExtraMetadata) Transform(_ context.Context, docs []vs.Document) ([]vs.Document, error)

type FilterMarkdownDocsNoContent

type FilterMarkdownDocsNoContent struct{}

FilterMarkdownDocsNoContent filters out Markdown documents with no content or only headings

TODO: this may be moved into the MarkdownTextSplitter

func (*FilterMarkdownDocsNoContent) Transform

func (f *FilterMarkdownDocsNoContent) Transform(_ context.Context, docs []vs.Document) ([]vs.Document, error)

type GenericTransformer

type GenericTransformer struct {
	TransformationFunc func(context.Context, []vs.Document) ([]vs.Document, error)
}

func (*GenericTransformer) Transform

func (g *GenericTransformer) Transform(ctx context.Context, docs []vs.Document) ([]vs.Document, error)

type KeywordExtractor

type KeywordExtractor struct {
	NumKeywords int
	LLM         llm.LLM
}

func NewKeyWordExtractor

func NewKeyWordExtractor(numKeywords int, llm llm.LLM) *KeywordExtractor

func (*KeywordExtractor) Transform

func (k *KeywordExtractor) Transform(ctx context.Context, docs []vs.Document) ([]vs.Document, error)