mirror of
https://github.com/lukaszraczylo/claude-mnemonic.git
synced 2026-06-06 23:13:50 +00:00
5c2685c7b6
* feat(leann-phase2): implement hybrid vector storage and graph-based search
- [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
* fix: add fts5 build tag to CI workflow
Pass build-tags: "fts5" to shared workflow to properly compile
sqlite-vec-go-bindings with SQLite FTS5 support.
This fixes test failures in hybrid vector storage tests that require
CGO and FTS5 build tags.
Requires shared-actions@8f7f235 or later.
* docs: add testing documentation and macOS ARM64 known issue
Document the macOS ARM64 CGO linking issue with sqlite-vec-go-bindings
that prevents hybrid package tests from compiling locally.
Added:
- .github/TESTING.md: Comprehensive testing guide with platform-specific
issues, workarounds, and CI configuration details
- internal/vector/hybrid/README.md: Package-specific documentation
explaining the macOS limitation
- .github/CI_FIX_SUMMARY.md: Technical details of the CI fix
Key points:
- 41 out of 42 packages test successfully on all platforms
- hybrid package tests fail only on macOS ARM64 (local dev issue)
- Linux CI tests pass with proper build-tags: "fts5" configuration
- Production builds and runtime functionality unaffected
This is a known limitation of sqlite-vec-go-bindings on macOS ARM64
and does not impact CI/CD or production deployments.
* fix: add SQLite busy_timeout to prevent database locked errors
Set PRAGMA busy_timeout=5000 (5 seconds) to allow SQLite to retry
when the database is locked instead of failing immediately.
This fixes race conditions when multiple goroutines try to write
simultaneously, particularly in tests where StoreObservation spawns
async cleanup goroutines.
Root cause:
- StoreObservation launches goroutine -> CleanupOldObservations
- Multiple concurrent cleanups caused "database is locked" errors
- Without busy_timeout, SQLite fails immediately on lock contention
Solution:
- Add 5-second busy timeout for automatic retry on lock
- Standard practice for concurrent SQLite usage
- Works with existing WAL mode configuration
Fixes TestObservationStore_CleanupOldObservations in CI.
* docs: complete summary of all CI test fixes
Comprehensive documentation of all fixes applied:
1. Missing build tags (fts5)
2. Database locked errors (busy_timeout)
All 41/42 packages now pass tests. The hybrid package has a known
macOS ARM64 limitation that doesn't affect CI or production.
No functionality was removed - all fixes are additive only.
* fix: add SQLite driver import to hybrid tests for CGO linking
Add blank import of mattn/go-sqlite3 to hybrid test files to ensure
the SQLite driver is linked into the test binary. This provides the
SQLite symbols that sqlite-vec-go-bindings requires.
Root cause:
- hybrid package imports sqlitevec (transitively depends on sqlite-vec CGO)
- Test binary needs SQLite symbols for linking
- sqlitevec tests already had this import, but hybrid tests didn't
- Without the driver import, linker fails with "undefined symbols"
This fix enables hybrid tests to run with -race flag on all platforms.
Before: 41/42 packages pass (hybrid failed to link)
After: 42/42 packages pass ✅
Fixes hybrid test compilation on macOS ARM64, Linux, and Windows.
* docs: remove outdated macOS limitation documentation
The hybrid test linking issue has been fixed by adding the SQLite
driver import. All tests now pass on all platforms including macOS.
Removed:
- internal/vector/hybrid/README.md (documented workaround no longer needed)
- .github/TESTING.md (macOS limitation section obsolete)
All 42/42 packages now test successfully with -race flag.
* docs: final comprehensive summary of all CI fixes
All three issues now resolved:
1. Missing fts5 build tags
2. Database busy_timeout for concurrent writes
3. Missing SQLite driver import in hybrid tests
Result: 42/42 packages pass with -race on all platforms.
Credit to reviewer for identifying the race detector concern.
422 lines
13 KiB
Go
422 lines
13 KiB
Go
// Package models contains domain models for claude-mnemonic.
|
|
package models
|
|
|
|
import (
|
|
"database/sql"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
// ConflictSuite is a test suite for conflict detection operations.
|
|
type ConflictSuite struct {
|
|
suite.Suite
|
|
}
|
|
|
|
func TestConflictSuite(t *testing.T) {
|
|
suite.Run(t, new(ConflictSuite))
|
|
}
|
|
|
|
// TestConflictTypeConstants tests conflict type constants.
|
|
func (s *ConflictSuite) TestConflictTypeConstants() {
|
|
s.Equal(ConflictType("superseded"), ConflictSuperseded)
|
|
s.Equal(ConflictType("contradicts"), ConflictContradicts)
|
|
s.Equal(ConflictType("outdated_pattern"), ConflictOutdatedPattern)
|
|
}
|
|
|
|
// TestResolutionConstants tests resolution constants.
|
|
func (s *ConflictSuite) TestResolutionConstants() {
|
|
s.Equal(ConflictResolution("prefer_newer"), ResolutionPreferNewer)
|
|
s.Equal(ConflictResolution("prefer_older"), ResolutionPreferOlder)
|
|
s.Equal(ConflictResolution("manual"), ResolutionManual)
|
|
}
|
|
|
|
// TestNewObservationConflict tests conflict creation.
|
|
func (s *ConflictSuite) TestNewObservationConflict() {
|
|
conflict := NewObservationConflict(2, 1, ConflictSuperseded, ResolutionPreferNewer, "Test reason")
|
|
|
|
s.Equal(int64(2), conflict.NewerObsID)
|
|
s.Equal(int64(1), conflict.OlderObsID)
|
|
s.Equal(ConflictSuperseded, conflict.ConflictType)
|
|
s.Equal(ResolutionPreferNewer, conflict.Resolution)
|
|
s.Equal("Test reason", conflict.Reason)
|
|
s.False(conflict.Resolved)
|
|
s.NotEmpty(conflict.DetectedAt)
|
|
s.Greater(conflict.DetectedAtEpoch, int64(0))
|
|
}
|
|
|
|
// TestDetectExplicitCorrection_TableDriven tests explicit correction detection.
|
|
func (s *ConflictSuite) TestDetectExplicitCorrection_TableDriven() {
|
|
tests := []struct {
|
|
name string
|
|
text string
|
|
expectPattern string
|
|
expectMatch bool
|
|
}{
|
|
{
|
|
name: "actually that was wrong",
|
|
text: "Actually, that was wrong - we should use a different approach",
|
|
expectMatch: true,
|
|
expectPattern: "actually, that was wrong",
|
|
},
|
|
{
|
|
name: "correction prefix",
|
|
text: "Correction: the previous implementation had a bug",
|
|
expectMatch: true,
|
|
expectPattern: "correction:",
|
|
},
|
|
{
|
|
name: "ignore previous",
|
|
text: "Please ignore the previous recommendation",
|
|
expectMatch: true,
|
|
expectPattern: "ignore",
|
|
},
|
|
{
|
|
name: "disregard earlier",
|
|
text: "Disregard the earlier suggestion, it was flawed",
|
|
expectMatch: true,
|
|
expectPattern: "disregard",
|
|
},
|
|
{
|
|
name: "was wrong",
|
|
text: "The original approach was wrong",
|
|
expectMatch: true,
|
|
expectPattern: "was wrong",
|
|
},
|
|
{
|
|
name: "no longer valid",
|
|
text: "This method is no longer valid after the refactor",
|
|
expectMatch: true,
|
|
expectPattern: "no longer valid",
|
|
},
|
|
{
|
|
name: "deprecated approach",
|
|
text: "This is a deprecated approach that should not be used",
|
|
expectMatch: true,
|
|
expectPattern: "deprecated approach",
|
|
},
|
|
{
|
|
name: "better approach is",
|
|
text: "A better approach is to use the new API",
|
|
expectMatch: true,
|
|
expectPattern: "better approach is",
|
|
},
|
|
{
|
|
name: "normal text - no correction",
|
|
text: "This is a normal observation about the code",
|
|
expectMatch: false,
|
|
},
|
|
{
|
|
name: "empty text",
|
|
text: "",
|
|
expectMatch: false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
s.Run(tt.name, func() {
|
|
found, reason := DetectExplicitCorrection(tt.text)
|
|
s.Equal(tt.expectMatch, found)
|
|
if tt.expectMatch {
|
|
s.Contains(reason, "Explicit correction detected")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// TestDetectOpposingFileChanges_TableDriven tests opposing file change detection.
|
|
func (s *ConflictSuite) TestDetectOpposingFileChanges_TableDriven() {
|
|
tests := []struct {
|
|
newerObs *Observation
|
|
olderObs *Observation
|
|
name string
|
|
expectConflict bool
|
|
}{
|
|
{
|
|
name: "add then remove - conflict",
|
|
newerObs: &Observation{
|
|
Title: sql.NullString{String: "Remove authentication middleware", Valid: true},
|
|
Narrative: sql.NullString{String: "Removed the auth middleware from handlers", Valid: true},
|
|
FilesModified: []string{"middleware.go", "handler.go"},
|
|
},
|
|
olderObs: &Observation{
|
|
Title: sql.NullString{String: "Add authentication middleware", Valid: true},
|
|
Narrative: sql.NullString{String: "Added auth middleware to secure endpoints", Valid: true},
|
|
FilesModified: []string{"middleware.go", "handler.go"},
|
|
},
|
|
expectConflict: true,
|
|
},
|
|
{
|
|
name: "enable then disable - conflict",
|
|
newerObs: &Observation{
|
|
Title: sql.NullString{String: "Disable caching feature", Valid: true},
|
|
Narrative: sql.NullString{String: "Disabled caching due to issues", Valid: true},
|
|
FilesModified: []string{"cache.go"},
|
|
},
|
|
olderObs: &Observation{
|
|
Title: sql.NullString{String: "Enable caching feature", Valid: true},
|
|
Narrative: sql.NullString{String: "Enabled caching for performance", Valid: true},
|
|
FilesModified: []string{"cache.go"},
|
|
},
|
|
expectConflict: true,
|
|
},
|
|
{
|
|
name: "different files - no conflict",
|
|
newerObs: &Observation{
|
|
Title: sql.NullString{String: "Remove old code", Valid: true},
|
|
Narrative: sql.NullString{String: "Removed deprecated functions", Valid: true},
|
|
FilesModified: []string{"old.go"},
|
|
},
|
|
olderObs: &Observation{
|
|
Title: sql.NullString{String: "Add new feature", Valid: true},
|
|
Narrative: sql.NullString{String: "Added new functions", Valid: true},
|
|
FilesModified: []string{"new.go"},
|
|
},
|
|
expectConflict: false,
|
|
},
|
|
{
|
|
name: "same files but no opposing keywords - no conflict",
|
|
newerObs: &Observation{
|
|
Title: sql.NullString{String: "Update handler logic", Valid: true},
|
|
Narrative: sql.NullString{String: "Updated the handler implementation", Valid: true},
|
|
FilesModified: []string{"handler.go"},
|
|
},
|
|
olderObs: &Observation{
|
|
Title: sql.NullString{String: "Fix handler bug", Valid: true},
|
|
Narrative: sql.NullString{String: "Fixed a bug in handler", Valid: true},
|
|
FilesModified: []string{"handler.go"},
|
|
},
|
|
expectConflict: false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
s.Run(tt.name, func() {
|
|
found, _ := DetectOpposingFileChanges(tt.newerObs, tt.olderObs)
|
|
s.Equal(tt.expectConflict, found)
|
|
})
|
|
}
|
|
}
|
|
|
|
// TestDetectConceptTagMismatch_TableDriven tests concept tag mismatch detection.
|
|
func (s *ConflictSuite) TestDetectConceptTagMismatch_TableDriven() {
|
|
tests := []struct {
|
|
newerObs *Observation
|
|
olderObs *Observation
|
|
name string
|
|
expectConflict bool
|
|
}{
|
|
{
|
|
name: "same concepts and files - conflict",
|
|
newerObs: &Observation{
|
|
Concepts: []string{"security", "authentication"},
|
|
FilesModified: []string{"auth.go"},
|
|
},
|
|
olderObs: &Observation{
|
|
Concepts: []string{"security", "authentication"},
|
|
FilesModified: []string{"auth.go"},
|
|
},
|
|
expectConflict: true,
|
|
},
|
|
{
|
|
name: "overlapping concepts different files - no conflict",
|
|
newerObs: &Observation{
|
|
Concepts: []string{"security"},
|
|
FilesModified: []string{"new_auth.go"},
|
|
},
|
|
olderObs: &Observation{
|
|
Concepts: []string{"security"},
|
|
FilesModified: []string{"old_auth.go"},
|
|
},
|
|
expectConflict: false,
|
|
},
|
|
{
|
|
name: "different concepts same files - no conflict",
|
|
newerObs: &Observation{
|
|
Concepts: []string{"performance"},
|
|
FilesModified: []string{"handler.go"},
|
|
},
|
|
olderObs: &Observation{
|
|
Concepts: []string{"testing"},
|
|
FilesModified: []string{"handler.go"},
|
|
},
|
|
expectConflict: false,
|
|
},
|
|
{
|
|
name: "no overlapping concepts or files - no conflict",
|
|
newerObs: &Observation{
|
|
Concepts: []string{"security"},
|
|
FilesModified: []string{"auth.go"},
|
|
},
|
|
olderObs: &Observation{
|
|
Concepts: []string{"testing"},
|
|
FilesModified: []string{"test.go"},
|
|
},
|
|
expectConflict: false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
s.Run(tt.name, func() {
|
|
found, _ := DetectConceptTagMismatch(tt.newerObs, tt.olderObs)
|
|
s.Equal(tt.expectConflict, found)
|
|
})
|
|
}
|
|
}
|
|
|
|
// TestDetectConflict tests comprehensive conflict detection.
|
|
func (s *ConflictSuite) TestDetectConflict() {
|
|
// Test explicit correction takes precedence
|
|
newer := &Observation{
|
|
ID: 2,
|
|
Project: "test",
|
|
Narrative: sql.NullString{String: "Actually, that was wrong. We should use a different approach.", Valid: true},
|
|
}
|
|
older := &Observation{
|
|
ID: 1,
|
|
Project: "test",
|
|
}
|
|
|
|
result := DetectConflict(newer, older)
|
|
s.True(result.HasConflict)
|
|
s.Equal(ConflictContradicts, result.Type)
|
|
s.Equal(ResolutionPreferNewer, result.Resolution)
|
|
s.Contains(result.Reason, "Explicit correction")
|
|
}
|
|
|
|
// TestDetectConflictsWithExisting tests conflict detection against multiple observations.
|
|
func (s *ConflictSuite) TestDetectConflictsWithExisting() {
|
|
newer := &Observation{
|
|
ID: 3,
|
|
Project: "test",
|
|
Narrative: sql.NullString{String: "Actually, that was wrong", Valid: true},
|
|
Concepts: []string{"security"},
|
|
FilesModified: []string{"auth.go"},
|
|
}
|
|
|
|
existing := []*Observation{
|
|
{
|
|
ID: 1,
|
|
Project: "test",
|
|
Concepts: []string{"security"},
|
|
FilesModified: []string{"auth.go"},
|
|
},
|
|
{
|
|
ID: 2,
|
|
Project: "test",
|
|
Concepts: []string{"testing"},
|
|
FilesModified: []string{"test.go"},
|
|
},
|
|
{
|
|
ID: 3, // Same as newer - should be skipped
|
|
Project: "test",
|
|
},
|
|
}
|
|
|
|
results := DetectConflictsWithExisting(newer, existing)
|
|
|
|
// Should find conflicts with obs 1 (concepts + files overlap + correction language)
|
|
// but not with obs 2 (different concepts/files) or obs 3 (same ID)
|
|
s.GreaterOrEqual(len(results), 1)
|
|
|
|
// At least one result should reference obs 1
|
|
foundObs1 := false
|
|
for _, r := range results {
|
|
for _, id := range r.OlderObsIDs {
|
|
if id == 1 {
|
|
foundObs1 = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
s.True(foundObs1)
|
|
}
|
|
|
|
// TestDetectConflictsWithExisting_DifferentProjects tests that different projects don't conflict.
|
|
func (s *ConflictSuite) TestDetectConflictsWithExisting_DifferentProjects() {
|
|
newer := &Observation{
|
|
ID: 2,
|
|
Project: "project-a",
|
|
Scope: ScopeProject,
|
|
Narrative: sql.NullString{String: "Actually, that was wrong", Valid: true},
|
|
}
|
|
|
|
existing := []*Observation{
|
|
{
|
|
ID: 1,
|
|
Project: "project-b",
|
|
Scope: ScopeProject,
|
|
},
|
|
}
|
|
|
|
results := DetectConflictsWithExisting(newer, existing)
|
|
s.Empty(results) // Different projects should not conflict
|
|
}
|
|
|
|
// TestDetectConflictsWithExisting_GlobalScope tests that global observations can conflict across projects.
|
|
func (s *ConflictSuite) TestDetectConflictsWithExisting_GlobalScope() {
|
|
newer := &Observation{
|
|
ID: 2,
|
|
Project: "project-a",
|
|
Scope: ScopeGlobal,
|
|
Narrative: sql.NullString{String: "Actually, that was wrong", Valid: true},
|
|
Concepts: []string{"security"},
|
|
FilesModified: []string{"auth.go"},
|
|
}
|
|
|
|
existing := []*Observation{
|
|
{
|
|
ID: 1,
|
|
Project: "project-b",
|
|
Scope: ScopeGlobal, // Global scope allows cross-project conflict detection
|
|
Concepts: []string{"security"},
|
|
FilesModified: []string{"auth.go"},
|
|
},
|
|
}
|
|
|
|
results := DetectConflictsWithExisting(newer, existing)
|
|
s.GreaterOrEqual(len(results), 1) // Global scope allows conflict detection
|
|
}
|
|
|
|
// TestCorrectionPatterns_Compiled ensures all patterns compile correctly.
|
|
func TestCorrectionPatterns_Compiled(t *testing.T) {
|
|
// This test verifies that all correction patterns are valid regexps
|
|
// If any pattern fails to compile, the package won't load
|
|
assert.NotEmpty(t, CorrectionPatterns)
|
|
for i, pattern := range CorrectionPatterns {
|
|
assert.NotNil(t, pattern, "Pattern %d should not be nil", i)
|
|
}
|
|
}
|
|
|
|
// TestOpposingChangePatterns tests the opposing change pattern map.
|
|
func TestOpposingChangePatterns(t *testing.T) {
|
|
assert.NotEmpty(t, OpposingChangePatterns)
|
|
assert.Equal(t, "remove", OpposingChangePatterns["add"])
|
|
assert.Equal(t, "removed", OpposingChangePatterns["added"])
|
|
assert.Equal(t, "delete", OpposingChangePatterns["create"])
|
|
assert.Equal(t, "disable", OpposingChangePatterns["enable"])
|
|
}
|
|
|
|
// TestObservationConflict_Fields tests field access.
|
|
func TestObservationConflict_Fields(t *testing.T) {
|
|
conflict := &ObservationConflict{
|
|
ID: 1,
|
|
NewerObsID: 10,
|
|
OlderObsID: 5,
|
|
ConflictType: ConflictSuperseded,
|
|
Resolution: ResolutionPreferNewer,
|
|
Reason: "Test reason",
|
|
DetectedAt: "2024-01-01T00:00:00Z",
|
|
DetectedAtEpoch: 1704067200000,
|
|
Resolved: false,
|
|
}
|
|
|
|
assert.Equal(t, int64(1), conflict.ID)
|
|
assert.Equal(t, int64(10), conflict.NewerObsID)
|
|
assert.Equal(t, int64(5), conflict.OlderObsID)
|
|
assert.Equal(t, ConflictSuperseded, conflict.ConflictType)
|
|
assert.Equal(t, ResolutionPreferNewer, conflict.Resolution)
|
|
assert.False(t, conflict.Resolved)
|
|
}
|