phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

feat - init galileo logger

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-05 19:17:09 -07:00
parent 9944e8d1a5
commit e9319085f8
1 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/litellm_core_utils/litellm_logging.py
View file

@ -56,6 +56,7 @@ from ..integrations.clickhouse import ClickhouseLogger
from ..integrations.custom_logger import CustomLogger from ..integrations.custom_logger import CustomLogger
from ..integrations.datadog import DataDogLogger from ..integrations.datadog import DataDogLogger
from ..integrations.dynamodb import DyanmoDBLogger from ..integrations.dynamodb import DyanmoDBLogger
from ..integrations.galileo import GalileoObserve
from ..integrations.greenscale import GreenscaleLogger from ..integrations.greenscale import GreenscaleLogger
from ..integrations.helicone import HeliconeLogger from ..integrations.helicone import HeliconeLogger
from ..integrations.lago import LagoLogger from ..integrations.lago import LagoLogger
@ -1929,6 +1930,15 @@ def _init_custom_logger_compatible_class(
_openmeter_logger = OpenMeterLogger() _openmeter_logger = OpenMeterLogger()
_in_memory_loggers.append(_openmeter_logger) _in_memory_loggers.append(_openmeter_logger)
return _openmeter_logger # type: ignore return _openmeter_logger # type: ignore
elif logging_integration == "galileo":
for callback in _in_memory_loggers:
if isinstance(callback, GalileoObserve):
return callback # type: ignore
galileo_logger = GalileoObserve()
_in_memory_loggers.append(galileo_logger)
return galileo_logger # type: ignore
elif logging_integration == "logfire": elif logging_integration == "logfire":
if "LOGFIRE_TOKEN" not in os.environ: if "LOGFIRE_TOKEN" not in os.environ:
raise ValueError("LOGFIRE_TOKEN not found in environment variables") raise ValueError("LOGFIRE_TOKEN not found in environment variables")
@ -1985,6 +1995,10 @@ def get_custom_logger_compatible_class(
for callback in _in_memory_loggers: for callback in _in_memory_loggers:
if isinstance(callback, OpenMeterLogger): if isinstance(callback, OpenMeterLogger):
return callback return callback
elif logging_integration == "galileo":
for callback in _in_memory_loggers:
if isinstance(callback, GalileoObserve):
return callback
elif logging_integration == "logfire": elif logging_integration == "logfire":
if "LOGFIRE_TOKEN" not in os.environ: if "LOGFIRE_TOKEN" not in os.environ:
raise ValueError("LOGFIRE_TOKEN not found in environment variables") raise ValueError("LOGFIRE_TOKEN not found in environment variables")