exp_pool add enabled

metagpt/configs/exp_pool_config.py

@@ -4,6 +4,10 @@ from metagpt.utils.yaml_model import YamlModel
 
 
 class ExperiencePoolConfig(YamlModel):
+    enabled: bool = Field(
+        default=False,
+        description="Flag to enable or disable the experience pool. When disabled, both reading and writing are ineffective.",
+    )
     enable_read: bool = Field(default=False, description="Enable to read from experience pool.")
     enable_write: bool = Field(default=False, description="Enable to write to experience pool.")
     persist_path: str = Field(default=".chroma_exp_data", description="The persist path for experience pool.")

metagpt/exp_pool/decorator.py

@@ -50,7 +50,7 @@ def exp_cache(
     """
 
     def decorator(func: Callable[..., ReturnType]) -> Callable[..., ReturnType]:
-        if not config.exp_pool.enable_read:
+        if not config.exp_pool.enabled:
             return func
 
         @functools.wraps(func)

metagpt/exp_pool/manager.py

@@ -74,7 +74,7 @@ class ExperienceManager(BaseModel):
             exp (Experience): The experience to add.
         """
 
-        if not self.config.exp_pool.enable_write:
+        if not self.config.exp_pool.enabled or not self.config.exp_pool.enable_write:
             return
 
         self.storage.add_objs([exp])
@@ -92,7 +92,7 @@ class ExperienceManager(BaseModel):
             list[Experience]: A list of experiences that match the args.
         """
 
-        if not self.config.exp_pool.enable_read:
+        if not self.config.exp_pool.enabled or not self.config.exp_pool.enable_read:
             return []
 
         nodes = await self.storage.aretrieve(req)

metagpt/exp_pool/schema.py

@@ -1,5 +1,5 @@
 """Experience schema."""
-
+import time
 from enum import Enum
 from typing import Optional
 
@@ -67,6 +67,7 @@ class Experience(BaseModel):
     entry_type: EntryType = Field(default=EntryType.AUTOMATIC, description="Type of entry: Manual or Automatic.")
     tag: str = Field(default="", description="Tagging experience.")
     traj: Optional[Trajectory] = Field(default=None, description="Trajectory.")
+    timestamp: Optional[float] = Field(default_factory=time.time)
 
     def rag_key(self):
         return self.req