retriever

Documentation

Overview

Package retriever provides functionality for retrieving relevant documents using various services.

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 AzureCognitiveSearch
  • func NewAzureCognitiveSearch(apiKey, serviceName, indexName string, ...) *AzureCognitiveSearch
  • func (r *AzureCognitiveSearch) GetRelevantDocuments(ctx context.Context, query string) ([]schema.Document, error)
• type AzureCognitiveSearchOptions
• type AzureCognitiveSearchRequest
• type HTTPClient
• type Merger
  • func NewMerger(retrievers ...schema.Retriever) *Merger
  • func (r *Merger) GetRelevantDocuments(ctx context.Context, query string) ([]schema.Document, error)
• 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 AzureCognitiveSearch

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

AzureCognitiveSearch is a retriever implementation for Azure Cognitive Search service.

func NewAzureCognitiveSearch

func NewAzureCognitiveSearch(apiKey, serviceName, indexName string, optFns ...func(o *AzureCognitiveSearchOptions)) *AzureCognitiveSearch

NewAzureCognitiveSearch creates a new instance of AzureCognitiveSearch retriever with the provided options.

func (*AzureCognitiveSearch) GetRelevantDocuments

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

GetRelevantDocuments retrieves relevant documents for the given query using Azure Cognitive Search.

type AzureCognitiveSearchOptions

type AzureCognitiveSearchOptions struct {
	// Number of documents to query for
	TopK uint

	// Azure Cognitive Search API version.
	APIVersion string

	// Key to extract content from response.
	ContentKey string

	// HTTP client to use for making requests.
	HTTPClient HTTPClient
}

AzureCognitiveSearchOptions contains options for configuring the AzureCognitiveSearch retriever.

type AzureCognitiveSearchRequest

type AzureCognitiveSearchRequest struct {
	Search string `json:"search"`
	Top    uint   `json:"top"`
}

AzureCognitiveSearchRequest represents the request payload for Azure Cognitive Search.

type HTTPClient

type HTTPClient interface {
	Do(req *http.Request) (*http.Response, error)
}

type Merger

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

func NewMerger

func NewMerger(retrievers ...schema.Retriever) *Merger

func (*Merger) GetRelevantDocuments

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

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"
)