o ưit@s>ddlmZddlmZmZmZddlmZGdddZdS))Union)DB_CONNECTION_ERROR_TYPESProxyErrorTypesProxyException) str_to_boolc@s`eZdZdZedefddZededefddZededefdd Z edefd d Z d S) PrismaDBExceptionHandlerz< Class to handle DB Exceptions or Connection Errors returncCs:ddlm}|dd}t|tr|St|durdSdS)zj Returns True if the request should be allowed to proceed despite the DB connection error r)general_settingsZ allow_requests_on_db_unavailableFT)Zlitellm.proxy.proxy_serverr get isinstanceboolr)r Z!_allow_requests_on_db_unavailabler Y/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/db/exception_handler.py&should_allow_request_on_db_unavailables   z?PrismaDBExceptionHandler.should_allow_request_on_db_unavailableecCsFddl}t|tr dSt||jjrdSt|tr!|jtjkr!dSdS)u Returns True if the exception is from a database outage / connection error. Any PrismaError qualifies — the DB failed to serve the request. Used by allow_requests_on_db_unavailable logic and endpoint 503 responses. rNTF) prismar rerrors PrismaErrorrtyperno_db_connection)rrr r ris_database_connection_error s z5PrismaDBExceptionHandler.is_database_connection_errorcsddl}t|tr dSt||jj|jjfrdSt||jjr4t|d}t fdd|Dr4dSt|t rA|j t j krAdSdS)u? Returns True only for transport/connectivity failures where a reconnect attempt makes sense (e.g. DB is unreachable, connection dropped). Use this for reconnect logic — data-layer errors like UniqueViolationError mean the DB IS reachable, so reconnecting would be pointless. rNT) zcan't reach database serverzcannot reach database serverz can't connectzcannot connectzconnection errorzconnection closedz timed outtimeoutzconnection refusedznetwork is unreachablezno route to hostz broken pipec3s|]}|vVqdS)Nr ).0keyword error_messager r RszGPrismaDBExceptionHandler.is_database_transport_error..F)rr rrZClientNotConnectedErrorZHTTPClientClosedErrorrstrloweranyrrrr)rrZconnection_keywordsr rris_database_transport_error1s  z4PrismaDBExceptionHandler.is_database_transport_errorcCst|r tr dS|)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)rrr)rr r rhandle_db_exceptionXs  z,PrismaDBExceptionHandler.handle_db_exceptionN) __name__ __module__ __qualname____doc__ staticmethodr r Exceptionrr r!r r r rr s&rN) typingrZlitellm.proxy._typesrrrZlitellm.secret_managers.mainrrr r r rs