fix env=None when init Team with env=xxx

metagpt/environment.py

@@ -57,6 +57,7 @@ class Environment(BaseModel):
     @model_validator(mode="after")
     def init_roles(self):
         self.add_roles(self.roles.values())
+        return self
 
     def serialize(self, stg_path: Path):
         roles_path = stg_path.joinpath("roles.json")

metagpt/schema.py

@@ -195,7 +195,7 @@ class Message(BaseModel):
 
     def dump(self) -> str:
         """Convert the object to json string"""
-        return self.model_dump_json(exclude_none=True)
+        return self.model_dump_json(exclude_none=True, warnings=False)
 
     @staticmethod
     @handle_exception(exception_type=JSONDecodeError, default_return=None)
@@ -250,15 +250,6 @@ class MessageQueue(BaseModel):
 
     _queue: Queue = PrivateAttr(default_factory=Queue)
 
-    # _private_attributes = {"_queue": Queue()}
-
-    # def __init__(self, **kwargs: Any):
-    #     for key in self._private_attributes.keys():
-    #         if key in kwargs:
-    #             object.__setattr__(self, key, kwargs[key])
-    #         else:
-    #             object.__setattr__(self, key, Queue())
-
     def pop(self) -> Message | None:
         """Pop one message from the queue."""
         try:

metagpt/team.py

@@ -71,9 +71,8 @@ class Team(BaseModel):
 
         # recover environment
         environment = Environment.deserialize(stg_path=stg_path.joinpath("environment"))
-        # team_info.update({"env": environment})
+        team_info.update({"env": environment})
         team = Team(**team_info)
-        team.env = environment
         return team
 
     def hire(self, roles: list[Role]):