cristiano.hoshikawa/nemo_guardrails_configuration
1
0
Fork 0
mirror of https://github.com/hoshikawa2/nemo_guardrails_configuration.git synced 2026-07-10 01:14:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ajuste segurança-extra

Browse Source
This commit is contained in:
Cristiano Hoshikawa
2026-05-05 13:13:23 -03:00
parent 25bc77f582
commit d698f8f834
15 changed files with 580 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
final_pkg/company_nemo_guardrails/llm_client.py
View File

@@ -10,6 +10,12 @@ from company_nemo_guardrails.prompts.aluc import build_aluc_prompt
from company_nemo_guardrails.prompts.rqlt import build_rqlt_prompt
from company_nemo_guardrails.prompts.supervisor import build_supervisor_prompt
# Segurança
from company_nemo_guardrails.prompts.dlex_in import build_dlex_in_prompt
from company_nemo_guardrails.prompts.dlex_out import build_dlex_out_prompt
from company_nemo_guardrails.prompts.pinj import build_pinj_prompt
from company_nemo_guardrails.prompts.ragsec import build_ragsec_prompt
class LLMClient:
def __init__(self):
self.use_mock=os.getenv('USE_MOCK_LLM','true').lower()=='true'
@@ -54,6 +60,17 @@ class LLMClient:
elif task == "SUPERVISOR_VAS":
prompt = build_supervisor_prompt(payload)
# Segurança Extra
elif task == "PINJ":
prompt = build_pinj_prompt(payload["text"])
elif task == "RAGSEC":
prompt = build_ragsec_prompt(payload["text"])
elif task == "DLEX_IN":
prompt = build_dlex_in_prompt(payload["text"])
elif task == "DLEX_OUT":
prompt = build_dlex_out_prompt(payload["text"])
else:
raise ValueError(f"Task não suportada: {task}")