Fixups to long memory, and avoid re-initing RAGDB recursively
Signed-off-by: mudler <mudler@localai.io>
This commit is contained in:
@@ -82,7 +82,7 @@ func main() {
|
||||
|
||||
if len(db.Database) > 0 && kbdisableIndexing != "true" {
|
||||
xlog.Info("Loading knowledgebase from disk, to skip run with KBDISABLEINDEX=true")
|
||||
if err := db.SaveAllToStore(); err != nil {
|
||||
if err := db.PopulateRAGDB(); err != nil {
|
||||
xlog.Info("Error storing in the KB", err)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -63,13 +63,13 @@ func NewInMemoryDB(knowledgebase string, store RAGDB) (*InMemoryDatabase, error)
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (db *InMemoryDatabase) SaveAllToStore() error {
|
||||
func (db *InMemoryDatabase) PopulateRAGDB() error {
|
||||
for _, d := range db.Database {
|
||||
if d == "" {
|
||||
// skip empty chunks
|
||||
continue
|
||||
}
|
||||
err := db.Store(d)
|
||||
err := db.RAGDB.Store(d)
|
||||
if err != nil {
|
||||
return fmt.Errorf("error storing in the KB: %w", err)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user