Fix: RAG updates

This commit is contained in:
Grail Finder
2025-01-09 15:49:59 +03:00
parent 7bbedd93cf
commit 363bbae2c7
8 changed files with 334 additions and 195 deletions

5
bot.go
View File

@@ -138,17 +138,12 @@ func chatRagUse(qText string) (string, error) {
logger.Error("failed to get embs", "error", err, "index", i, "question", q)
continue
}
// e := &models.EmbeddingResp{
// Embedding: emb,
// }
// vecs, err := ragger.SearchEmb(e)
vecs, err := store.SearchClosest(emb)
if err != nil {
logger.Error("failed to query embs", "error", err, "index", i, "question", q)
continue
}
respVecs = append(respVecs, vecs...)
// logger.Info("returned vector from query search", "question", q, "vec", vec)
}
// get raw text
resps := []string{}