tPjgUddlZddlmZmZddlmZddlZddlZddlm Z ddl m Z ddl m Z mZddlmZddlmZmZdd lmZdd lmZdd lmZmZmZdd lmZd Zee d<dZ!ee d<dZ"ee d<Gdde Z#y)N)AnyFinal)urlparse)verbose_proxy_logger)GuardrailRaisedException)CustomGuardraillog_guardrail_information)Logging)get_async_httpx_clienthttpxSpecialProvider)GuardrailEventHooks)GuardrailConfigModel)SingulrGuardrailPayloadSingulrGuardrailRequestSingulrGuardrailResponse)GenericGuardrailAPIInputszhttp://localhost:8003_DEFAULT_API_BASEz/api/v1/ai-gateway/litellm_GUARD_ENDPOINTg>@_DEFAULT_TIMEOUTc eZdZ ddedzdedzdedzdedzdedzdedzded dffd Zed e d dzfd Z d e eefde ded e eeffdZ d e eeffdZde eefd edzfdZe dde d e deddd e f dZxZS)SingulrGuardrailNsingulr_api_keysingulr_api_basesingulr_application_idsingulr_guardrail_idblock_on_errortimeoutkwargsreturnc T|xstjjd|_|xs'tjjdxstj d|_t|j }|jdk(r'|jdvrtd|j d|xstjjd|_ |xstjjd |_ |8tjjd d } | jd v|_n||_|tn||_t#t$j& |_d|vr#t*j,t*j.g|d<t1 |ddi|y)NSINGULR_API_KEYSINGULR_API_BASE/http) localhostz 127.0.0.1zSingulr: api_base z uses plain HTTP for a non-local endpoint. Guardrail payloads contain the API token, full conversation content, and the guardrail decision, so this endpoint must use HTTPS.SINGULR_ENFORCEMENT_ENTITY_IDSINGULR_GUARDRAIL_IDSINGULR_BLOCK_ON_ERRORtrue)r)1yes) llm_providersupported_event_hooks)osenvirongetrrrstriprrschemehostname ValueErrorrrlowerrrrr r GuardrailCallback async_handlerr pre_call post_callsuper__init__) selfrrrrrrrparsedenv __class__s }/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/guardrails/guardrail_hooks/singulr/singulr.pyr<zSingulrGuardrail.__init__&sr /S"**..AR2S!1!lRZZ^^DV5W!l[l t t ! !!6!67 ==F "v? ( $T%:%:$;<"" '=&o On@o#$8$bBJJNNKaz2SingulrGuardrail._build_headers..s%  UO  s)z Content-Typezapplication/jsonzX-Singulr-Gateway-TokenzX-Singulr-Enforcement-Entity-IdzX-Singulr-Guardrail-Id)dictrrr)r=s rA_build_headerszSingulrGuardrail._build_headers~s\  5*D,@,@A5//52*4+D+D+JK"   rBrYcTK|jt}tjd| |jj ||j ||jd{}|jtj|j}tjd||S7O#tj$r}tjd|jj t#||j$rFt'|j(d|jj d|jj*|Yd}~yd}~wtj,$rP}tjdt#||j$rt'|j(d ||Yd}~yd}~wt.t0j2f$rP}tjd t#||j$rt'|j(d ||Yd}~yd}~wwxYww) Nz Singulr: %s)urlheadersrSrzSingulr: result=%sz Singulr API returned HTTP %s: %szSingulr API returned HTTP z: )guardrail_namemessagezSingulr API unreachable: %sz/Singulr API unreachable (block_on_error=True): z,Singulr API returned an invalid response: %sz*Singulr API returned an invalid response: )rrrdebugr8postrerraise_for_statusrmodel_validaterShttpxHTTPStatusErrorerrorrO status_codestrrrritextTransportErrorr5pydanticValidationError)r=rYendpointrOresultexcs rA _call_apizSingulrGuardrail._call_apis!223O3DE""=(;) $($6$6$;$;++- %<%H  % % '4CCHMMOTF & &';V DM$$  & &2 ((C  "".#'#6#69#,,:R:R9SSUVYVbVbVgVgUhi##  & &'Dc#h O"".#'#6#6McUSH445  & &'UWZ[^W_ `"".#'#6#6HN sh+H(:B:(B8)AB:7H(8B::H% BEH(H%.AF94H(9H%AH H( H%%H( logging_objzLiteLLMLoggingObj | Nonec:K|j|||}|s|S|j|d{}||Stjd|j|j |jr)t |jd|j xsdd|S7jw)Nz+Singulr: should_block=%s blocking_due_to=%szBlocked by Singulr: unknownT)rirjblocked_content)r[r{rrk should_blockblocking_due_torri)r=rIrHrJr|rYrys rAapply_guardrailz SingulrGuardrail.apply_guardrails,,\6:NM"nnW55 >M"" 9     " "   *#22.v/E/E/R.ST $   #6s,BBA+B)NNNNNNr^)__name__ __module__ __qualname__rsboolfloatrr< staticmethodtyperGrdrr[rerr{r r __classcell__)r@s@rArr%sd'+'+-1+/&* $/#t/#*/#!$d /# "Dj /# t /#/#/# /#b+d#9:TA++/38n/*/ / c3h />  S#X  -tCH~-:RUY:Y-^ 37 )  0  # rBr)$r/typingrr urllib.parserrorvlitellm._loggingrlitellm.exceptionsr%litellm.integrations.custom_guardrailrr *litellm.litellm_core_utils.litellm_loggingr LiteLLMLoggingObj&litellm.llms.custom_httpx.http_handlerr r litellm.types.guardrailsr 3litellm.types.proxy.guardrails.guardrail_hooks.baserrFrrrlitellm.types.utilsrr__annotations__rrrr.rBrArsp ! 179 :25255%ttrB