summaryrefslogtreecommitdiff
path: root/storage
diff options
context:
space:
mode:
authorGrail Finder <wohilas@gmail.com>2026-02-25 20:06:56 +0300
committerGrail Finder <wohilas@gmail.com>2026-02-25 20:06:56 +0300
commit888c9fec652b82174702c710f54f7d64f194315c (patch)
tree883051d653dda2d57b227670bfd3721bf6cf426a /storage
parent4f07994bdc3d23421cf3941af3edc18c05ffc94b (diff)
Chore: linter complaints
Diffstat (limited to 'storage')
-rw-r--r--storage/storage.go1
-rw-r--r--storage/vector.go18
2 files changed, 1 insertions, 18 deletions
diff --git a/storage/storage.go b/storage/storage.go
index a092f8d..7294261 100644
--- a/storage/storage.go
+++ b/storage/storage.go
@@ -103,7 +103,6 @@ func NewProviderSQL(dbPath string, logger *slog.Logger) FullRepo {
return nil
}
p := ProviderSQL{db: db, logger: logger}
-
p.Migrate()
return p
}
diff --git a/storage/vector.go b/storage/vector.go
index 32b4731..75f5c9a 100644
--- a/storage/vector.go
+++ b/storage/vector.go
@@ -73,12 +73,9 @@ func (p ProviderSQL) WriteVector(row *models.VectorRow) error {
if err != nil {
return err
}
-
serializedEmbeddings := SerializeVector(row.Embeddings)
-
query := fmt.Sprintf("INSERT INTO %s(embeddings, slug, raw_text, filename) VALUES (?, ?, ?, ?)", tableName)
_, err = p.db.Exec(query, serializedEmbeddings, row.Slug, row.RawText, row.FileName)
-
return err
}
@@ -87,27 +84,22 @@ func (p ProviderSQL) SearchClosest(q []float32) ([]models.VectorRow, error) {
if err != nil {
return nil, err
}
-
querySQL := "SELECT embeddings, slug, raw_text, filename FROM " + tableName
rows, err := p.db.Query(querySQL)
if err != nil {
return nil, err
}
defer rows.Close()
-
type SearchResult struct {
vector models.VectorRow
distance float32
}
-
var topResults []SearchResult
-
for rows.Next() {
var (
- embeddingsBlob []byte
+ embeddingsBlob []byte
slug, rawText, fileName string
)
-
if err := rows.Scan(&embeddingsBlob, &slug, &rawText, &fileName); err != nil {
continue
}
@@ -152,7 +144,6 @@ func (p ProviderSQL) SearchClosest(q []float32) ([]models.VectorRow, error) {
result.vector.Distance = result.distance
results[i] = result.vector
}
-
return results, nil
}
@@ -161,18 +152,15 @@ func cosineSimilarity(a, b []float32) float32 {
if len(a) != len(b) {
return 0.0
}
-
var dotProduct, normA, normB float32
for i := 0; i < len(a); i++ {
dotProduct += a[i] * b[i]
normA += a[i] * a[i]
normB += b[i] * b[i]
}
-
if normA == 0 || normB == 0 {
return 0.0
}
-
return dotProduct / (sqrt(normA) * sqrt(normB))
}
@@ -229,13 +217,11 @@ func (p ProviderSQL) ListFiles() ([]string, error) {
}
}
}
-
return allFiles, nil
}
func (p ProviderSQL) RemoveEmbByFileName(filename string) error {
var errors []string
-
tableNames := []string{
"embeddings_384", "embeddings_768", "embeddings_1024", "embeddings_1536",
"embeddings_2048", "embeddings_3072", "embeddings_4096", "embeddings_5120",
@@ -246,10 +232,8 @@ func (p ProviderSQL) RemoveEmbByFileName(filename string) error {
errors = append(errors, err.Error())
}
}
-
if len(errors) > 0 {
return fmt.Errorf("errors occurred: %v", errors)
}
-
return nil
}