From 7bbedd93cf078fc7496a6779cf9eda6e588e64c0 Mon Sep 17 00:00:00 2001
From: Grail Finder <wohilas@gmail.com>
Date: Tue, 7 Jan 2025 11:11:44 +0300
Subject: Enha (RAG): raw text as primary key in vector db

---
 storage/vector.go | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

(limited to 'storage/vector.go')

diff --git a/storage/vector.go b/storage/vector.go
index 1579686..fe479d8 100644
--- a/storage/vector.go
+++ b/storage/vector.go
@@ -68,7 +68,6 @@ func (p ProviderSQL) WriteVector(row *models.VectorRow) error {
 	}
 	err = stmt.Exec()
 	if err != nil {
-		p.logger.Error("failed exec a stmt", "error", err)
 		return err
 	}
 	return nil
@@ -85,7 +84,6 @@ func (p ProviderSQL) SearchClosest(q []float32) ([]models.VectorRow, error) {
 	}
 	stmt, _, err := p.s3Conn.Prepare(
 		fmt.Sprintf(`SELECT
-			id,
 			distance,
 			embedding,
 			slug,
@@ -109,12 +107,11 @@ func (p ProviderSQL) SearchClosest(q []float32) ([]models.VectorRow, error) {
 	resp := []models.VectorRow{}
 	for stmt.Step() {
 		res := models.VectorRow{}
-		res.ID = uint32(stmt.ColumnInt64(0))
-		res.Distance = float32(stmt.ColumnFloat(1))
-		emb := stmt.ColumnRawText(2)
+		res.Distance = float32(stmt.ColumnFloat(0))
+		emb := stmt.ColumnRawText(1)
 		res.Embeddings = decodeUnsafe(emb)
-		res.Slug = stmt.ColumnText(3)
-		res.RawText = stmt.ColumnText(4)
+		res.Slug = stmt.ColumnText(2)
+		res.RawText = stmt.ColumnText(3)
 		resp = append(resp, res)
 	}
 	if err := stmt.Err(); err != nil {
-- 
cgit v1.2.3