Prompt Shields

prompt-shields.py

# --------------------
# PROMPT AND DOCUMENT ATTACK
# --------------------

import os
from dotenv import load_dotenv
import requests

load_dotenv()

key = os.environ["CONTENT_SAFETY_KEY"]
endpoint = os.environ["CONTENT_SAFETY_ENDPOINT"]

url = f'{endpoint}/contentsafety/text:shieldPrompt?api-version=2024-02-15-preview'
subscription_key = key

headers = {
    'Ocp-Apim-Subscription-Key': subscription_key,
    'Content-Type': 'application/json'
}

data = {
    "userPrompt": "<Your user prompt>",
    "documents":["<Your documents>"]
}

response = requests.post(url, headers=headers, json=data)

print(response.json())