diff options
| author | Grail Finder <wohilas@gmail.com> | 2025-01-07 11:11:44 +0300 |
|---|---|---|
| committer | Grail Finder <wohilas@gmail.com> | 2025-01-07 11:11:44 +0300 |
| commit | 7bbedd93cf078fc7496a6779cf9eda6e588e64c0 (patch) | |
| tree | 954a8c767362fab8da11fe1fad4643f2a44b62c9 /storage/vector.go | |
| parent | b822b3a1613ef7f1c9ed8fa5aaddfaffbfc513a4 (diff) | |
Enha (RAG): raw text as primary key in vector db
Diffstat (limited to 'storage/vector.go')
| -rw-r--r-- | storage/vector.go | 11 |
1 files changed, 4 insertions, 7 deletions
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 { |