Wired tools to agent

1 files changed, 4 insertions, 0 deletions

diff --git a/agent.py b/agent.py
index 886cae0..a79fd74 100644
--- a/agent.py
+++ b/agent.py
@@ -7,6 +7,8 @@ from memory import WorkingMemory
 from tools import (
     MemoryAddFactTool,
     MemoryAddNoteTool,
+    MemoryRemoveFactTool,
+    MemoryRemoveNoteTool,
     TaskStartTool,
     ToolError,
     ToolRegistry,
@@ -32,6 +34,8 @@ class Agent:
         self.working_memory = WorkingMemory()
 
         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))