index

Documentation

Index

• Variables
• type Embedder
• type Option
  • func WithTopK(topK int) Option
• type Pinecone
  • func NewPinecone(options PineconeOptions, embedder Embedder) *Pinecone
  • func (p *Pinecone) IsEmpty(ctx context.Context) (bool, error)
  • func (p *Pinecone) LoadFromDocuments(ctx context.Context, documents []document.Document) error
  • func (p *Pinecone) SimilaritySearch(ctx context.Context, query string, opts ...Option) (SearchResponses, error)
  • func (p *Pinecone) WithAPIKeyAndEnvironment(apiKey, environment string) *Pinecone
• type PineconeCreateIndexOptions
• type PineconeOptions
• type SearchResponse
• type SearchResponses
  • func (s SearchResponses) ToDocuments() []document.Document
• type SimpleVectorIndex
  • func NewSimpleVectorIndex(name string, outputPath string, embedder Embedder) *SimpleVectorIndex
  • func (s *SimpleVectorIndex) IsEmpty() (bool, error)
  • func (s *SimpleVectorIndex) LoadFromDocuments(ctx context.Context, documents []document.Document) error
  • func (s *SimpleVectorIndex) SimilaritySearch(ctx context.Context, query string, opts ...Option) (SearchResponses, error)

Variables

var (
	ErrInternal = "internal index error"
)

Types

type Embedder

type Embedder interface {
	Embed(ctx context.Context, texts []string) ([]embedder.Embedding, error)
}

type Option

type Option func(*options)

func WithTopK

func WithTopK(topK int) Option

type Pinecone

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

func NewPinecone

func NewPinecone(options PineconeOptions, embedder Embedder) *Pinecone

func (*Pinecone) IsEmpty

func (p *Pinecone) IsEmpty(ctx context.Context) (bool, error)

func (*Pinecone) LoadFromDocuments

func (p *Pinecone) LoadFromDocuments(ctx context.Context, documents []document.Document) error

func (*Pinecone) SimilaritySearch

func (p *Pinecone) SimilaritySearch(ctx context.Context, query string, opts ...Option) (SearchResponses, error)

func (*Pinecone) WithAPIKeyAndEnvironment

func (p *Pinecone) WithAPIKeyAndEnvironment(apiKey, environment string) *Pinecone

type PineconeCreateIndexOptions

type PineconeCreateIndexOptions struct {
	Dimension int
	Replicas  int
	Metric    string
	PodType   string
}

type PineconeOptions

type PineconeOptions struct {
	IndexName       string
	Namespace       string
	IncludeContent  bool
	BatchUpsertSize *int
	CreateIndex     *PineconeCreateIndexOptions
}

type SearchResponse

type SearchResponse struct {
	ID       string
	Document document.Document
	Score    float64
}

type SearchResponses

type SearchResponses []SearchResponse

func (SearchResponses) ToDocuments

func (s SearchResponses) ToDocuments() []document.Document

type SimpleVectorIndex

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

func NewSimpleVectorIndex

func NewSimpleVectorIndex(name string, outputPath string, embedder Embedder) *SimpleVectorIndex

func (*SimpleVectorIndex) IsEmpty

func (s *SimpleVectorIndex) IsEmpty() (bool, error)

func (*SimpleVectorIndex) LoadFromDocuments

func (s *SimpleVectorIndex) LoadFromDocuments(ctx context.Context, documents []document.Document) error

func (*SimpleVectorIndex) SimilaritySearch

func (s *SimpleVectorIndex) SimilaritySearch(ctx context.Context, query string, opts ...Option) (SearchResponses, error)