From cb01e42645eeb0580db82ee1c1d875e2fe14fd78 Mon Sep 17 00:00:00 2001
From: geekan <geekan@foxmail.com>
Date: Mon, 8 Jan 2024 20:02:37 +0800
Subject: [PATCH] fix memory

---
 tests/metagpt/memory/test_longterm_memory.py | 8 +++-----
 tests/metagpt/memory/test_memory.py          | 2 +-
 tests/metagpt/memory/test_memory_storage.py  | 4 ++--
 3 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/tests/metagpt/memory/test_longterm_memory.py b/tests/metagpt/memory/test_longterm_memory.py
index 0f7a4fac4..a9ef56bad 100644
--- a/tests/metagpt/memory/test_longterm_memory.py
+++ b/tests/metagpt/memory/test_longterm_memory.py
@@ -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
 
diff --git a/tests/metagpt/memory/test_memory.py b/tests/metagpt/memory/test_memory.py
index 36d7ad488..a072b61de 100644
--- a/tests/metagpt/memory/test_memory.py
+++ b/tests/metagpt/memory/test_memory.py
@@ -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")
diff --git a/tests/metagpt/memory/test_memory_storage.py b/tests/metagpt/memory/test_memory_storage.py
index 0eb1069d5..e82a82fc8 100644
--- a/tests/metagpt/memory/test_memory_storage.py
+++ b/tests/metagpt/memory/test_memory_storage.py
@@ -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():