database

Documentation

Index

• Constants
• func Store(filePath string, embeddings []Document) error
• type Database
  • func New(filePath string) (*Database, error)
  • func (db *Database) Add(doc Document)
  • func (db *Database) GetEmbeddingByIndex(i int) []float64
  • func (db *Database) GetEmbeddings() [][]float64
  • func (db *Database) GetTextByIndex(i int) string
  • func (db *Database) GetTexts() []string
  • func (db *Database) RemoveByIndex(i int)
  • func (db *Database) RemoveByText(text string)
  • func (db *Database) Save() error
• type DatabaseInterface
• type Document
  • func Load(filePath string) ([]Document, error)

Constants

const (
	ErrorFileNotFound = "file not found"
)

Functions

func Store

func Store(filePath string, embeddings []Document) error

Types

type Database

type Database struct {
	Documents []Document
	// contains filtered or unexported fields
}

func New

func New(filePath string) (*Database, error)

func (*Database) Add

func (db *Database) Add(doc Document)

func (*Database) GetEmbeddingByIndex

func (db *Database) GetEmbeddingByIndex(i int) []float64

func (*Database) GetEmbeddings

func (db *Database) GetEmbeddings() [][]float64

func (*Database) GetTextByIndex

func (db *Database) GetTextByIndex(i int) string

func (*Database) GetTexts

func (db *Database) GetTexts() []string

func (*Database) RemoveByIndex

func (db *Database) RemoveByIndex(i int)

func (*Database) RemoveByText

func (db *Database) RemoveByText(text string)

func (*Database) Save

func (db *Database) Save() error

type DatabaseInterface

type DatabaseInterface interface {
	Add(doc Document)
	Save() error
	Close() error
	GetTextByIndex(i int) string
	GetEmbeddingByIndex(i int) []float64
	GetEmbeddings() [][]float64
	GetTexts() []string
}

type Document

type Document struct {
	Text      string    `json:"text"`
	Embedding []float64 `json:"embedding"`
}

func Load

func Load(filePath string) ([]Document, error)