Wired long-term memory into agent

1 files changed, 9 insertions, 0 deletions

diff --git a/personal_agent/agent.py b/personal_agent/agent.py
index 73cc422..9c456a8 100644
--- a/personal_agent/agent.py
+++ b/personal_agent/agent.py
@@ -4,7 +4,11 @@ from personal_agent.context import ContextBuilder
 from personal_agent.identity import AgentIdentity, UserProfile
 from personal_agent.llm import LLM
 from personal_agent.memory.models import WorkingMemory
+from personal_agent.memory.store import LongTermMemoryStore
 from personal_agent.tools import (
+    LongTermMemoryDeleteTool,
+    LongTermMemorySaveTool,
+    LongTermMemorySearchTool,
     MemoryAddFactTool,
     MemoryAddNoteTool,
     MemoryRemoveFactTool,
@@ -22,6 +26,7 @@ class Agent:
         agent_identity: AgentIdentity,
         user_profile: UserProfile,
         tools: ToolRegistry,
+        long_term_memory_store: LongTermMemoryStore,
     ) -> None:
         self.llm = llm
         self.tools = tools
@@ -32,12 +37,16 @@ class Agent:
         )
 
         self.working_memory = WorkingMemory()
+        self.long_term_memory_store = long_term_memory_store
 
         self.tools.register(MemoryAddFactTool(self.working_memory))
         self.tools.register(MemoryRemoveFactTool(self.working_memory))
         self.tools.register(MemoryRemoveNoteTool(self.working_memory))
         self.tools.register(MemoryAddNoteTool(self.working_memory))
         self.tools.register(TaskStartTool(self.working_memory))
+        self.tools.register(LongTermMemorySaveTool(self.long_term_memory_store))
+        self.tools.register(LongTermMemorySearchTool(self.long_term_memory_store))
+        self.tools.register(LongTermMemoryDeleteTool(self.long_term_memory_store))
 
         self.messages: list[dict[str, object]] = []