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

[Feat] Allow setting custom arize endpoint (#5709)

Browse source 
* set arize endpoint

* docs arize endpoint

* fix arize endpoint
This commit is contained in:
Ishaan Jaff 2024-09-21 13:12:00 -07:00 • committed by GitHub
parent 16176da9fa
commit a9caba33ef
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
docs/my-website/docs/observability/arize_integration.md
View file

@ -62,6 +62,7 @@ litellm_settings:
environment_variables: environment_variables:
ARIZE_SPACE_KEY: "d0*****" ARIZE_SPACE_KEY: "d0*****"
ARIZE_API_KEY: "141a****" ARIZE_API_KEY: "141a****"
ARIZE_ENDPOINT: "https://otlp.arize.com/v1" # OPTIONAL - your custom arize api endpoint
``` ```
## Support & Talk to Founders ## Support & Talk to Founders

1
docs/my-website/docs/proxy/logging.md
View file

@ -1279,6 +1279,7 @@ litellm_settings:
environment_variables: environment_variables:
ARIZE_SPACE_KEY: "d0*****" ARIZE_SPACE_KEY: "d0*****"
ARIZE_API_KEY: "141a****" ARIZE_API_KEY: "141a****"
ARIZE_ENDPOINT: "https://otlp.arize.com/v1" # OPTIONAL - your custom arize api endpoint
``` ```
2. Start Proxy 2. Start Proxy

5
litellm/litellm_core_utils/litellm_logging.py
View file

@ -2175,9 +2175,12 @@ def _init_custom_logger_compatible_class(
OpenTelemetryConfig, OpenTelemetryConfig,
) )
arize_endpoint = (
os.environ.get("ARIZE_ENDPOINT", None) or "https://otlp.arize.com/v1"
)
otel_config = OpenTelemetryConfig( otel_config = OpenTelemetryConfig(
exporter="otlp_grpc", exporter="otlp_grpc",
endpoint="https://otlp.arize.com/v1", endpoint=arize_endpoint,
) )
os.environ["OTEL_EXPORTER_OTLP_TRACES_HEADERS"] = ( os.environ["OTEL_EXPORTER_OTLP_TRACES_HEADERS"] = (
f"space_key={os.getenv('ARIZE_SPACE_KEY')},api_key={os.getenv('ARIZE_API_KEY')}" f"space_key={os.getenv('ARIZE_SPACE_KEY')},api_key={os.getenv('ARIZE_API_KEY')}"