scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

fix(proxy): route config-file custom loggers via module helper

Browse source 
Keeps the private-access suppression on one short line so it carries a
reason without exceeding the line budget.
This commit is contained in:
wadrzl55333 2026-08-28 16:44:38 +08:00
parent 82b789a99d
commit 84afc7e924
1 changed files with 14 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
litellm/proxy/proxy_server.py
View file

@ -704,6 +704,18 @@ from litellm.types.utils import CredentialItem, CustomHuggingfaceTokenizer, RawR
from litellm.types.utils import ModelInfo as ModelMapInfo
from litellm.utils import _add_custom_logger_callback_to_specific_event
def _register_config_custom_logger_callback(callback: str, logging_event: Literal["success", "failure"]) -> None:
"""Register a config-file custom-logger name as an instance, not a string.
Pass-through endpoints log through the async success path only; a string in
``litellm.success_callback`` never reaches them. Mirrors the DB-config path.
"""
_add_custom_logger_callback_to_specific_event( # pyright: ignore[reportPrivateUsage] # mirrors DB-config path
callback, logging_event
)
try:
from litellm._version import version
except Exception:
@ -5127,10 +5139,7 @@ class ProxyConfig:
# instance already exists from an earlier registration
if callback in litellm.success_callback:
litellm.success_callback.remove(callback)
_add_custom_logger_callback_to_specific_event( # pyright: ignore[reportPrivateUsage]
callback,
"success", # mirrors the DB-config registration path
)
_register_config_custom_logger_callback(callback, "success")
else:
litellm.logging_callback_manager.add_litellm_success_callback(callback)
if "prometheus" in callback:
@ -5162,10 +5171,7 @@ class ProxyConfig:
if callback in litellm._known_custom_logger_compatible_callbacks:
if callback in litellm.failure_callback:
litellm.failure_callback.remove(callback)
_add_custom_logger_callback_to_specific_event( # pyright: ignore[reportPrivateUsage]
callback,
"failure", # mirrors the DB-config registration path
)
_register_config_custom_logger_callback(callback, "failure")
else:
litellm.logging_callback_manager.add_litellm_failure_callback(callback)
print( # noqa: T201