requirements update16

app.py

@@ -5,7 +5,7 @@ import os
 import sys
 from pathlib import Path
 import concurrent.futures
-import multiprocessing  # 添加multiprocessing模块以获取CPU核心数
+import multiprocessing
 
 # 添加当前目录和src目录到模块搜索路径
 current_dir = os.path.dirname(os.path.abspath(__file__))
@@ -25,18 +25,13 @@ def evaluate(input_data):
         list: 包含评估结果的列表
     """
     try:
-        # 确保输入是列表
         if not isinstance(input_data, list):
             return {"status": "Exception", "error": "Input must be a list"}
             
         results = []
-        # 获取CPU核心数作为最大线程数
         max_workers = multiprocessing.cpu_count()
-        # 使用线程池并行处理多个测试用例，限制最大线程数为CPU核心数
         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}
-            # 获取结果
             for future in concurrent.futures.as_completed(future_to_item):
                 item = future_to_item[future]
                 try:
@@ -61,7 +56,6 @@ def evaluate_single_case(input_data):
         dict: 包含评估结果的字典
     """
     try:
-        # 只处理字典类型输入
         if not isinstance(input_data, dict):
             return {"status": "Exception", "error": "Input item must be a dictionary"}
             
@@ -71,19 +65,16 @@ def evaluate_single_case(input_data):
         if not completions:
             return {"status": "Exception", "error": "No code provided"}
 
-        # 评估所有完成的代码
         results = []
         for comp in completions:
             code = input_data.get('prompt') + comp + '\n' + input_data.get('tests')
             result = evaluate_code(code, language)
-            # 如果当前代码执行成功，立即返回pass，不再评估后续代码
             if result["status"] == "OK":
                 return {"status": "pass", "compiled_code": code}
             print(f'Code failed to compile: \n{code}')
             result["compiled_code"] = code
             results.append(result)
             
-        # 所有代码都执行失败，返回第一个失败结果
         return results[0]
                 
     except Exception as e:
@@ -100,22 +91,18 @@ def evaluate_code(code, language):
         dict: 包含评估结果的字典
     """
     try:
-        # 动态导入对应语言的评估模块
+        language = language.split('.')[-1] if '.' in language else language  # just for go
         module_name = f"src.eval_{language.lower()}"
         module = importlib.import_module(module_name)
 
-        # 使用系统临时目录而不是固定的temp目录
         import tempfile
         
-        # 创建临时文件
         with tempfile.NamedTemporaryFile(suffix=f".{language}", delete=False) as temp_file:
             temp_file_path = temp_file.name
             temp_file.write(code.encode('utf-8'))
         
-        # 调用对应语言的评估函数
         result = module.eval_script(temp_file_path)
 
-        # 清理临时文件
         if os.path.exists(temp_file_path):
             os.unlink(temp_file_path)