import requests
import json
import os
url = "https://api.enkryptai.com/guardrails/add-guardrail"
payload = json.dumps({
"name": "My Guardrail",
"description": "Production guardrail with input/output scanning",
"input": {
"injection_attack": {
"enabled": True,
"block_message": "Potential injection attack detected"
},
"toxicity": {
"enabled": True,
"block_message": "Toxic content detected"
},
"pii": {
"enabled": True,
"entities": ["pii", "secrets", "ip_address", "url"]
},
"keyword_detector": {
"enabled": True,
"banned_keywords": ["confidential", "internal-only"]
},
"policy_violation": {
"enabled": True,
"need_explanation": True,
"coc_policy_name": "My CoC Policy",
"block_message": "Policy violation detected"
}
},
"output": {
"toxicity": {
"enabled": True,
"block_message": "Toxic response detected"
},
"bias": {
"enabled": True
},
"nsfw": {
"enabled": True,
"block_message": "NSFW content in response"
},
"policy_violation": {
"enabled": True,
"need_explanation": True,
"policy_text": "Do not provide medical, legal, or financial advice.",
"block_message": "Response violates policy"
}
}
})
headers = {
'Content-Type': 'application/json',
'apikey': os.getenv('ENKRYPTAI_API_KEY')
}
response = requests.post(url, headers=headers, data=payload)
print(json.dumps(response.json(), indent=4))