Skip to content

Commit e1a305d

Browse files
wenxi-onyxWeves
authored andcommitted
mask llm api key from logs
1 parent e2233d2 commit e1a305d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

backend/onyx/llm/chat_llm.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -313,14 +313,14 @@ def __init__(
313313

314314
self._model_kwargs = model_kwargs
315315

316-
def log_model_configs(self) -> None:
317-
logger.debug(f"Config: {self.config}")
318-
319316
def _safe_model_config(self) -> dict:
320317
dump = self.config.model_dump()
321318
dump["api_key"] = mask_string(dump.get("api_key", ""))
322319
return dump
323320

321+
def log_model_configs(self) -> None:
322+
logger.debug(f"Config: {self._safe_model_config()}")
323+
324324
def _record_call(self, prompt: LanguageModelInput) -> None:
325325
if self._long_term_logger:
326326
self._long_term_logger.record(

0 commit comments

Comments
 (0)