import os
from dataframer import Dataframer

client = Dataframer(
    api_key=os.environ.get("DATAFRAMER_API_KEY"),  # This is the default and can be omitted
)
red_team_spec = client.dataframer.red_team_specs.create(
    app_description="A patient-facing chatbot that helps schedule appointments and answer general health questions",
    domain_description="Healthcare and medical information services",
    name="Healthcare Chatbot",
    concerns="Should not provide medical diagnoses, prescription recommendations, or treatment advice",
)
print(red_team_spec.id)

curl https://df-api.dataframer.ai/api/dataframer/red-team-specs/ \
    -H 'Content-Type: application/json' \
    -H "Authorization: Bearer $DATAFRAMER_API_KEY" \
    -d '{
          "app_description": "A patient-facing chatbot that helps schedule appointments and answer general health questions",
          "domain_description": "Healthcare and medical information services",
          "name": "Healthcare Chatbot",
          "concerns": "Should not provide medical diagnoses, prescription recommendations, or treatment advice"
        }'

{
  "id": "3c90c3cc-0d44-4b50-8888-8dd25736052a",
  "name": "<string>",
  "domain_description": "<string>",
  "app_description": "<string>",
  "concerns": "<string>",
  "created_at": "2023-11-07T05:31:56Z",
  "created_by": 123,
  "created_by_email": "<string>"
}

{
  "name": [
    "Red Team Spec with name \"my_spec\" already exists"
  ],
  "domain_description": [
    "This field is required."
  ],
  "app_description": [
    "This field is required."
  ]
}