Phoenix21 commited on
Commit
3bc6f69
·
verified ·
1 Parent(s): dd52b12

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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
- if not moderation_result.is_safe:
204
- return "Sorry, this query contains harmful or inappropriate content."
205
-
206
- classification = classify_query(moderation_result.original_text)
 
 
 
 
 
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})