refine code

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):

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