diff --git a/metagpt/config2.py b/metagpt/config2.py index 230e090af..6b6f4935b 100644 --- a/metagpt/config2.py +++ b/metagpt/config2.py @@ -153,12 +153,13 @@ def merge_dict(dicts: Iterable[Dict]) -> Dict: return result -class ConfigMixin: +class ConfigMixin(BaseModel): """Mixin class for configurable objects""" _config: Optional[Config] = None def __init__(self, config: Optional[Config] = None): + super().__init__() self._config = config def try_set_parent_config(self, parent_config): diff --git a/tests/metagpt/test_config.py b/tests/metagpt/test_config.py index eecabb546..85e32818d 100644 --- a/tests/metagpt/test_config.py +++ b/tests/metagpt/test_config.py @@ -5,8 +5,9 @@ @Author : alexanderwu @File : test_config.py """ +from pydantic import BaseModel -from metagpt.config2 import Config, config +from metagpt.config2 import Config, ConfigMixin, config from metagpt.configs.llm_config import LLMType from tests.metagpt.provider.mock_llm_config import mock_llm_config @@ -26,3 +27,24 @@ def test_config_from_dict(): cfg = Config(llm={"default": mock_llm_config}) assert cfg assert cfg.llm["default"].api_key == "mock_api_key" + + +class NewModel(ConfigMixin, BaseModel): + a: str = "a" + b: str = "b" + + +def test_config_mixin(): + new_model = NewModel() + assert new_model.a == "a" + assert new_model.b == "b" + assert new_model._config == new_model.config + assert new_model._config is None + + +def test_config_mixin_2(): + i = Config(llm={"default": mock_llm_config}) + new_model = NewModel(config=i) + assert new_model.config == i + assert new_model._config == i + assert new_model.config.llm["default"] == mock_llm_config