Adding params.

philterd · Oct 18, 2023 · eed495b · eed495b
1 parent b830147
commit eed495b
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/philter.py b/philter.py
@@ -1,19 +1,21 @@
 import requests
 
 
-def filter(philter_endpoint, text, verify=True):
+def filter(philter_endpoint, text, policy="default", verify=True):
     headers = {"Content-Type": "text/plain"}
-    response = requests.post(philter_endpoint + "/api/filter", headers=headers, data=text, verify=verify)
+    params = {"p": policy}
+    response = requests.post(philter_endpoint + "/api/filter", params=params, headers=headers, data=text, verify=verify)
 
     if response.status_code == 200:
         return response.text
     else:
         raise Exception("Error response received from Philter.")
 
 
-def explain(philter_endpoint, text, verify=True):
+def explain(philter_endpoint, text, policy="default", verify=True):
     headers = {"Content-Type": "text/plain"}
-    response = requests.post(philter_endpoint + "/api/explain", headers=headers, data=text, verify=verify)
+    params = {"p": policy}
+    response = requests.post(philter_endpoint + "/api/explain", params=params, headers=headers, data=text, verify=verify)
 
     if response.status_code == 200:
         return response.json()