o ưi@sddlZddlmZmZmZmZddlmZer5ddlm Z ddlm Z ddl mZeZe Z ee efZ neZeZ eZ GdddZGd d d eZdS) N) TYPE_CHECKINGAnyOptionalUnion) OpenTelemetry)Span)OpenTelemetryConfig)Protocolc@s*eZdZdZdeededefddZdS) LevoConfigz(Configuration for Levo OTLP integration.otlp_auth_headersprotocolendpointcCs||_||_||_dS)Nr r r )selfr r r rU/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/levo/levo.py__init__s zLevoConfig.__init__N)__name__ __module__ __qualname____doc__rstrr rrrrrr sr c@s*eZdZdZedefddZddZdS) LevoLoggerz   zLevoLogger.get_levo_configc s\z|}|jsdddWSdddWSty-}z dt|dWYd}~Sd}~ww)z Health check for Levo integration. Returns: dict: Health status with status and message/error_message keys. Z unhealthyz+LEVOAI_API_KEY environment variable not set)status error_messageZhealthyz(Levo credentials are configured properly)r"messageN)r!r rr)rconfigerrrasync_health_check\s zLevoLogger.async_health_checkN)rrrr staticmethodr r!r'rrrrr#s  5r)rtypingrrrrZ"litellm.integrations.opentelemetryrZopentelemetry.tracer_SpanrZ_OpenTelemetryConfigZ litellm.types.integrations.arizer Z _Protocolr rrrrrs