Added more memory management

1 files changed, 6 insertions, 0 deletions

diff --git a/context.py b/context.py
index d702706..669e237 100644
--- a/context.py
+++ b/context.py
@@ -25,6 +25,9 @@ class ContextBuilder:
             f"Directness: {agent.values['directness']}.\n"
             f"Favorite color: {agent.quirks['favorite_color']}.\n"
             f"\n"
+            f"Task management:\n"
+            f"{agent.behavior['task_management']}"
+            f"\n"
             f"The user's name is {user.name}"
             f"\n"
             f"{self.build_working_memory(memory)}"
@@ -36,6 +39,9 @@ class ContextBuilder:
         if memory.objective:
             lines.append(f"Objective: {memory.objective}")
 
+        if memory.status:
+            lines.append(f"Status: {memory.status.value}")
+
         if memory.facts:
             lines.append("Facts:")
             lines.extend(f"- {fact}" for fact in memory.facts)