o ưi @s6dZddlmZmZddlmZmZGdddZdS)z Compliance checker for EU AI Act and GDPR regulations. Provides guardrail-agnostic compliance validation based on guardrail modes and execution results rather than specific guardrail names. )DictList)ComplianceCheckRequestComplianceCheckResultc@seZdZdZdefddZdedeefddZ d eede fd d Z d eede fd d Z de fddZde fddZde fddZde fddZde fddZde fddZdee fddZdee fddZdS)ComplianceCheckera% Validates compliance with EU AI Act and GDPR regulations. Uses guardrail-agnostic checks based on: - Whether any guardrails ran - Guardrail execution mode (pre-call, post-call, etc.) - Whether guardrails intervened/blocked content - Completeness of audit records datacCs||_|jpg|_dS)N)rZguardrail_information guardrails)selfrr V/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/compliance_checks.py__init__szComplianceChecker.__init__modereturncCsLg}|jD]}|d}|dur|dkr||q||kr#||q|S)z Get all guardrails that ran in a specific mode. If a guardrail doesn't have a mode specified, it's treated as pre-call (the most common case). Zguardrail_modeNpre_call)rgetappend)r r resultgZg_moder r r _get_guardrails_by_modes    z)ComplianceChecker._get_guardrails_by_modercCs(|D]}|dd}|dvrdSqdS)z;Check if any guardrail intervened (blocked/masked content).guardrail_status)Zguardrail_intervenedfailedZblockedTFr)r rrstatusr r r _has_guardrail_intervention0s  z-ComplianceChecker._has_guardrail_interventioncCs|sdStdd|DS)z4Check if all guardrails passed (no issues detected).Fcss|] }|ddkVqdS)rsuccessNr).0rr r r <sz;ComplianceChecker._all_guardrails_passed..)all)r rr r r _all_guardrails_passed8sz(ComplianceChecker._all_guardrails_passedcCs6t|jdk}tdd||rt|jddSddS)z-Art. 9: Check if any guardrails were applied.rzGuardrails appliedzArt. 9z guardrail(s) appliedzNo guardrails appliedZ check_nameZarticleZpasseddetail)lenrr)r has_guardrailsr r r _check_art_9_guardrails_applied@sz1ComplianceChecker._check_art_9_guardrails_appliedcC<|d}t|dk}tdd||rt|ddSddS)z)a Check GDPR compliance. Returns: List of compliance check results for: - Art. 32: Data protection applied (pre-call screening) - Art. 5(1)(c): Sensitive data protected - Art. 30: Audit record complete )r;r<r=r?r r r check_gdprrAzComplianceChecker.check_gdprN)__name__ __module__ __qualname____doc__rr strrrrr5rrrr$r)r9r;r<r=r@rBr r r r rs rN)rFtypingrrZ(litellm.types.proxy.compliance_endpointsrrrr r r r s