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

fix memory

Browse source
This commit is contained in:
geekan 2024-01-08 20:02:37 +08:00
parent 3677d44b47
commit cb01e42645
3 changed files with 6 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -10,17 +10,15 @@ import os
import pytest
from metagpt.actions import UserRequirement
from metagpt.config import CONFIG
from metagpt.config2 import config
from metagpt.memory.longterm_memory import LongTermMemory
from metagpt.roles.role import RoleContext
from metagpt.schema import Message
os.environ.setdefault("OPENAI_API_KEY", config.get_openai_llm().api_key)
def test_ltm_search():
assert hasattr(CONFIG, "long_term_memory") is True
os.environ.setdefault("OPENAI_API_KEY", CONFIG.openai_api_key)
assert len(CONFIG.openai_api_key) > 20
role_id = "UTUserLtm(Product Manager)"
from metagpt.environment import Environment

2
tests/metagpt/memory/test_memory.py
View file

@ -32,7 +32,7 @@ def test_memory():
messages = memory.get_by_action(UserRequirement)
assert len(messages) == 2
messages = memory.get_by_actions([UserRequirement])
messages = memory.get_by_actions({UserRequirement})
assert len(messages) == 2
messages = memory.try_remember("test message")

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

@ -11,12 +11,12 @@ from typing import List
from metagpt.actions import UserRequirement, WritePRD
from metagpt.actions.action_node import ActionNode
from metagpt.config import CONFIG
from metagpt.config2 import config
from metagpt.const import DATA_PATH
from metagpt.memory.memory_storage import MemoryStorage
from metagpt.schema import Message
os.environ.setdefault("OPENAI_API_KEY", CONFIG.openai_api_key)
os.environ.setdefault("OPENAI_API_KEY", config.get_openai_llm().api_key)
def test_idea_message():