Added more memory management

1 files changed, 8 insertions, 3 deletions

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(
             {