uPj UdZddlmZmZddlmZGddeZidddd d d d d dddddddddddd dddd dddd ddd d d!d"Zeee d#<d$ed%e d&zfd'Z d$ed%e d&zfd(Z d$e d%efd)Zy&)*zB Common utilities and exceptions for the NVIDIA Riva STT provider )AnyFinal)BaseLLMExceptionceZdZdZy)NvidiaRivaExceptiona Exception raised for NVIDIA Riva (gRPC) errors. ``status_code`` is an HTTP-equivalent code derived from the underlying gRPC ``StatusCode`` (when available) so that litellm's existing error classifiers (RateLimitError, AuthenticationError, etc.) keep working. N)__name__ __module__ __qualname____doc__j/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/nvidia_riva/common_utils.pyrr sr rOK CANCELLEDiUNKNOWNINVALID_ARGUMENTiDEADLINE_EXCEEDEDi NOT_FOUNDiALREADY_EXISTSiPERMISSION_DENIEDiRESOURCE_EXHAUSTEDiFAILED_PRECONDITIONABORTED OUT_OF_RANGE UNIMPLEMENTEDiINTERNAL UNAVAILABLEi DATA_LOSSUNAUTHENTICATEDi_GRPC_STATUS_CODE_TO_HTTPerrorreturnNct|dd}t|r' |}t|dd}t|tr|Sy#t$rd}Y-wxYw)z Best-effort extraction of a gRPC StatusCode name from an arbitrary error. Works for ``grpc.RpcError`` instances (which expose ``.code()``) as well as plain exceptions whose string representation contains a status name. codeNnamegetattrcallable Exception isinstancestr)r#code_fnr&r's r_extract_grpc_status_namer/,s^UFD1G 9DdFD1 dC K   D sA AAct|dd}t|r |}t|tr|r|Sy#t$rd}Y"wxYw)zKBest-effort extraction of a human-readable detail string from a gRPC error.detailsNr()r# details_fnr1s r_extract_grpc_detailsr3?sPy$7J   lG gs #N   G s6 AAct|}tj|xsdd}t|xs t |}|rd|d|nd|}t ||S)z Convert a gRPC error (or any exception raised from the Riva client) into a ``NvidiaRivaException`` with an appropriate HTTP-equivalent status code. rzNVIDIA Riva gRPC error (z): zNVIDIA Riva error: ) status_codemessage)r/r"getr3r-r)r# status_name http_statusdetailr7s rgrpc_error_to_litellm_exceptionr<Lsf 359K266{7Hb#NK)%0>CJFEP( SAXklrksVtG ; HHr )r typingrr)litellm.llms.base_llm.chat.transformationrrr"dict__annotations__r-r/r3r+r<r r rrAs?F**#**s* *  *  *c**#*3*s*C*S**3* !*"s#*5;*SS4Z&  t  I9 I9L Ir