diff --git a/metagpt/actions/action.py b/metagpt/actions/action.py index 4136d7599..9b94ce461 100644 --- a/metagpt/actions/action.py +++ b/metagpt/actions/action.py @@ -19,12 +19,12 @@ from metagpt.schema import ( CodeSummarizeContext, CodingContext, RunCodeContext, - SerDeserMixin, + SerializationMixin, TestingContext, ) -class Action(SerDeserMixin, is_polymorphic_base=True): +class Action(SerializationMixin, is_polymorphic_base=True): model_config = ConfigDict(arbitrary_types_allowed=True, exclude=["llm"]) name: str = "" diff --git a/metagpt/roles/role.py b/metagpt/roles/role.py index 2b8209758..29f3b0595 100644 --- a/metagpt/roles/role.py +++ b/metagpt/roles/role.py @@ -36,7 +36,7 @@ from metagpt.llm import LLM, HumanProvider from metagpt.logs import logger from metagpt.memory import Memory from metagpt.provider.base_llm import BaseLLM -from metagpt.schema import Message, MessageQueue, SerDeserMixin +from metagpt.schema import Message, MessageQueue, SerializationMixin from metagpt.utils.common import ( any_to_name, any_to_str, @@ -126,7 +126,7 @@ class RoleContext(BaseModel): return self.memory.get() -class Role(SerDeserMixin, is_polymorphic_base=True): +class Role(SerializationMixin, is_polymorphic_base=True): """Role/Agent""" model_config = ConfigDict(arbitrary_types_allowed=True, exclude=["llm"]) diff --git a/metagpt/schema.py b/metagpt/schema.py index 46064472f..41303ea46 100644 --- a/metagpt/schema.py +++ b/metagpt/schema.py @@ -54,7 +54,7 @@ from metagpt.utils.serialize import ( ) -class SerDeserMixin(BaseModel): +class SerializationMixin(BaseModel): """SereDeserMixin for subclass' ser&deser""" __is_polymorphic_base = False @@ -62,7 +62,7 @@ class SerDeserMixin(BaseModel): @classmethod def __get_pydantic_core_schema__( - cls, source: type["SerDeserMixin"], handler: Callable[[Any], core_schema.CoreSchema] + cls, source: type["SerializationMixin"], handler: Callable[[Any], core_schema.CoreSchema] ) -> core_schema.CoreSchema: schema = handler(source) og_schema_ref = schema["ref"]