github.com/weaviate/weaviate@v1.24.6/modules/qna-transformers/additional/provider.go (about)

     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 additional
    13  
    14  import (
    15  	"context"
    16  
    17  	"github.com/tailor-inc/graphql"
    18  	"github.com/tailor-inc/graphql/language/ast"
    19  	"github.com/weaviate/weaviate/entities/modulecapabilities"
    20  	"github.com/weaviate/weaviate/entities/moduletools"
    21  	"github.com/weaviate/weaviate/entities/search"
    22  )
    23  
    24  type AdditionalProperty interface {
    25  	AdditionalPropertyFn(ctx context.Context,
    26  		in []search.Result, params interface{}, limit *int,
    27  		argumentModuleParams map[string]interface{}, cfg moduletools.ClassConfig) ([]search.Result, error)
    28  	ExtractAdditionalFn(param []*ast.Argument) interface{}
    29  	AdditionalPropertyDefaultValue() interface{}
    30  	AdditionalFieldFn(classname string) *graphql.Field
    31  }
    32  
    33  type GraphQLAdditionalArgumentsProvider struct {
    34  	answerProvider AdditionalProperty
    35  }
    36  
    37  func New(answerProvider AdditionalProperty) *GraphQLAdditionalArgumentsProvider {
    38  	return &GraphQLAdditionalArgumentsProvider{answerProvider}
    39  }
    40  
    41  func (p *GraphQLAdditionalArgumentsProvider) AdditionalProperties() map[string]modulecapabilities.AdditionalProperty {
    42  	additionalProperties := map[string]modulecapabilities.AdditionalProperty{}
    43  	additionalProperties["answer"] = p.getAnswer()
    44  	return additionalProperties
    45  }
    46  
    47  func (p *GraphQLAdditionalArgumentsProvider) getAnswer() modulecapabilities.AdditionalProperty {
    48  	return modulecapabilities.AdditionalProperty{
    49  		GraphQLNames:           []string{"answer"},
    50  		GraphQLFieldFunction:   p.answerProvider.AdditionalFieldFn,
    51  		GraphQLExtractFunction: p.answerProvider.ExtractAdditionalFn,
    52  		SearchFunctions: modulecapabilities.AdditionalSearch{
    53  			ExploreGet:  p.answerProvider.AdditionalPropertyFn,
    54  			ExploreList: p.answerProvider.AdditionalPropertyFn,
    55  		},
    56  	}
    57  }