aimoderationproject / chatgptmoderationapptest.py
lg3394's picture
Update chatgptmoderationapptest.py
8c9240e verified
raw
history blame contribute delete
808 Bytes
import os
from openai import OpenAI
# Initialize the client with your API key
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
# Test a moderation request
moderation = client.moderations.create(input="I want to kill them.")
print(moderation)
# Print more detailed information
print("\nModeration results:")
print(f"Flagged: {moderation.results[0].flagged}")
print("\nCategories:")
for category_name, flagged in vars(moderation.results[0].categories).items():
if not category_name.startswith('_'): # Skip private attributes
print(f"- {category_name}: {flagged}")
print("\nCategory scores:")
for score_name, score_value in vars(moderation.results[0].category_scores).items():
if not score_name.startswith('_'): # Skip private attributes
print(f"- {score_name}: {score_value}")