diff options
Diffstat (limited to 'personal_agent/agent.py')
| -rw-r--r-- | personal_agent/agent.py | 9 |
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]] = [] |