mirror of
https://github.com/hoshikawa2/nemo_guardrails_configuration.git
synced 2026-07-09 17:04:20 +00:00
Funcional
This commit is contained in:
43
nemo_guardrails_tracing_project/config/config.py
Normal file
43
nemo_guardrails_tracing_project/config/config.py
Normal file
@@ -0,0 +1,43 @@
|
||||
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
|
||||
)
|
||||
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)
|
||||
@@ -1,11 +1,27 @@
|
||||
colang_version: "1.0"
|
||||
|
||||
models:
|
||||
- type: main
|
||||
engine: openai
|
||||
model: gpt-5
|
||||
api_key_env_var: OPENAI_API_KEY
|
||||
parameters:
|
||||
temperature: 0
|
||||
base_url: http://127.0.0.1:8051/v1
|
||||
max_tokens: 50 # 🔥 evita respostas longas do LLM
|
||||
|
||||
# 🔥 usado apenas se você chamar explicitamente no flow
|
||||
- type: self_check_input
|
||||
engine: openai
|
||||
model: openai.gpt-oss-120b
|
||||
api_key_env_var: OPENAI_API_KEY
|
||||
parameters:
|
||||
temperature: 0
|
||||
base_url: http://127.0.0.1:8051/v1
|
||||
|
||||
|
||||
rails:
|
||||
input:
|
||||
flows:
|
||||
- main
|
||||
- check_input_terms
|
||||
output:
|
||||
flows:
|
||||
- check_output_terms
|
||||
|
||||
@@ -1,9 +0,0 @@
|
||||
define user express input
|
||||
$input_text
|
||||
|
||||
define flow main
|
||||
user express input
|
||||
|
||||
execute executar_pipeline_validacoes
|
||||
|
||||
bot say $nemo_response
|
||||
Reference in New Issue
Block a user