summaryrefslogtreecommitdiff
path: root/rag/rag.go
diff options
context:
space:
mode:
Diffstat (limited to 'rag/rag.go')
-rw-r--r--rag/rag.go4
1 files changed, 2 insertions, 2 deletions
diff --git a/rag/rag.go b/rag/rag.go
index c560f33..7fbbe98 100644
--- a/rag/rag.go
+++ b/rag/rag.go
@@ -163,9 +163,9 @@ func (r *RAG) writeVectors(vectorCh chan []models.VectorRow) error {
for batch := range vectorCh {
for _, vector := range batch {
if err := r.storage.WriteVector(&vector); err != nil {
- r.logger.Error("failed to write vector", "error", err, "slug", vector.Slug)
+ r.logger.Error("failed to write vector to DB", "error", err, "slug", vector.Slug)
LongJobStatusCh <- ErrRAGStatus
- continue // a duplicate is not critical
+ return err // Stop the entire RAG operation on DB error
}
}
r.logger.Debug("wrote batch to db", "size", len(batch), "vector_chan_len", len(vectorCh))