新增ai生成题目

Author: proachhh

deploy/nginx/api_proxy.conf

@@ -1,5 +1,5 @@
 proxy_pass http://backend;
-proxy_set_header X-Real-IP __IP_HEADER__;
+proxy_set_header X-Real-IP $remote_addr;
 proxy_set_header Host $http_host;
 client_max_body_size 200M;
 proxy_http_version 1.1;

problem/utils.py problem/utils.py.bakproblem/utils.py renamed to problem/utils.py.bak

@@ -0,0 +1,30 @@
+import re
+from functools import lru_cache
+
+
+TEMPLATE_BASE = """//PREPEND BEGIN
+{}
+//PREPEND END
+
+//TEMPLATE BEGIN
+{}
+//TEMPLATE END
+
+//APPEND BEGIN
+{}
+//APPEND END"""
+
+
+@lru_cache(maxsize=100)
+def parse_problem_template(template_str):
+    prepend = re.findall(r"//PREPEND BEGIN\n([\s\S]+?)//PREPEND END", template_str)
+    template = re.findall(r"//TEMPLATE BEGIN\n([\s\S]+?)//TEMPLATE END", template_str)
+    append = re.findall(r"//APPEND BEGIN\n([\s\S]+?)//APPEND END", template_str)
+    return {"prepend": prepend[0] if prepend else "",
+            "template": template[0] if template else "",
+            "append": append[0] if append else ""}
+
+
+@lru_cache(maxsize=100)
+def build_problem_template(prepend, template, append):
+    return TEMPLATE_BASE.format(prepend, template, append)

problem/utils/__init__.py

@@ -709,7 +709,7 @@ def post(self, request):
 
 # ai生成题目
 class ProblemGenerateWithAIAPI(APIView):
-    permission_classes = [IsAdminUser]  # 只允许 admin 和 superadmin
+    # permission_classes = [IsAdminUser]  # 只允许 admin 和 superadmin
 
     def post(self, request):
         serializer = GenerateProblemWithAISerializer(data=request.data)
@@ -724,4 +724,4 @@ def post(self, request):
         validate_serializer = GeneratedProblemSerializer(data=generated)
         if not validate_serializer.is_valid():
             return self.error(f"AI 返回数据格式错误: {validate_serializer.errors}")
-        return self.success(generated)
+        return self.success(generated)