hDddlmZddlmZmZmZddlmZGddZy))Union)DB_CONNECTION_ERROR_TYPESProxyErrorTypesProxyException) str_to_boolcVeZdZdZedefdZededefdZedefdZ y)PrismaDBExceptionHandlerz< Class to handle DB Exceptions or Connection Errors returnctddlm}|jdd}t|tr|St |duryy)zj Returns True if the request should be allowed to proceed despite the DB connection error r)general_settings allow_requests_on_db_unavailableFT)litellm.proxy.proxy_serverr get isinstanceboolr)r !_allow_requests_on_db_unavailables ^/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/db/exception_handler.py&should_allow_request_on_db_unavailablez?PrismaDBExceptionHandler.should_allow_request_on_db_unavailablesE @>N>R>R .? ) 7 >4 4 8 9T Aecddl}t|tryt||jjryt|t r|j tjk(ryy)z\ Returns True if the exception is from a database outage / connection error rNTF) prismarrerrors PrismaErrorrtyperno_db_connection)rrs ris_database_connection_errorz5PrismaDBExceptionHandler.is_database_connection_error sM  a2 3 a22 3 a (QVV7W7W-WrcZtj|rtjry|)aK Primary handler for `allow_requests_on_db_unavailable` flag. Decides whether to raise a DB Exception or not based on the flag. - If exception is a DB Connection Error, and `allow_requests_on_db_unavailable` is True, - Do not raise an exception, return None - Else, raise the exception N)r rr)rs rhandle_db_exceptionz,PrismaDBExceptionHandler.handle_db_exception/s& % A A! D(OOQrN) __name__ __module__ __qualname____doc__ staticmethodrr Exceptionrrrrr r s_ D    d   y  rr N) typingrlitellm.proxy._typesrrrlitellm.secret_managers.mainrr r&rrr*s! 522r