From a3e492d29c4ebb59a11dd6d6ffd1be04dca6834e Mon Sep 17 00:00:00 2001
From: batsumaru <>
Date: Mon, 14 Sep 2026 17:21:33 +0900
Subject: Added more memory management

---
 agent.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'agent.py')

diff --git a/agent.py b/agent.py
index 03abfe9..886cae0 100644
--- a/agent.py
+++ b/agent.py
@@ -4,7 +4,13 @@ from context import ContextBuilder
 from identity import AgentIdentity, UserProfile
 from llm import LLM
 from memory import WorkingMemory
-from tools import MemoryAddFactTool, MemoryAddNoteTool, ToolError, ToolRegistry
+from tools import (
+    MemoryAddFactTool,
+    MemoryAddNoteTool,
+    TaskStartTool,
+    ToolError,
+    ToolRegistry,
+)
 
 
 class Agent:
@@ -27,12 +33,11 @@ class Agent:
 
         self.tools.register(MemoryAddFactTool(self.working_memory))
         self.tools.register(MemoryAddNoteTool(self.working_memory))
+        self.tools.register(TaskStartTool(self.working_memory))
 
         self.messages: list[dict[str, object]] = []
 
     def chat(self, user_input: str) -> str:
-        if self.working_memory.objective is None:
-            self.working_memory.start_task(user_input)
         print(f"\n[memory: {self.working_memory}]")
         self.messages.append(
             {
-- 
cgit v1.3