Wired long-term memory into agent

1 files changed, 10 insertions, 0 deletions

diff --git a/personal_agent/cli.py b/personal_agent/cli.py
index 7bd747a..57df7b0 100644
--- a/personal_agent/cli.py
+++ b/personal_agent/cli.py
@@ -1,6 +1,10 @@
+from pathlib import Path
+
 from personal_agent.agent import Agent
+from personal_agent.database.database import Database
 from personal_agent.identity import load_agent_identity, load_user_profile
 from personal_agent.llm import LLM
+from personal_agent.memory.store import LongTermMemoryStore
 from personal_agent.tools import (
     CalculatorTool,
     FileSystemListTool,
@@ -23,11 +27,17 @@ tools.register(WeatherTool())
 tools.register(FileSystemListTool())
 tools.register(FileSystemReadTool())
 
+database = Database(Path("data/agent.db"))
+database.migrate()
+
+long_term_memory_store = LongTermMemoryStore(database)
+
 agent = Agent(
     llm,
     agent_identity,
     user_profile,
     tools,
+    long_term_memory_store,
 )
 
 while True: