apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-15 11:02:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix memory unittest

Browse source
This commit is contained in:
better629 2023-12-29 03:10:23 +08:00
parent c8e351f3c8
commit edce4ac47a
3 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
metagpt/memory/longterm_memory.py
View file

@ -12,6 +12,7 @@ from pydantic import ConfigDict, Field
from metagpt.logs import logger
from metagpt.memory import Memory
from metagpt.memory.memory_storage import MemoryStorage
from metagpt.roles.role import RoleContext
from metagpt.schema import Message
@ -25,10 +26,10 @@ class LongTermMemory(Memory):
model_config = ConfigDict(arbitrary_types_allowed=True)
memory_storage: MemoryStorage = Field(default_factory=MemoryStorage)
rc: Optional["RoleContext"] = None
rc: Optional[RoleContext] = None
msg_from_recover: bool = False
def recover_memory(self, role_id: str, rc: "RoleContext"):
def recover_memory(self, role_id: str, rc: RoleContext):
messages = self.memory_storage.recover_memory(role_id)
self.rc = rc
if not self.memory_storage.is_initialized: