From 604568cc7935847bee7aeaad982d4eadada783fd Mon Sep 17 00:00:00 2001 From: "fangyaozheng@bytedance.com" Date: Mon, 18 Aug 2025 14:17:11 +0800 Subject: [PATCH 1/2] remove secret keys from logging --- veadk/memory/long_term_memory.py | 2 +- veadk/memory/short_term_memory.py | 10 +++------- 2 files changed, 4 insertions(+), 8 deletions(-) diff --git a/veadk/memory/long_term_memory.py b/veadk/memory/long_term_memory.py index 24d572aa..cca18089 100644 --- a/veadk/memory/long_term_memory.py +++ b/veadk/memory/long_term_memory.py @@ -63,7 +63,7 @@ def __init__( self.adapter = get_long_term_memory_database_adapter(self.db_client) logger.info( - f"Initialized long term memory: db_client={self.db_client} adapter={self.adapter}" + f"Initialized long term memory: db_client={self.db_client.__class__.__name__} adapter={self.adapter}" ) def _filter_and_convert_events(self, events: list[Event]) -> list[str]: diff --git a/veadk/memory/short_term_memory.py b/veadk/memory/short_term_memory.py index 6acb3cde..d3878c1f 100644 --- a/veadk/memory/short_term_memory.py +++ b/veadk/memory/short_term_memory.py @@ -60,18 +60,14 @@ def __init__( self.session_service = InMemorySessionService() elif self.backend == "database": if self.db_url == "" or self.db_url is None: - logger.warning( - f"The `db_url` {self.db_url} is an empty or None string." - ) + logger.warning("The `db_url` is an empty or None string.") self._use_default_database() else: try: self.session_service = DatabaseSessionService(db_url=self.db_url) - logger.info("Connected to database.") + logger.info("Connected to database with db_url.") except Exception as e: - logger.error( - f"Failed to connect to database {self.db_url}, error: {e}." - ) + logger.error(f"Failed to connect to database, error: {e}.") self._use_default_database() else: raise ValueError(f"Unknown short term memory backend: {self.backend}") From f510151f06cf5fddeefccd76d6f867be676aedc4 Mon Sep 17 00:00:00 2001 From: "fangyaozheng@bytedance.com" Date: Mon, 18 Aug 2025 14:20:53 +0800 Subject: [PATCH 2/2] remove secret output from kb logging --- veadk/knowledgebase/knowledgebase.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/veadk/knowledgebase/knowledgebase.py b/veadk/knowledgebase/knowledgebase.py index 01c8c55b..f9e92556 100644 --- a/veadk/knowledgebase/knowledgebase.py +++ b/veadk/knowledgebase/knowledgebase.py @@ -32,9 +32,7 @@ def __init__( top_k: int = 10, db_config=None, ): - logger.info( - f"Initializing knowledgebase: backend={backend} top_k={top_k} db_config={db_config}" - ) + logger.info(f"Initializing knowledgebase: backend={backend} top_k={top_k}") self.backend = backend self.top_k = top_k @@ -43,7 +41,7 @@ def __init__( self.adapter = get_knowledgebase_database_adapter(self.db_client) logger.info( - f"Initialized knowledgebase: db_client={self.db_client} adapter={self.adapter}" + f"Initialized knowledgebase: db_client={self.db_client.__class__.__name__} adapter={self.adapter}" ) def add(