scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

feat(logging): register the clickhouse callback

Browse source
This commit is contained in:
Ishaan Jaff 2026-09-30 00:40:11 -07:00
parent e73ab9eaf9
commit 23e332a607
No known key found for this signature in database
1 changed files with 13 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
litellm/litellm_core_utils/litellm_logging.py
View file

@ -180,6 +180,7 @@ from ..integrations.arize.arize_phoenix import ArizePhoenixLogger
from ..integrations.athina import AthinaLogger
from ..integrations.azure_sentinel.azure_sentinel import AzureSentinelLogger
from ..integrations.azure_storage.azure_storage import AzureBlobStorageLogger
from ..integrations.clickhouse.clickhouse_spend_logger import ClickHouseSpendLogger
from ..integrations.custom_prompt_management import CustomPromptManagement
from ..integrations.datadog.datadog import DataDogLogger
from ..integrations.datadog.datadog_llm_obs import DataDogLLMObsLogger
@ -4638,6 +4639,14 @@ def _init_custom_logger_compatible_class(
_s3_v2_logger: Final = S3V2Logger()
_in_memory_loggers.append(_s3_v2_logger)
return _s3_v2_logger
elif logging_integration == "clickhouse":
for callback in _in_memory_loggers:
if isinstance(callback, ClickHouseSpendLogger):
return callback
_clickhouse_spend_logger: Final = ClickHouseSpendLogger()
_in_memory_loggers.append(_clickhouse_spend_logger)
return _clickhouse_spend_logger
elif logging_integration == "pointfive":
for callback in _in_memory_loggers:
if isinstance(callback, PointFiveLogger):
@ -5374,6 +5383,10 @@ def get_custom_logger_compatible_class(
for callback in _in_memory_loggers:
if isinstance(callback, S3V2Logger):
return callback
elif logging_integration == "clickhouse":
for callback in _in_memory_loggers:
if isinstance(callback, ClickHouseSpendLogger):
return callback
elif logging_integration == "pointfive":
for callback in _in_memory_loggers:
if isinstance(callback, PointFiveLogger):