Skip to content

Commit 4508aa3

Browse files
fix: refine logging (#70)
1 parent b3954e9 commit 4508aa3

File tree

3 files changed

+6
-12
lines changed

3 files changed

+6
-12
lines changed

veadk/knowledgebase/knowledgebase.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,7 @@ def __init__(
3232
top_k: int = 10,
3333
db_config=None,
3434
):
35-
logger.info(
36-
f"Initializing knowledgebase: backend={backend} top_k={top_k} db_config={db_config}"
37-
)
35+
logger.info(f"Initializing knowledgebase: backend={backend} top_k={top_k}")
3836

3937
self.backend = backend
4038
self.top_k = top_k
@@ -43,7 +41,7 @@ def __init__(
4341
self.adapter = get_knowledgebase_database_adapter(self.db_client)
4442

4543
logger.info(
46-
f"Initialized knowledgebase: db_client={self.db_client} adapter={self.adapter}"
44+
f"Initialized knowledgebase: db_client={self.db_client.__class__.__name__} adapter={self.adapter}"
4745
)
4846

4947
def add(

veadk/memory/long_term_memory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def __init__(
6363
self.adapter = get_long_term_memory_database_adapter(self.db_client)
6464

6565
logger.info(
66-
f"Initialized long term memory: db_client={self.db_client} adapter={self.adapter}"
66+
f"Initialized long term memory: db_client={self.db_client.__class__.__name__} adapter={self.adapter}"
6767
)
6868

6969
def _filter_and_convert_events(self, events: list[Event]) -> list[str]:

veadk/memory/short_term_memory.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -60,18 +60,14 @@ def __init__(
6060
self.session_service = InMemorySessionService()
6161
elif self.backend == "database":
6262
if self.db_url == "" or self.db_url is None:
63-
logger.warning(
64-
f"The `db_url` {self.db_url} is an empty or None string."
65-
)
63+
logger.warning("The `db_url` is an empty or None string.")
6664
self._use_default_database()
6765
else:
6866
try:
6967
self.session_service = DatabaseSessionService(db_url=self.db_url)
70-
logger.info("Connected to database.")
68+
logger.info("Connected to database with db_url.")
7169
except Exception as e:
72-
logger.error(
73-
f"Failed to connect to database {self.db_url}, error: {e}."
74-
)
70+
logger.error(f"Failed to connect to database, error: {e}.")
7571
self._use_default_database()
7672
else:
7773
raise ValueError(f"Unknown short term memory backend: {self.backend}")

0 commit comments

Comments
 (0)