requirements update26

app.py

@@ -30,6 +30,12 @@ def evaluate(input_data):
             return {"status": "Exception", "error": "Input must be a list"}
             
         results = []
+        # 定义系统错误关键词，用于判断是否需要重试
+        system_error_keywords = [
+            "resource", "timeout", "busy", "congestion", "memory", 
+            "connection", "system", "overload", "refused", "reset"
+        ]
+        
         max_workers = multiprocessing.cpu_count()
         with concurrent.futures.ThreadPoolExecutor(max_workers=max_workers) as executor:
             future_to_item = {executor.submit(evaluate_single_case, item): item for item in input_data}
@@ -37,9 +43,72 @@ def evaluate(input_data):
                 item = future_to_item[future]
                 try:
                     result = future.result()
+                    
+                    # 检查是否是系统错误，如果是，立即重试一次
+                    if isinstance(result, dict) and result.get("status") == "Exception":
+                        error_msg = str(result.get("error", "")).lower()
+                        
+                        # 如果错误信息包含系统错误关键词，则重试
+                        if any(keyword in error_msg for keyword in system_error_keywords):
+                            print(f"检测到系统错误: {error_msg}，正在重试...")
+                            # 立即重试
+                            retry_result = evaluate_single_case(item)
+                            if isinstance(retry_result, dict) and retry_result.get("status") != "Exception":
+                                # 重试成功，使用重试结果
+                                result = retry_result
+                                print(f"重试成功")
+                            else:
+                                print(f"重试失败")
+                    
+                    # 检查结果列表
+                    if isinstance(result, list):
+                        for i, res in enumerate(result):
+                            if isinstance(res, dict) and res.get("status") == "Exception":
+                                error_msg = str(res.get("error", "")).lower()
+                                
+                                # 如果错误信息包含系统错误关键词，则重试
+                                if any(keyword in error_msg for keyword in system_error_keywords):
+                                    print(f"检测到列表中的系统错误: {error_msg}，正在重试...")
+                                    # 仅重试这个失败的情况
+                                    code = item.get('prompt') + item.get('processed_completions', [])[i] + '\n' + item.get('tests')
+                                    retry_result = evaluate_code(code, item.get('language'))
+                                    if isinstance(retry_result, dict) and retry_result.get("status") != "Exception":
+                                        # 重试成功，更新结果
+                                        result[i] = retry_result
+                                        print(f"重试成功")
+                                    else:
+                                        print(f"重试失败")
+                    
+                    # 如果是超时错误，也尝试重试一次
+                    if isinstance(result, dict) and result.get("status") == "Timeout":
+                        print(f"检测到超时错误，正在重试...")
+                        # 立即重试
+                        retry_result = evaluate_single_case(item)
+                        if isinstance(retry_result, dict) and retry_result.get("status") != "Timeout":
+                            # 重试成功，使用重试结果
+                            result = retry_result
+                            print(f"重试成功")
+                        else:
+                            print(f"重试失败")
+                    
                     item.update(result)
                     results.append(item)
                 except Exception as e:
+                    error_msg = str(e).lower()
+                    # 检查是否是系统错误
+                    if any(keyword in error_msg for keyword in system_error_keywords):
+                        print(f"执行过程中检测到系统错误: {error_msg}，正在重试...")
+                        try:
+                            # 立即重试
+                            retry_result = evaluate_single_case(item)
+                            item.update(retry_result)
+                            results.append(item)
+                            print(f"重试成功")
+                            continue
+                        except Exception as retry_e:
+                            print(f"重试失败: {str(retry_e)}")
+                    
+                    # 如果重试失败或不是系统错误，记录原始错误
                     item.update({"status": "Exception", "error": str(e)})
                     results.append(item)
         return results
@@ -95,7 +164,6 @@ def evaluate_code(code, language):
         return result
 
     except Exception as e:
-        print(1)
         return {"status": "Exception", "error": str(e)}
 
 # 创建Gradio接口