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

fix _opik logger

Browse source
This commit is contained in:
Ishaan Jaff 2024-10-10 18:43:39 +05:30
parent fbf756806e
commit aadbbe9841
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/litellm_core_utils/litellm_logging.py
View file

@ -76,6 +76,7 @@ from ..integrations.literal_ai import LiteralAILogger
from ..integrations.logfire_logger import LogfireLevel, LogfireLogger from ..integrations.logfire_logger import LogfireLevel, LogfireLogger
from ..integrations.lunary import LunaryLogger from ..integrations.lunary import LunaryLogger
from ..integrations.openmeter import OpenMeterLogger from ..integrations.openmeter import OpenMeterLogger
from ..integrations.opik.opik import OpikLogger
from ..integrations.prometheus import PrometheusLogger from ..integrations.prometheus import PrometheusLogger
from ..integrations.prometheus_services import PrometheusServicesLogger from ..integrations.prometheus_services import PrometheusServicesLogger
from ..integrations.prompt_layer import PromptLayerLogger from ..integrations.prompt_layer import PromptLayerLogger
@ -2430,6 +2431,14 @@ def _init_custom_logger_compatible_class(
_gcs_bucket_logger = GCSBucketLogger() _gcs_bucket_logger = GCSBucketLogger()
_in_memory_loggers.append(_gcs_bucket_logger) _in_memory_loggers.append(_gcs_bucket_logger)
return _gcs_bucket_logger # type: ignore return _gcs_bucket_logger # type: ignore
elif logging_integration == "opik":
for callback in _in_memory_loggers:
if isinstance(callback, OpikLogger):
return callback # type: ignore
_opik_logger = OpikLogger()
_in_memory_loggers.append(_opik_logger)
return _opik_logger # type: ignore
elif logging_integration == "arize": elif logging_integration == "arize":
if "ARIZE_SPACE_KEY" not in os.environ: if "ARIZE_SPACE_KEY" not in os.environ:
raise ValueError("ARIZE_SPACE_KEY not found in environment variables") raise ValueError("ARIZE_SPACE_KEY not found in environment variables")
@ -2563,6 +2572,10 @@ def get_custom_logger_compatible_class(
for callback in _in_memory_loggers: for callback in _in_memory_loggers:
if isinstance(callback, GCSBucketLogger): if isinstance(callback, GCSBucketLogger):
return callback return callback
elif logging_integration == "opik":
for callback in _in_memory_loggers:
if isinstance(callback, OpikLogger):
return callback
elif logging_integration == "otel": elif logging_integration == "otel":
from litellm.integrations.opentelemetry import OpenTelemetry from litellm.integrations.opentelemetry import OpenTelemetry