storage

Documentation

Index

• type BadgerStore
  • func NewBadgerStore(path string) (*BadgerStore, error)
  • func (s *BadgerStore) Close()
  • func (s *BadgerStore) Count(ctx context.Context) (int64, error)
  • func (s *BadgerStore) Delete(ctx context.Context, key string) error
  • func (s *BadgerStore) DeleteByPrefix(ctx context.Context, prefix string) (int, 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) GetAllKeys(ctx context.Context, prefix string) ([]string, error)
  • func (s *BadgerStore) GetPrompt(ctx context.Context, key string) (string, error)
  • func (s *BadgerStore) RunGC() error
  • func (s *BadgerStore) Set(ctx context.Context, key string, value []byte) error
  • func (s *BadgerStore) SetWithTTL(ctx context.Context, key string, value []byte, ttl time.Duration) error
  • func (s *BadgerStore) Sync() 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) Count

func (s *BadgerStore) Count(ctx context.Context) (int64, error)

Count returns the total number of keys in the store

func (*BadgerStore) Delete

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

func (*BadgerStore) DeleteByPrefix

func (s *BadgerStore) DeleteByPrefix(ctx context.Context, prefix string) (int, error)

DeleteByPrefix deletes all keys with the given prefix

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) GetAllKeys

func (s *BadgerStore) GetAllKeys(ctx context.Context, prefix string) ([]string, error)

GetAllKeys returns all keys with the given prefix

func (*BadgerStore) GetPrompt

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

func (*BadgerStore) RunGC

func (s *BadgerStore) RunGC() error

RunGC runs garbage collection on the database

func (*BadgerStore) Set

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

func (*BadgerStore) SetWithTTL

func (s *BadgerStore) SetWithTTL(ctx context.Context, key string, value []byte, ttl time.Duration) error

SetWithTTL sets a key-value pair with a time-to-live

func (*BadgerStore) Sync

func (s *BadgerStore) Sync() error

Sync flushes all writes to disk

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()
}