mirror of
https://github.com/lukaszraczylo/claude-mnemonic.git
synced 2026-06-06 23:13:50 +00:00
lukaszraczylo
74ae8ed4c1
- [x] Add AST-aware code chunking for Go, Python, and TypeScript using tree-sitter - [x] Implement LEANN-inspired hybrid vector storage with hub detection and selective embedding storage (60-80% savings) - [x] Add observation relationship graph with CSR format and edge detection (file overlap, semantic similarity, temporal, concept) - [x] Implement graph-aware search with two-level traversal and relationship-based ranking - [x] Add auto-tuning system for dynamic hub threshold adjustment based on query performance - [x] Add comprehensive metrics tracking for vector storage, queries, latency, and graph traversals - [x] Update configuration system with graph and hybrid storage settings - [x] Add graph stats and vector metrics endpoints to worker service - [x] Enhance UI sidebar with advanced metrics display and graph visualization - [x] Optimize struct field alignment throughout codebase for memory efficiency - [x] Update documentation with LEANN Phase 2 features and performance benefits - [x] Add tree-sitter dependency for AST parsing
43 lines
1.5 KiB
Go
43 lines
1.5 KiB
Go
// Package vector provides common interfaces for vector storage implementations
|
|
package vector
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/lukaszraczylo/claude-mnemonic/internal/vector/sqlitevec"
|
|
)
|
|
|
|
// Client defines the interface for vector storage operations.
|
|
// Both sqlitevec.Client and hybrid.Client implement this interface.
|
|
type Client interface {
|
|
// AddDocuments adds documents with their embeddings to the vector store
|
|
AddDocuments(ctx context.Context, docs []sqlitevec.Document) error
|
|
|
|
// DeleteDocuments removes documents by their IDs
|
|
DeleteDocuments(ctx context.Context, ids []string) error
|
|
|
|
// Query performs a vector similarity search
|
|
Query(ctx context.Context, query string, limit int, where map[string]any) ([]sqlitevec.QueryResult, error)
|
|
|
|
// IsConnected checks if the vector store is available
|
|
IsConnected() bool
|
|
|
|
// Close releases resources
|
|
Close() error
|
|
|
|
// Count returns the total number of vectors in the store
|
|
Count(ctx context.Context) (int64, error)
|
|
|
|
// ModelVersion returns the current embedding model version
|
|
ModelVersion() string
|
|
|
|
// NeedsRebuild checks if vectors need to be rebuilt due to model version change
|
|
NeedsRebuild(ctx context.Context) (bool, string)
|
|
|
|
// GetStaleVectors returns doc_ids of vectors with mismatched or null model versions
|
|
GetStaleVectors(ctx context.Context) ([]sqlitevec.StaleVectorInfo, error)
|
|
|
|
// DeleteVectorsByDocIDs removes vectors by their doc_ids
|
|
DeleteVectorsByDocIDs(ctx context.Context, docIDs []string) error
|
|
}
|