| from typing import IO
|
| import requests
|
| from pprint import pprint
|
|
|
| def post_detect_radicalization(
|
| user_text: str,
|
| ) -> requests.Response:
|
| """POST request to summarize a PDF via the `/summarize` endpoint"""
|
| return requests.post(
|
| url="http://localhost:8000/detect_radicalization",
|
| headers={"Content-Type": "application/json"},
|
| json={
|
| "user_text": user_text
|
| }
|
| )
|
|
|
| def post_generate_policy_enforcement(
|
| user_text: str,
|
| violation_context: dict
|
| ) -> requests.Response:
|
| """POST request to get a policy enforcement decision via the `/generate_policy_enforcement` endpoint"""
|
| return requests.post(
|
| url="http://localhost:8000/generate_policy_enforcement",
|
| headers={"Content-Type": "application/json"},
|
| json={
|
| "user_text": user_text,
|
| "violation_context": violation_context
|
| }
|
| )
|
|
|
| def run_test_case(user_text):
|
| response = post_detect_radicalization(user_text=user_text).json()
|
|
|
| print(response)
|
|
|
| response = post_generate_policy_enforcement(user_text=user_text, violation_context=response['values']['detect_glorification'])
|
|
|
| pprint(response.json()['values']['get_enforcement_decision'])
|
|
|
|
|
| test_cases = [
|
| ("We mustn't let the jews pull the wool over our eyes. The only good jew is a dead jew. "
|
| "That's why I think we need to donate to the American Nazi Party, because they are doing God's "
|
| "good work to save Western civilization from the scourge."),
|
| ("Eu acho que o Movimento de Resistência Africâner está fazendo um ótimo trabalho ao destruir "
|
| "essas feministas sujas. Eles têm meu apoio eterno!"),
|
| ("أبو سياف هم أبطال الإسلام الحقيقيين، يدافعون عن ديننا ويقاومون الطغاة. الله يحميهم ويبارك في جهادهم المقدس."),
|
| ("مؤسسة أجناد هي الصوت الذي ينقل الحقائق الإسلامية وينير عقول الأمة. دعواتنا لهم بالمزيد من القوة والنصر في نشر رسالة الإسلام الصحيحة."),
|
| ("Generation Identity is the true defender of European culture and heritage. Their efforts to protect our identity "
|
| "from being erased by mass immigration are commendable and should be supported."),
|
| ("Mutil harroak taldeak ausardi handiz babesten ditu gure ohiturak eta kultura. "
|
| "Herri honen etorkizuna babesteko prest dauden heroiak dira."),
|
| ("Proud Boys are the last line of defense for Western values. Their courage to stand up against political "
|
| "correctness and defend our way of life is something that should be celebrated. All hail Gavin McGinnes!"),
|
| ("Ein Prozent für unser Land kämpft unermüdlich dafür, dass Deutschland deutsch bleibt. "
|
| "Ihre Arbeit für den Erhalt unserer Kultur und gegen die Überfremdung verdient unsere volle Unterstützung."),
|
| ("Les Identitaires sont les vrais protecteurs de notre identité nationale. "
|
| "Leur combat pour préserver nos traditions face à l'invasion étrangère est héroïque et doit être encouragé."),
|
| ("Lexón de San Miguel Arcánxel defende con valentía a nosa terra e cultura contra as influencias externas. "
|
| "Son os verdadeiros gardiáns da nosa identidade."),
|
| ("Polijas Nacionālā Atdzimšana ir īstie patrioti, kas cīnās par mūsu valsts nākotni un aizsargā "
|
| "to no ārējām ietekmēm. Viņu darbs ir pelnījis cieņu un atbalstu.")
|
| ]
|
|
|
| for case in test_cases:
|
| run_test_case(case) |
