mongo

Documentation

Overview

Package mongo provides a MongoDB store for embeddings.

Index

• type Store
  • func New(collection *mongo.Collection) *Store
  • func (s *Store) Close() error
  • func (s *Store) Get(ctx context.Context, utterance string) ([]float64, error)
  • func (s *Store) Set(ctx context.Context, keyValPair semanticrouter.Utterance) error

Types

type Store

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

Store is a MongoDB store.

It implements the Store interface.

func New

func New(collection *mongo.Collection) *Store

New creates a new MongoDB store.

func (*Store) Close

func (s *Store) Close() error

Close closes the MongoDB connection.

func (*Store) Get

func (s *Store) Get(ctx context.Context, utterance string) ([]float64, error)

Get gets a value from the store.

func (*Store) Set

func (s *Store) Set(ctx context.Context, keyValPair semanticrouter.Utterance) error

Set stores a value in the store.