增加评估类型枚举接口

pqai_agent_server/api_server.py

@@ -18,6 +18,7 @@ from pqai_agent.utils.db_utils import create_ai_agent_db_engine
 from pqai_agent.utils.prompt_utils import format_agent_profile, format_user_profile
 from pqai_agent_server.const import AgentApiConst
 from pqai_agent_server.const.status_enum import TestTaskStatus
+from pqai_agent_server.const.type_enum import EvaluateType
 from pqai_agent_server.dataset_service import DatasetService
 from pqai_agent_server.models import MySQLSessionManager
 from pqai_agent_server.task_server import TaskManager
@@ -620,6 +621,21 @@ def resume_test_task():
     return wrap_response(200)
 
 
+@app.route("/api/getEvaluateType", methods=["GET"])
+def get_evaluate_type():
+    """
+       获取评估类型
+       :return:
+    """
+    name_desc_list = [
+        {
+            "type": item.value,
+            "desc": item.description
+        }
+        for item in EvaluateType]
+    return wrap_response(code=200, data=name_desc_list)
+
+
 @app.route("/api/getDatasetList", methods=["GET"])
 def get_dataset_list():
     """

pqai_agent_server/const/type_enum.py

@@ -13,10 +13,34 @@ class DatasetType(Enum):
         }
         return descriptions.get(self)
 
+
 # 使用示例
 def get_dataset_type_desc(type_code):
     try:
         type = DatasetType(type_code)
         return type.description
     except ValueError:
-        return f"未知类型: {type_code}"
+        return f"未知类型: {type_code}"
+
+
+class EvaluateType(Enum):
+    REPLAY = 0
+    PUSH = 1
+
+    @property
+    def description(self):
+        descriptions = {
+            self.REPLAY: "回复",
+            self.PUSH: "唤起"
+        }
+        return descriptions.get(self)
+
+
+# 使用示例
+def get_evaluate_type_desc(type_code):
+    try:
+        type = EvaluateType(type_code)
+        return type.description
+    except ValueError:
+        return f"未知类型: {type_code}"
+