diff --git a/docs/my-website/docs/observability/arize_integration.md b/docs/my-website/docs/observability/arize_integration.md index 652d79af5..17be003f8 100644 --- a/docs/my-website/docs/observability/arize_integration.md +++ b/docs/my-website/docs/observability/arize_integration.md @@ -62,6 +62,7 @@ litellm_settings: environment_variables: ARIZE_SPACE_KEY: "d0*****" ARIZE_API_KEY: "141a****" + ARIZE_ENDPOINT: "https://otlp.arize.com/v1" # OPTIONAL - your custom arize api endpoint ``` ## Support & Talk to Founders diff --git a/docs/my-website/docs/proxy/logging.md b/docs/my-website/docs/proxy/logging.md index 5523b95f8..89ed0bda5 100644 --- a/docs/my-website/docs/proxy/logging.md +++ b/docs/my-website/docs/proxy/logging.md @@ -1279,6 +1279,7 @@ litellm_settings: environment_variables: ARIZE_SPACE_KEY: "d0*****" ARIZE_API_KEY: "141a****" + ARIZE_ENDPOINT: "https://otlp.arize.com/v1" # OPTIONAL - your custom arize api endpoint ``` 2. Start Proxy diff --git a/litellm/litellm_core_utils/litellm_logging.py b/litellm/litellm_core_utils/litellm_logging.py index ca02ea3aa..2b5cd2187 100644 --- a/litellm/litellm_core_utils/litellm_logging.py +++ b/litellm/litellm_core_utils/litellm_logging.py @@ -2175,9 +2175,12 @@ def _init_custom_logger_compatible_class( OpenTelemetryConfig, ) + arize_endpoint = ( + os.environ.get("ARIZE_ENDPOINT", None) or "https://otlp.arize.com/v1" + ) otel_config = OpenTelemetryConfig( exporter="otlp_grpc", - endpoint="https://otlp.arize.com/v1", + endpoint=arize_endpoint, ) os.environ["OTEL_EXPORTER_OTLP_TRACES_HEADERS"] = ( f"space_key={os.getenv('ARIZE_SPACE_KEY')},api_key={os.getenv('ARIZE_API_KEY')}"