engine

package

v0.4.1 Latest Latest Go to latest Published: Jun 9, 2025 License: MIT Imports: 7 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

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

func NewChromemDBCollection(collection, path string, openaiClient *openai.Client, embeddingsModel string) (*ChromemDB, error)

func (c *ChromemDB) Count() int

func (c *ChromemDB) Delete(where map[string]string, whereDocuments map[string]string, ids ...string) error

func (c *ChromemDB) GetByID(id string) (types.Result, error)

func (c *ChromemDB) GetEmbeddingDimensions() (int, error)

func (c *ChromemDB) Reset() error

func (c *ChromemDB) Search(s string, similarEntries int) ([]types.Result, error)

func (c *ChromemDB) Store(s string, metadata map[string]string) (Result, error)

func (c *ChromemDB) StoreDocuments(s []string, metadata map[string]string) ([]Result, error)

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

func NewLocalAIRAGDB(storeClient *localai.StoreClient, openaiClient *openai.Client, embeddingsModel string) *LocalAIRAGDB

func (db *LocalAIRAGDB) Count() int

func (db *LocalAIRAGDB) Delete(where map[string]string, whereDocuments map[string]string, ids ...string) error

func (db *LocalAIRAGDB) GetByID(id string) (types.Result, error)

func (db *LocalAIRAGDB) GetEmbeddingDimensions() (int, error)

func (db *LocalAIRAGDB) Reset() error

func (db *LocalAIRAGDB) Search(s string, similarEntries int) ([]types.Result, error)

func (db *LocalAIRAGDB) Store(s string, metadata map[string]string) (Result, error)

func (db *LocalAIRAGDB) StoreDocuments(s []string, metadata map[string]string) ([]Result, error)

type Result struct {
	ID string `json:"id"`
}

Path	Synopsis
localai TODO: this is a duplicate of LocalAI/core/store/client.go	TODO: this is a duplicate of LocalAI/core/store/client.go