apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-29 10:56:22 +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

4
tests/metagpt/memory/test_longterm_memory.py
View file

@ -20,6 +20,10 @@ def test_ltm_search():
assert len(CONFIG.openai_api_key) > 20
role_id = "UTUserLtm(Product Manager)"
from metagpt.environment import Environment
Environment
RoleContext.model_rebuild()
rc = RoleContext(watch={"metagpt.actions.add_requirement.UserRequirement"})
ltm = LongTermMemory()
ltm.recover_memory(role_id, rc)

4
tests/metagpt/memory/test_memory_storage.py
View file

@ -24,7 +24,7 @@ def test_idea_message():
role_id = "UTUser1(Product Manager)"
message = Message(role="User", content=idea, cause_by=UserRequirement)
shutil.rmtree(Path(DATA_PATH / f"role_mem/{role_id}/"))
shutil.rmtree(Path(DATA_PATH / f"role_mem/{role_id}/"), ignore_errors=True)
memory_storage: MemoryStorage = MemoryStorage()
messages = memory_storage.recover_memory(role_id)
@ -58,7 +58,7 @@ def test_actionout_message():
content=content, instruct_content=ic_obj(**out_data), role="user", cause_by=WritePRD
) # WritePRD as test action
shutil.rmtree(Path(DATA_PATH / f"role_mem/{role_id}/"))
shutil.rmtree(Path(DATA_PATH / f"role_mem/{role_id}/"), ignore_errors=True)
memory_storage: MemoryStorage = MemoryStorage()
messages = memory_storage.recover_memory(role_id)