retriever

Documentation

Index

• func AmazonKendraLanguageCodeAttributeFilter(languageCode string) *types.AttributeFilter
• type AmazonKendra
  • func NewAmazonKendra(client AmazonKendraClient, index string, ...) *AmazonKendra
  • func (r *AmazonKendra) GetRelevantDocuments(ctx context.Context, query string) ([]schema.Document, error)
• type AmazonKendraClient
• type AmazonKendraOptions
• type VectorStore
  • func NewVectorStore(vectorStore schema.VectorStore, optFns ...func(o *VectorStoreOptions)) *VectorStore
  • func (r *VectorStore) GetRelevantDocuments(ctx context.Context, query string) ([]schema.Document, error)
• type VectorStoreOptions
• type VectorStoreSearchType

Functions

func AmazonKendraLanguageCodeAttributeFilter

func AmazonKendraLanguageCodeAttributeFilter(languageCode string) *types.AttributeFilter

Types

type AmazonKendra

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

func NewAmazonKendra

func NewAmazonKendra(client AmazonKendraClient, index string, optFns ...func(o *AmazonKendraOptions)) *AmazonKendra

func (*AmazonKendra) GetRelevantDocuments

func (r *AmazonKendra) GetRelevantDocuments(ctx context.Context, query string) ([]schema.Document, error)

type AmazonKendraClient

type AmazonKendraClient interface {
	// Retrieve retrieves documents from Amazon Kendra based on the provided input parameters.
	// It returns the retrieval output or an error if the retrieval operation fails.
	Retrieve(ctx context.Context, params *kendra.RetrieveInput, optFns ...func(*kendra.Options)) (*kendra.RetrieveOutput, error)

	// Query performs a query operation on Amazon Kendra using the specified input parameters.
	// It returns the query output or an error if the query operation fails.
	Query(ctx context.Context, params *kendra.QueryInput, optFns ...func(*kendra.Options)) (*kendra.QueryOutput, error)
}

AmazonKendraClient represents a client for interacting with Amazon Kendra.

type AmazonKendraOptions

type AmazonKendraOptions struct {
	// Number of documents to query for
	TopK int32

	// Provides filtering the results based on document attributes or metadata
	// fields.
	AttributeFilter *types.AttributeFilter
}

type VectorStore

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

func NewVectorStore

func NewVectorStore(vectorStore schema.VectorStore, optFns ...func(o *VectorStoreOptions)) *VectorStore

func (*VectorStore) GetRelevantDocuments

func (r *VectorStore) GetRelevantDocuments(ctx context.Context, query string) ([]schema.Document, error)

GetRelevantDocuments returns documents using the vector store.

type VectorStoreOptions

type VectorStoreOptions struct {
	SearchType VectorStoreSearchType
}

type VectorStoreSearchType

type VectorStoreSearchType string

const (
	VectorStoreSearchTypeSimilarity VectorStoreSearchType = "similarity"
)