weaviate
36 строк · 1009.0 Байт
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
12package modopenai13
14import (15"github.com/weaviate/weaviate/entities/modulecapabilities"16"github.com/weaviate/weaviate/usecases/modulecomponents/arguments/nearText"17)
18
19func (m *OpenAIModule) initNearText() error {20m.searcher = nearText.NewSearcher(m.vectorizer)21m.graphqlProvider = nearText.New(m.nearTextTransformer)22return nil23}
24
25func (m *OpenAIModule) Arguments() map[string]modulecapabilities.GraphQLArgument {26return m.graphqlProvider.Arguments()27}
28
29func (m *OpenAIModule) VectorSearches() map[string]modulecapabilities.VectorForParams {30return m.searcher.VectorSearches()31}
32
33var (34_ = modulecapabilities.GraphQLArguments(New())35_ = modulecapabilities.Searcher(New())36)
37