fix bug in test

metagpt/memory/brain_memory.py

@@ -33,6 +33,9 @@ class BrainMemory(BaseModel):
     cacheable: bool = True
     llm: Optional[BaseLLM] = None
 
+    class Config:
+        arbitrary_types_allowed = True
+
     def add_talk(self, msg: Message):
         """
         Add message from user.

tests/metagpt/document_store/test_lancedb_store.py

@@ -7,12 +7,9 @@
 """
 import random
 
-import pytest
-
 from metagpt.document_store.lancedb_store import LanceStore
 
 
-@pytest
 def test_lance_store():
     # This simply establishes the connection to the database, so we can drop the table if it exists
     store = LanceStore("test")

tests/metagpt/test_message.py

@@ -8,7 +8,7 @@
 """
 import pytest
 
-from metagpt.schema import AIMessage, Message, RawMessage, SystemMessage, UserMessage
+from metagpt.schema import AIMessage, Message, SystemMessage, UserMessage
 
 
 def test_message():
@@ -29,13 +29,5 @@ def test_all_messages():
         assert msg.content == test_content
 
 
-def test_raw_message():
-    msg = RawMessage(role="user", content="raw")
-    assert msg["role"] == "user"
-    assert msg["content"] == "raw"
-    with pytest.raises(KeyError):
-        assert msg["1"] == 1, "KeyError: '1'"
-
-
 if __name__ == "__main__":
     pytest.main([__file__, "-s"])