Spaces:
Sleeping
Sleeping
Update pipeline.py
Browse files- pipeline.py +9 -4
pipeline.py
CHANGED
|
@@ -200,10 +200,15 @@ def run_pipeline(query: str) -> str:
|
|
| 200 |
query = sanitize_message(query)
|
| 201 |
topic=extract_main_topic(query)
|
| 202 |
moderation_result = moderate_text(query)
|
| 203 |
-
|
| 204 |
-
|
| 205 |
-
|
| 206 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 207 |
|
| 208 |
if classification == "OutOfScope":
|
| 209 |
refusal_text = refusal_chain.run({"topic": topic})
|
|
|
|
| 200 |
query = sanitize_message(query)
|
| 201 |
topic=extract_main_topic(query)
|
| 202 |
moderation_result = moderate_text(query)
|
| 203 |
+
try:
|
| 204 |
+
if not moderation_result.is_safe:
|
| 205 |
+
return "Sorry, this query contains harmful or inappropriate content."
|
| 206 |
+
except Exception as e:
|
| 207 |
+
raise RuntimeError(f"Error in run_runpipeline check moderation: {str(e)}")
|
| 208 |
+
try:
|
| 209 |
+
classification = classify_query(moderation_result.original_text)
|
| 210 |
+
except Exception as e:
|
| 211 |
+
raise RuntimeError(f"Error in run_runpipeline check classify_query: {str(e)}")
|
| 212 |
|
| 213 |
if classification == "OutOfScope":
|
| 214 |
refusal_text = refusal_chain.run({"topic": topic})
|