mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 02:34:29 +00:00
9b7ebb6a7d
* build(pyproject.toml): add new dev dependencies - for type checking * build: reformat files to fit black * ci: reformat to fit black * ci(test-litellm.yml): make tests run clear * build(pyproject.toml): add ruff * fix: fix ruff checks * build(mypy/): fix mypy linting errors * fix(hashicorp_secret_manager.py): fix passing cert for tls auth * build(mypy/): resolve all mypy errors * test: update test * fix: fix black formatting * build(pre-commit-config.yaml): use poetry run black * fix(proxy_server.py): fix linting error * fix: fix ruff safe representation error
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
import os
|
|
from typing import TYPE_CHECKING, Any, Union
|
|
|
|
from litellm._logging import verbose_logger
|
|
from litellm.integrations.arize import _utils
|
|
from litellm.types.integrations.arize_phoenix import ArizePhoenixConfig
|
|
|
|
if TYPE_CHECKING:
|
|
from opentelemetry.trace import Span as _Span
|
|
|
|
from litellm.types.integrations.arize import Protocol as _Protocol
|
|
|
|
from .opentelemetry import OpenTelemetryConfig as _OpenTelemetryConfig
|
|
|
|
Protocol = _Protocol
|
|
OpenTelemetryConfig = _OpenTelemetryConfig
|
|
Span = Union[_Span, Any]
|
|
else:
|
|
Protocol = Any
|
|
OpenTelemetryConfig = Any
|
|
Span = Any
|
|
|
|
|
|
ARIZE_HOSTED_PHOENIX_ENDPOINT = "https://app.phoenix.arize.com/v1/traces"
|
|
|
|
|
|
class ArizePhoenixLogger:
|
|
@staticmethod
|
|
def set_arize_phoenix_attributes(span: Span, kwargs, response_obj):
|
|
_utils.set_attributes(span, kwargs, response_obj)
|
|
return
|
|
|
|
@staticmethod
|
|
def get_arize_phoenix_config() -> ArizePhoenixConfig:
|
|
"""
|
|
Retrieves the Arize Phoenix configuration based on environment variables.
|
|
|
|
Returns:
|
|
ArizePhoenixConfig: A Pydantic model containing Arize Phoenix configuration.
|
|
"""
|
|
api_key = os.environ.get("PHOENIX_API_KEY", None)
|
|
grpc_endpoint = os.environ.get("PHOENIX_COLLECTOR_ENDPOINT", None)
|
|
http_endpoint = os.environ.get("PHOENIX_COLLECTOR_HTTP_ENDPOINT", None)
|
|
|
|
endpoint = None
|
|
protocol: Protocol = "otlp_http"
|
|
|
|
if http_endpoint:
|
|
endpoint = http_endpoint
|
|
protocol = "otlp_http"
|
|
elif grpc_endpoint:
|
|
endpoint = grpc_endpoint
|
|
protocol = "otlp_grpc"
|
|
else:
|
|
endpoint = ARIZE_HOSTED_PHOENIX_ENDPOINT
|
|
protocol = "otlp_http"
|
|
verbose_logger.debug(
|
|
f"No PHOENIX_COLLECTOR_ENDPOINT or PHOENIX_COLLECTOR_HTTP_ENDPOINT found, using default endpoint with http: {ARIZE_HOSTED_PHOENIX_ENDPOINT}"
|
|
)
|
|
|
|
otlp_auth_headers = None
|
|
# If the endpoint is the Arize hosted Phoenix endpoint, use the api_key as the auth header as currently it is uses
|
|
# a slightly different auth header format than self hosted phoenix
|
|
if endpoint == ARIZE_HOSTED_PHOENIX_ENDPOINT:
|
|
if api_key is None:
|
|
raise ValueError(
|
|
"PHOENIX_API_KEY must be set when the Arize hosted Phoenix endpoint is used."
|
|
)
|
|
otlp_auth_headers = f"api_key={api_key}"
|
|
elif api_key is not None:
|
|
# api_key/auth is optional for self hosted phoenix
|
|
otlp_auth_headers = f"Authorization=Bearer {api_key}"
|
|
|
|
return ArizePhoenixConfig(
|
|
otlp_auth_headers=otlp_auth_headers, protocol=protocol, endpoint=endpoint
|
|
)
|