index

package

v0.0.3-alpha1 Latest Latest Go to latest Published: Apr 27, 2023 License: MIT Imports: 14 Imported by: 2

Details

This section is empty.

This section is empty.

This section is empty.

type Embedder interface {
	Embed(ctx context.Context, docs []document.Document) ([]embedder.Embedding, error)
}

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

func NewPinecone(indexName, projectID string, embedder Embedder) (*Pinecone, error)

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

func (s *Pinecone) SimilaritySearch(ctx context.Context, query string, topK *int) ([]SearchResponse, error)

func (s *Pinecone) Size() (int64, error)

type SearchResponse struct {
	Document document.Document
	Score    float32
	Index    int
}

type SimpleVectorIndex struct {
	Data []SimpleVectorIndexData `json:"data"`
	// contains filtered or unexported fields
}

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

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

func (s *SimpleVectorIndex) SimilaritySearch(ctx context.Context, query string, topK *int) ([]SearchResponse, error)

func (s *SimpleVectorIndex) Size() (int64, error)

type SimpleVectorIndexData struct {
	Document  document.Document  `json:"document"`
	Embedding embedder.Embedding `json:"embedding"`
}