From 606e264c9c6ba99ae0c04a8463d37d85568f0c86 Mon Sep 17 00:00:00 2001
From: batsumaru <>
Date: Mon, 14 Sep 2026 17:33:50 +0900
Subject: Wired tools to agent

---
 agent.py | 4 ++++
 tools.py | 4 ++--
 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}")
 
-- 
cgit v1.3