Added more memory management

1 files changed, 2 insertions, 0 deletions

diff --git a/identity.py b/identity.py
index 312e3d7..ec1cc19 100644
--- a/identity.py
+++ b/identity.py
@@ -8,6 +8,7 @@ class AgentIdentity:
     personality: dict[str, str]
     values: dict[str, bool]
     quirks: dict[str, str]
+    behavior: dict[str, str]
 
 
 @dataclass
@@ -25,6 +26,7 @@ def load_agent_identity(path: str) -> AgentIdentity:
         personality=data["personality"],
         values=data["values"],
         quirks=data["quirks"],
+        behavior=data["behavior"],
     )