From b39fb0abe72e4cfd7f87ff8a479e8cdd5a505d59 Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Fri, 10 Oct 2025 16:32:12 +0800
Subject: [PATCH] better masking

---
 lang_agent/config.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/lang_agent/config.py b/lang_agent/config.py
index 2fda88b..f28c174 100644
--- a/lang_agent/config.py
+++ b/lang_agent/config.py
@@ -16,7 +16,8 @@ class PrintableConfig:
         for key, val in vars(self).items():
 
             if key.endswith("_secret") or ("key" in key):
-                val = "****"
+                val = str(val)
+                val = val[:3] + "*"*(len(val) - 3)
                 
             if isinstance(val, Tuple):
                 flattened_val = "["
@@ -26,6 +27,14 @@ class PrintableConfig:
                 val = flattened_val + "]"
             lines += f"{key}: {str(val)}".split("\n")
         return "\n"  + "\n    ".join(lines)
+    
+    def is_secrete(self, inp:str):
+        sec_list = ["secret", "api_key"]
+        for sec in sec_list:
+            if sec in inp:
+                return True
+        
+        return False
 
 
 # Base instantiate configs