storage

Documentation

Index

• type BadgerStore
  • func NewBadgerStore(path string) (*BadgerStore, error)
  • func (s *BadgerStore) Close()
  • func (s *BadgerStore) Delete(ctx context.Context, key string) error
  • func (s *BadgerStore) Get(ctx context.Context, key string) ([]byte, error)
  • func (s *BadgerStore) GetAllEmbeddings(ctx context.Context) (map[string][]byte, error)
  • func (s *BadgerStore) GetPrompt(ctx context.Context, key string) (string, error)
  • func (s *BadgerStore) Set(ctx context.Context, key string, value []byte) error
• type Storage

Types

type BadgerStore

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

func NewBadgerStore

func NewBadgerStore(path string) (*BadgerStore, error)

func (*BadgerStore) Close

func (s *BadgerStore) Close()

func (*BadgerStore) Delete

func (s *BadgerStore) Delete(ctx context.Context, key string) error

func (*BadgerStore) Get

func (s *BadgerStore) Get(ctx context.Context, key string) ([]byte, error)

func (*BadgerStore) GetAllEmbeddings

func (s *BadgerStore) GetAllEmbeddings(ctx context.Context) (map[string][]byte, error)

func (*BadgerStore) GetPrompt

func (s *BadgerStore) GetPrompt(ctx context.Context, key string) (string, error)

func (*BadgerStore) Set

func (s *BadgerStore) Set(ctx context.Context, key string, value []byte) error

type Storage

type Storage interface {
	Set(ctx context.Context, key string, value []byte) error
	Get(ctx context.Context, key string) ([]byte, error)
	Delete(ctx context.Context, key string) error
	GetAllEmbeddings(ctx context.Context) (map[string][]byte, error)
	GetPrompt(ctx context.Context, key string) (string, error)
	Close()
}