添加异常捕获

rag2_0/dify/test_dify_chatapi.py

@@ -57,7 +57,7 @@ class DifyComparisonTester:
         with ThreadPoolExecutor(max_workers=2) as executor:
             future_old = executor.submit(get_old_answer)
             future_new = executor.submit(get_new_answer)
-            
+            try:
                 old_result = future_old.result()
                 new_result = future_new.result()
                 old_message_id = old_result["message_id"]
@@ -70,7 +70,8 @@ class DifyComparisonTester:
                         rewrite_query = outputs["optimize_query"]
                 old_answer = old_result["answer"]
                 new_answer = new_result["answer"]
-            
+            except Exception as e:
+                return None
         return {"问题": q, "问题改写": rewrite_query, "旧流程答案": old_answer, "新流程答案": new_answer}
         
     def run_comparison(self):
@@ -89,6 +90,7 @@ class DifyComparisonTester:
         with tqdm(total=len(questions), desc="处理问题进度") as pbar:
             for q in questions:
                 result = self.process_question(q)
+                if result is not None:
                     results.append(result)
                 pbar.update(1)