Cleanup mypy errors from tests

camel-ai · Jul 3, 2023 · be1d700 · be1d700
1 parent 04f4496
commit be1d700
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/camel/messages/base.py b/camel/messages/base.py
@@ -221,7 +221,7 @@ def to_openai_assistant_message(self) -> OpenAIAssistantMessage:
         """
         return {"role": "assistant", "content": self.content}
 
-    def to_dict(self) -> Dict:
+    def to_dict(self) -> Dict[str, Any]:
         r"""Converts the message to a dictionary.
 
         Returns:

diff --git a/test/messages/test_chat_message.py b/test/messages/test_chat_message.py
@@ -11,6 +11,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
+from typing import Any, Dict
+
 import pytest
 
 from camel.messages import BaseMessage
@@ -59,12 +61,12 @@ def test_chat_message(chat_message: BaseMessage) -> None:
     assert chat_message.content == content
 
     dictionary = chat_message.to_dict()
-    assert dictionary == {
+    reference_dict: Dict[str, Any] = {
         "role_name": role_name,
         "role_type": role_type.name,
-        **(meta_dict or {}),
         "content": content,
     }
+    assert dictionary == reference_dict
 
 
 def test_assistant_chat_message(assistant_chat_message: BaseMessage) -> None:
@@ -79,12 +81,12 @@ def test_assistant_chat_message(assistant_chat_message: BaseMessage) -> None:
     assert assistant_chat_message.content == content
 
     dictionary = assistant_chat_message.to_dict()
-    assert dictionary == {
+    reference_dict: Dict[str, Any] = {
         "role_name": role_name,
         "role_type": role_type.name,
-        **(meta_dict or {}),
         "content": content,
     }
+    assert dictionary == reference_dict
 
 
 def test_user_chat_message(user_chat_message: BaseMessage) -> None:
@@ -99,9 +101,9 @@ def test_user_chat_message(user_chat_message: BaseMessage) -> None:
     assert user_chat_message.content == content
 
     dictionary = user_chat_message.to_dict()
-    assert dictionary == {
+    reference_dict: Dict[str, Any] = {
         "role_name": role_name,
         "role_type": role_type.name,
-        **(meta_dict or {}),
         "content": content,
     }
+    assert dictionary == reference_dict