embedstore

package

v0.8.6 Latest Latest Go to latest Published: Jun 14, 2023 License: MIT Imports: 9 Imported by: 2

Details

This section is empty.

This section is empty.

func TightenSearchResults(searchResults []types.SearchResult) types.SearchResults

Sorts search results in a way that preserves original order based on filenames

type EmbedStore struct {
	types.UnimplementedTGenerator
}

func NewEmbedStore() (*EmbedStore, error)

func (idx *EmbedStore) AddEmbeddingDocument(ctx context.Context, docID string, embedding [1536]float32, metadata types.Metadata) error

func (idx *EmbedStore) AddEmbeddingDocuments(ctx context.Context, vectors []types.Vector) (int, error)

func (idx *EmbedStore) DeleteEmbeddingDocument(ctx context.Context, docID string) error

func (idx *EmbedStore) DeleteEmbeddingDocuments(ctx context.Context, docIDs []string) error

func (idx *EmbedStore) GetEmbeddingDocument(ctx context.Context, docID string) (types.Vector, bool, error)

func (idx *EmbedStore) ListAllEmbeddingsIds(ctx context.Context) (types.SearchResults, error)

func (idx *EmbedStore) SearchWithEmbedding(ctx context.Context, embedding types.QueryFilter, k int) (types.SearchResults, error)