From aadbbe9841cf465a36e964df1ba959dc31d95a63 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 10 Oct 2024 18:43:39 +0530 Subject: [PATCH] fix _opik logger --- litellm/litellm_core_utils/litellm_logging.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/litellm/litellm_core_utils/litellm_logging.py b/litellm/litellm_core_utils/litellm_logging.py index 9bea6b9a9..2d70975c9 100644 --- a/litellm/litellm_core_utils/litellm_logging.py +++ b/litellm/litellm_core_utils/litellm_logging.py @@ -76,6 +76,7 @@ from ..integrations.literal_ai import LiteralAILogger from ..integrations.logfire_logger import LogfireLevel, LogfireLogger from ..integrations.lunary import LunaryLogger from ..integrations.openmeter import OpenMeterLogger +from ..integrations.opik.opik import OpikLogger from ..integrations.prometheus import PrometheusLogger from ..integrations.prometheus_services import PrometheusServicesLogger from ..integrations.prompt_layer import PromptLayerLogger @@ -2430,6 +2431,14 @@ def _init_custom_logger_compatible_class( _gcs_bucket_logger = GCSBucketLogger() _in_memory_loggers.append(_gcs_bucket_logger) 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": if "ARIZE_SPACE_KEY" not in os.environ: 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: if isinstance(callback, GCSBucketLogger): return callback + elif logging_integration == "opik": + for callback in _in_memory_loggers: + if isinstance(callback, OpikLogger): + return callback elif logging_integration == "otel": from litellm.integrations.opentelemetry import OpenTelemetry