Commit 01571b1
Changed files (1)
pkg
memory
pkg/memory/server.go
@@ -16,6 +16,7 @@ type Server struct {
memoryFile string
graph *KnowledgeGraph
mu sync.RWMutex
+ loaded bool
}
// KnowledgeGraph represents the in-memory knowledge graph
@@ -67,9 +68,8 @@ func (ms *Server) ensureGraphLoaded() error {
defer ms.mu.Unlock()
// Check if graph is already loaded
- if ms.graph.Entities == nil {
- ms.graph.Entities = make(map[string]*Entity)
- ms.graph.Relations = make(map[string]Relation)
+ if !ms.loaded {
+ ms.loaded = true
// Load from file if it exists
return ms.loadGraphInternal()