o ưi @s*ddlZddlmZmZGdddZdS)N) HTTPExceptionstatusc@sJeZdZedefddZedefddZddgZedefd d Z d S) EnterpriseRouteChecksreturncCVddlm}ddlm}ddlm}dtjvr%|s%tt j d|j j d|dduS) z4 Check if llm api route is disabled rCommonProxyErrors premium_userget_secret_boolZDISABLE_LLM_API_ENDPOINTSuG🚨🚨🚨 DISABLING LLM API ENDPOINTS is an Enterprise feature 🚨  status_codedetailT Zlitellm.proxy._typesrZlitellm.proxy.proxy_serverr Zlitellm.secret_managers.mainr osenvironrrZHTTP_500_INTERNAL_SERVER_ERRORZnot_premium_uservaluerr r ra/home/app/Keep/.python/lib/python3.10/site-packages/litellm_enterprise/proxy/auth/route_checks.pyis_llm_api_route_disableds      z/EnterpriseRouteChecks.is_llm_api_route_disabledcCr) z7 Check if management route is disabled rrr r ZDISABLE_ADMIN_ENDPOINTSuE🚨🚨🚨 DISABLING ADMIN ENDPOINTS is an Enterprise feature 🚨 r Trrrrris_management_routes_disableds      z3EnterpriseRouteChecks.is_management_routes_disabledz/modelsz /v1/modelsroutecCsfddlm}|j|drtrttjdd|j|dr-|tj vr/t r1ttjdddSdSdS)zK Check if management route is disabled and raise exception r) RouteChecks)rz1Management routes are disabled for this instance.r z.LLM API routes are disabled for this instance.N) Zlitellm.proxy.auth.route_checksrZis_management_routerrrrZHTTP_403_FORBIDDENZis_llm_api_routeLLM_API_EXEMPT_ROUTESr)rrrrrshould_call_route0s(    z'EnterpriseRouteChecks.should_call_routeN) __name__ __module__ __qualname__ staticmethodboolrrrstrrrrrrrsr)rZfastapirrrrrrrs