Skip to content

Commit 9554f40

Browse files
authored
chore: log LITELLM_SETTINGS when first use (#1074)
* log LITELLM_SETTINGS when first use * fix ci
1 parent d0f33c5 commit 9554f40

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

rdagent/oai/backend/litellm.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,15 +40,19 @@ class Config:
4040

4141

4242
LITELLM_SETTINGS = LiteLLMSettings()
43-
logger.info(f"{LITELLM_SETTINGS}")
44-
logger.log_object(LITELLM_SETTINGS.model_dump(), tag="LITELLM_SETTINGS")
4543
ACC_COST = 0.0
4644

4745

4846
class LiteLLMAPIBackend(APIBackend):
4947
"""LiteLLM implementation of APIBackend interface"""
5048

49+
_has_logged_settings: bool = False
50+
5151
def __init__(self, *args: Any, **kwargs: Any) -> None:
52+
if not self.__class__._has_logged_settings:
53+
logger.info(f"{LITELLM_SETTINGS}")
54+
logger.log_object(LITELLM_SETTINGS.model_dump(), tag="LITELLM_SETTINGS")
55+
self.__class__._has_logged_settings = True
5256
super().__init__(*args, **kwargs)
5357

5458
def _calculate_token_from_messages(self, messages: list[dict[str, Any]]) -> int:

0 commit comments

Comments
 (0)