weaviate

1
//                           _       _
2
// __      _____  __ ___   ___  __ _| |_ ___
3
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
4
//  \ V  V /  __/ (_| |\ V /| | (_| | ||  __/
5
//   \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
6
//
7
//  Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
8
//
9
//  CONTACT: hello@weaviate.io
10
//
11

12
package texttransformer
13

14
import (
15
	"github.com/weaviate/weaviate/entities/modulecapabilities"
16
)
17

18
type TextTransformerProvider struct {
19
	autocorrecProvider modulecapabilities.TextTransform
20
}
21

22
func New(autocorrecProvider modulecapabilities.TextTransform) *TextTransformerProvider {
23
	return &TextTransformerProvider{autocorrecProvider}
24
}
25

26
func (p *TextTransformerProvider) TextTransformers() map[string]modulecapabilities.TextTransform {
27
	textTransformers := map[string]modulecapabilities.TextTransform{}
28
	textTransformers["nearText"] = p.autocorrecProvider
29
	textTransformers["ask"] = p.autocorrecProvider
30
	return textTransformers
31
}
32