From d9ab81bbae03dd8ea166b0dda535b0f3ec655434 Mon Sep 17 00:00:00 2001
From: seehi <6580@pm.me>
Date: Thu, 12 Sep 2024 11:21:59 +0800
Subject: [PATCH] add `enable_longterm_memory` config

---
 metagpt/roles/di/role_zero.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/metagpt/roles/di/role_zero.py b/metagpt/roles/di/role_zero.py
index 586e5345f..9b68378cb 100644
--- a/metagpt/roles/di/role_zero.py
+++ b/metagpt/roles/di/role_zero.py
@@ -173,8 +173,7 @@ class RoleZero(Role):
         The role name will be used as the collection name.
         """
 
-        enable_longterm_memory = bool(self.config.role_zero and self.config.role_zero.enable_longterm_memory)
-        if enable_longterm_memory:
+        if self.config.role_zero and self.config.role_zero.enable_longterm_memory:
             self.rc.memory = RoleZeroLongTermMemory(collection_name=self.name.replace(" ", ""), memory_k=self.memory_k)
             logger.info(f"Long-term memory set for role '{self.name}'")