Merge branch 'main' of https://huggingface.co/spaces/dongsheng/docker_test

app.py

@@ -71,9 +71,7 @@ def evaluate_single_case(input_data):
             code = input_data.get('prompt') + comp + '\n' + input_data.get('tests')
             result = evaluate_code(code, language)
             if result["status"] == "OK":
-                return {"status": "pass", "compiled_code": code}
-            print(f'Code failed to compile: \n{code}')
-            result["compiled_code"] = code
+                return result
             results.append(result)
             
         return results[0]
@@ -94,17 +92,10 @@ def evaluate_code(code, language):
     try:
         # 使用containerized_eval中的eval_string_script函数
         result = eval_string_script(language, code)
-        
-        if result["exit_code"] == 0:
-            return {"status": "OK", "output": result["stdout"]}
-        else:
-            return {
-                "status": "Error",
-                "error": result["stderr"] if result["stderr"] else "Unknown error",
-                "output": result["stdout"]
-            }
+        return result
 
     except Exception as e:
+        print(1)
         return {"status": "Exception", "error": str(e)}
 
 # 创建Gradio接口

src/containerized_eval.py

@@ -24,6 +24,7 @@ from . import eval_clj
 from . import eval_v
 from . import eval_lean
 from . import eval_dart
+from . import eval_go
 import tempfile
 
 
@@ -60,6 +61,8 @@ EVALUATORS = {
     "coq": (eval_v.eval_script, ".v"),
     "lean": (eval_lean.eval_script, ".lean"),
     "dart": (eval_dart.eval_script, ".dart"),
+    "go": (eval_go.eval_script, ".go"),
+    "go_test.go": (eval_go.eval_script, "_test.go"),
 }
 
 def eval_string_script(language, program):

src/eval_go.py

@@ -2,6 +2,8 @@ import argparse
 from sys import exit
 import subprocess
 from pathlib import Path
+import os
+import tempfile
 from .generic_eval import main as gmain
 
 
@@ -11,22 +13,30 @@ def eval_script(path: Path):
     stderr = None
     exit_code = None
     try:
-        build = subprocess.run(["go", "test", path],
-                               timeout=30,
-                               stdout=subprocess.PIPE,
-                               stderr=subprocess.PIPE)
+        # 创建临时目录用于Go缓存
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # 设置Go环境变量
+            env = os.environ.copy()
+            env["GOCACHE"] = os.path.join(temp_dir, "go-build")
+            env["GOPATH"] = os.path.join(temp_dir, "gopath")
+            
+            build = subprocess.run(["go", "test", path],
+                                env=env,
+                                timeout=30,
+                                stdout=subprocess.PIPE,
+                                stderr=subprocess.PIPE)
 
-        stdout = build.stdout.decode("utf-8", errors="ignore")
-        stderr = build.stderr.decode("utf-8", errors="ignore")
-        exit_code = build.returncode
-        # write to stderr just so that we can redirect stdout to a csv
+            stdout = build.stdout.decode("utf-8", errors="ignore")
+            stderr = build.stderr.decode("utf-8", errors="ignore")
+            exit_code = build.returncode
+            # write to stderr just so that we can redirect stdout to a csv
 
-        if "[setup failed]" in stdout or "[build failed]" in stdout:
-            status = "SyntaxError"
-        elif "FAIL" in stdout:
-            status = "Exception"
-        else:
-            status = "OK"
+            if "[setup failed]" in stdout or "[build failed]" in stdout:
+                status = "SyntaxError"
+            elif "FAIL" in stdout:
+                status = "Exception"
+            else:
+                status = "OK"
     except subprocess.TimeoutExpired:
         status = "Timeout"