Wired tools to agent

2 files changed, 6 insertions, 2 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))
 
diff --git a/tools.py b/tools.py
index 54c837e..78299b8 100644
--- a/tools.py
+++ b/tools.py
@@ -213,7 +213,7 @@ class MemoryRemoveFactTool(Tool):
         if not isinstance(fact, str):
             raise ToolError("fact must be a string")
 
-        self.memory.add_fact(fact)
+        self.memory.remove_fact(fact)
 
         return ToolResult(content=f"Removed fact: {fact}")
 
@@ -249,7 +249,7 @@ class MemoryRemoveNoteTool(Tool):
         if not isinstance(note, str):
             raise ToolError("note must be a string")
 
-        self.memory.add_note(note)
+        self.memory.remove_note(note)
 
         return ToolResult(content=f"Removed note: {note}")