mirror of
https://github.com/hoshikawa2/nemo_guardrails_configuration.git
synced 2026-07-09 17:04:20 +00:00
51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
from nemoguardrails import LLMRails, RailsConfig
|
|
from src.actions import (
|
|
mask_pii_action,
|
|
detectar_toxicidade_action,
|
|
detectar_out_of_scope_action,
|
|
validar_alcada_action,
|
|
verbalizacao_prematura_action,
|
|
validar_groundedness_action,
|
|
supervisor_vas_avulso_action,
|
|
enforce_compliance_anatel_action,
|
|
calcular_tcr_action,
|
|
detectar_fallback_action,
|
|
registrar_violacao_action,
|
|
validar_consistencia_historica_action,
|
|
contabilizar_tokens_action,
|
|
calcular_eficiencia_nlu_action,
|
|
calcular_eficiencia_nlu_action,
|
|
detectar_loop_action,
|
|
medir_tamanho_mensagem_action,
|
|
calcular_precisao_revocacao_action,
|
|
avaliar_acuracia_semantica_action,
|
|
detectar_prompt_injection_jailbreak_action,
|
|
detectar_rag_injection_context_poisoning_action,
|
|
detectar_data_leakage_input_action,
|
|
detectar_data_leakage_output_action
|
|
)
|
|
def init(app: LLMRails):
|
|
|
|
app.register_action(mask_pii_action)
|
|
app.register_action(detectar_toxicidade_action)
|
|
app.register_action(detectar_out_of_scope_action)
|
|
app.register_action(validar_alcada_action)
|
|
app.register_action(verbalizacao_prematura_action)
|
|
app.register_action(validar_groundedness_action)
|
|
app.register_action(supervisor_vas_avulso_action)
|
|
app.register_action(enforce_compliance_anatel_action)
|
|
app.register_action(calcular_tcr_action)
|
|
app.register_action(detectar_fallback_action)
|
|
app.register_action(registrar_violacao_action)
|
|
app.register_action(validar_consistencia_historica_action)
|
|
app.register_action(contabilizar_tokens_action)
|
|
app.register_action(calcular_eficiencia_nlu_action)
|
|
app.register_action(calcular_eficiencia_nlu_action)
|
|
app.register_action(detectar_loop_action)
|
|
app.register_action(medir_tamanho_mensagem_action)
|
|
app.register_action(calcular_precisao_revocacao_action)
|
|
app.register_action(avaliar_acuracia_semantica_action)
|
|
app.register_action(detectar_prompt_injection_jailbreak_action)
|
|
app.register_action(detectar_rag_injection_context_poisoning_action)
|
|
app.register_action(detectar_data_leakage_input_action)
|
|
app.register_action(detectar_data_leakage_output_action) |