From bda75d255891980eb8055a6af957124e52455c74 Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Tue, 3 Mar 2026 19:27:45 +0800
Subject: [PATCH] change save location

---
 lang_agent/config/core_config.py | 58 ++++----------------------------
 1 file changed, 7 insertions(+), 51 deletions(-)

diff --git a/lang_agent/config/core_config.py b/lang_agent/config/core_config.py
index ee15b2b..9b1afed 100644
--- a/lang_agent/config/core_config.py
+++ b/lang_agent/config/core_config.py
@@ -72,56 +72,12 @@ class InstantiateConfig(PrintableConfig):
         将配置保存到 YAML 文件
         
         """
-
-        def mask_value(key, value):
-            """
-            Apply masking if key is secret-like
-            如果键是敏感的，应用掩码
-
-            检查键是否敏感（如包含 "secret" 或 "api_key"），如果是，则对值进行掩码处理
-            """
-            if isinstance(value, str) and self.is_secrete(str(key)):
-                sval = str(value)
-                return sval[:3] + "*" * (len(sval) - 6) + sval[-3:]
-            return value
-
-        def to_serializable(obj, apply_mask: bool):
-            """
-            Recursively convert dataclasses and containers to serializable format,
-            optionally masking secrets.
-
-            递归地将数据类和容器转换为可序列化的格式，可选地对敏感信息进行掩码处理
-            """
-            if is_dataclass(obj):
-                out = {}
-                for k, v in vars(obj).items():
-                    if is_dataclass(v) or isinstance(v, (dict, list, tuple)):
-                        out[k] = to_serializable(v, apply_mask)
-                    else:
-                        out[k] = mask_value(k, v) if apply_mask else v
-                return out
-            if isinstance(obj, dict):
-                out = {}
-                for k, v in obj.items():
-                    if is_dataclass(v) or isinstance(v, (dict, list, tuple)):
-                        out[k] = to_serializable(v, apply_mask)
-                    else:
-                        key_str = str(k)
-                        out[k] = mask_value(key_str, v) if apply_mask else v
-                return out
-            if isinstance(obj, list):
-                return [to_serializable(v, apply_mask) for v in obj]
-            if isinstance(obj, tuple):
-                return tuple(to_serializable(v, apply_mask) for v in obj)
-            return obj
-
-        # NOTE: we intentionally do NOT mask secrets when saving to disk so that
-        # configs can be reloaded with real values. Masking is handled in __str__
-        # for safe logging/printing. If you need a redacted copy, call
-        # to_serializable(self, apply_mask=True) manually and dump it yourself.
-        serializable = to_serializable(self, apply_mask=False)
-        with open(filename, 'w') as f:
-            yaml.dump(serializable, f)
+        # Persist the full config object (including type tags) so it can be
+        # deserialized back into config instances with methods like .setup().
+        # Secret masking is intentionally handled by __str__ for printing/logging,
+        # not when writing to disk.
+        with open(filename, "w", encoding="utf-8") as f:
+            yaml.dump(self, f)
         logger.info(f"[yellow]config saved to: {filename}[/yellow]")
     
     def get_name(self):
@@ -182,7 +138,7 @@ def load_tyro_conf(filename: str, inp_conf = None) -> InstantiateConfig:
     """
     config = yaml.load(Path(filename).read_text(), Loader=yaml.Loader)
 
-    config = ovewrite_config(config, inp_conf) if inp_conf is not None else config
+    # config = ovewrite_config(config, inp_conf) if inp_conf is not None else config
     return config
 
 def is_default(instance, field_):