save key

lang_agent/config/core_config.py

@@ -71,55 +71,57 @@ 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(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_masked_serializable(obj):
-           
+        def to_serializable(obj, apply_mask: bool):
             """
-            Recursively convert dataclasses and containers to serializable with masked secrets
-            
-            递归地将数据类和容器转换为可序列化的格式，同时对敏感信息进行掩码处理
-            
+            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_masked_serializable(v)
+                        out[k] = to_serializable(v, apply_mask)
                     else:
-                        out[k] = mask_value(k, v)
+                        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_masked_serializable(v)
+                        out[k] = to_serializable(v, apply_mask)
                     else:
-                        # k might be a non-string; convert to str for is_secrete check consistency
                         key_str = str(k)
-                        out[k] = mask_value(key_str, v)
+                        out[k] = mask_value(key_str, v) if apply_mask else v
                 return out
             if isinstance(obj, list):
-                return [to_masked_serializable(v) for v in obj]
+                return [to_serializable(v, apply_mask) for v in obj]
             if isinstance(obj, tuple):
-                return tuple(to_masked_serializable(v) for v in obj)
+                return tuple(to_serializable(v, apply_mask) for v in obj)
             return obj
 
-        masked = to_masked_serializable(self)
+        # 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(masked, f)
+            yaml.dump(serializable, f)
         logger.info(f"[yellow]config saved to: {filename}[/yellow]")
     
     def get_name(self):