From c4eb028a8303a2dfb9fbb8018d751e5343c01d91 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= <mashenquan@fuzhi.ai>
Date: Sat, 4 Nov 2023 14:26:48 +0800
Subject: [PATCH] refactor: save -> dump

---
 metagpt/environment.py       |  4 ++--
 metagpt/schema.py            | 14 +++++++-------
 tests/metagpt/test_schema.py |  4 ++--
 3 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/metagpt/environment.py b/metagpt/environment.py
index 0fa330a83..a7e6322ff 100644
--- a/metagpt/environment.py
+++ b/metagpt/environment.py
@@ -58,7 +58,7 @@ class Environment(BaseModel):
         route the message to the message recipient is a problem addressed by the transport framework designed
         in RFC 113.
         """
-        logger.info(f"publish_message: {message.save()}")
+        logger.info(f"publish_message: {message.dump()}")
         found = False
         # According to the routing feature plan in Chapter 2.2.3.2 of RFC 113
         for obj, subscribed_tags in self.consumers.items():
@@ -66,7 +66,7 @@ class Environment(BaseModel):
                 obj.put_message(message)
                 found = True
         if not found:
-            logger.warning(f"Message no recipients: {message.save()}")
+            logger.warning(f"Message no recipients: {message.dump()}")
 
         return True
 
diff --git a/metagpt/schema.py b/metagpt/schema.py
index 34e6fa07b..bb8d8b42c 100644
--- a/metagpt/schema.py
+++ b/metagpt/schema.py
@@ -188,7 +188,7 @@ class Message(BaseModel):
         """Return a dict containing `role` and `content` for the LLM call.l"""
         return {"role": self.role, "content": self.content}
 
-    def save(self) -> str:
+    def dump(self) -> str:
         """Convert the object to json string"""
         return self.json(exclude_none=True)
 
@@ -264,7 +264,7 @@ class MessageQueue:
         """Return true if the queue is empty."""
         return self._queue.empty()
 
-    async def save(self) -> str:
+    async def dump(self) -> str:
         """Convert the `MessageQueue` object to a json string."""
         if self.empty():
             return "[]"
@@ -299,7 +299,7 @@ class MessageQueue:
 if __name__ == "__main__":
     m = Message("a", role="v1")
     m.set_role("v2")
-    v = m.save()
+    v = m.dump()
     m = Message.load(v)
 
     test_content = "test_message"
@@ -312,9 +312,9 @@ if __name__ == "__main__":
     logger.info(msgs)
 
     jsons = [
-        UserMessage(test_content).save(),
-        SystemMessage(test_content).save(),
-        AIMessage(test_content).save(),
-        Message(test_content, role="QA").save(),
+        UserMessage(test_content).dump(),
+        SystemMessage(test_content).dump(),
+        AIMessage(test_content).dump(),
+        Message(test_content, role="QA").dump(),
     ]
     logger.info(jsons)
diff --git a/tests/metagpt/test_schema.py b/tests/metagpt/test_schema.py
index 2fa76fcad..21ba3fd14 100644
--- a/tests/metagpt/test_schema.py
+++ b/tests/metagpt/test_schema.py
@@ -31,13 +31,13 @@ def test_messages():
 @pytest.mark.asyncio
 def test_message():
     m = Message("a", role="v1")
-    v = m.save()
+    v = m.dump()
     d = json.loads(v)
     assert d
     assert d.get("content") == "a"
     assert d.get("meta_info") == {"role": "v1"}
     m.set_role("v2")
-    v = m.save()
+    v = m.dump()
     assert v
     m = Message.load(v)
     assert m.content == "a"