From e7e3202714e9f929b9236966dd48fc8f45df22ee Mon Sep 17 00:00:00 2001
From: bruce <sunzhiye01@outlook.com>
Date: Thu, 4 Jun 2026 01:32:35 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=94=AF=E6=8C=81=E5=A4=84=E7=90=86?=
 =?UTF-8?q?=E5=8E=86=E5=8F=B2=E6=8C=89=E9=A3=8E=E9=99=A9=E7=8A=B6=E6=80=81?=
 =?UTF-8?q?=E7=AD=9B=E9=80=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/audit/services.py        |  1 +
 apps/audit/views.py           |  9 +++++++++
 templates/audit/log_list.html | 13 +++++++++++-
 tests/test_audit.py           | 38 +++++++++++++++++++++++++++++++++++
 4 files changed, 60 insertions(+), 1 deletion(-)

diff --git a/apps/audit/services.py b/apps/audit/services.py
index 93b4ed4..fcabe6d 100644
--- a/apps/audit/services.py
+++ b/apps/audit/services.py
@@ -117,6 +117,7 @@ def build_history_rows(logs) -> list[dict]:
                 or structured_output.get("risk_level")
                 or "-",
                 "notify_status": notification.message_status if notification else "-",
+                "notify_reason": notification.notify_reason if notification else "-",
             }
         )
     return rows
diff --git a/apps/audit/views.py b/apps/audit/views.py
index 294ff1e..1f23da8 100644
--- a/apps/audit/views.py
+++ b/apps/audit/views.py
@@ -10,6 +10,7 @@ def log_list(request):
     scenario_id = (request.GET.get("scenario_id") or "").strip()
     keyword = (request.GET.get("keyword") or "").strip()
     notify_status = (request.GET.get("notify_status") or "").strip()
+    risk_status = (request.GET.get("risk_status") or "").strip()
     logs = AgentAuditLog.objects.all()
     if scenario_id:
         logs = logs.filter(scenario_id=scenario_id)
@@ -27,6 +28,13 @@ def log_list(request):
             for log in logs
             if (log.batch_id, log.conversation_id) in matched_pairs
         ]
+    if risk_status:
+        logs = [
+            log
+            for log in logs
+            if (log.structured_output or {}).get("highest_risk_level") == risk_status
+            or (log.structured_output or {}).get("risk_level") == risk_status
+        ]
     return render(
         request,
         "audit/log_list.html",
@@ -35,6 +43,7 @@ def log_list(request):
             "selected_scenario_id": scenario_id,
             "keyword": keyword,
             "notify_status": notify_status,
+            "risk_status": risk_status,
         },
     )
 
diff --git a/templates/audit/log_list.html b/templates/audit/log_list.html
index ecdbda6..db3cdb2 100644
--- a/templates/audit/log_list.html
+++ b/templates/audit/log_list.html
@@ -21,6 +21,15 @@
         <label for="id_keyword">产品名称 / 批次号</label>
         <input id="id_keyword" type="text" name="keyword" value="{{ keyword }}" placeholder="例如：新型冠状病毒 或 SUB-20260604-001">
       </div>
+      <div>
+        <label for="id_risk_status">风险状态</label>
+        <select id="id_risk_status" name="risk_status">
+          <option value="">全部风险</option>
+          <option value="high"{% if risk_status == "high" %} selected{% endif %}>high</option>
+          <option value="medium"{% if risk_status == "medium" %} selected{% endif %}>medium</option>
+          <option value="low"{% if risk_status == "low" %} selected{% endif %}>low</option>
+        </select>
+      </div>
       <div>
         <label for="id_notify_status">通知状态</label>
         <select id="id_notify_status" name="notify_status">
@@ -55,6 +64,7 @@
             <th>输入摘要</th>
             <th>状态</th>
             <th>风险状态</th>
+            <th>通知原因</th>
             <th>通知状态</th>
             <th>模型</th>
             <th>时间</th>
@@ -74,13 +84,14 @@
                 <span class="pill {% if row.log.status == 'success' %}pill-success{% else %}pill-danger{% endif %}">{{ row.log.get_status_display_text }}</span>
               </td>
               <td>{{ row.risk_status }}</td>
+              <td>{{ row.notify_reason }}</td>
               <td>{{ row.notify_status }}</td>
               <td>{{ row.log.model_name }}</td>
               <td>{{ row.log.created_at|date:"Y-m-d H:i" }}</td>
               <td><a class="button" href="{% url 'audit:detail' row.log.id %}">查看详情</a></td>
             </tr>
           {% empty %}
-            <tr><td colspan="12">暂无处理历史，先去执行一次审核任务。</td></tr>
+            <tr><td colspan="13">暂无处理历史，先去执行一次审核任务。</td></tr>
           {% endfor %}
         </tbody>
       </table>
diff --git a/tests/test_audit.py b/tests/test_audit.py
index 9ffe012..be7f2b1 100644
--- a/tests/test_audit.py
+++ b/tests/test_audit.py
@@ -232,6 +232,8 @@ def test_audit_list_shows_risk_and_notification_status(client, db):
     assert "high" in content
     assert "通知状态" in content
     assert "sent" in content
+    assert "通知原因" in content
+    assert "task_completed" in content
 
 
 def test_audit_list_can_filter_by_notification_status(client, db):
@@ -286,6 +288,42 @@ def test_audit_list_can_filter_by_notification_status(client, db):
     assert "产品A" not in content
 
 
+def test_audit_list_can_filter_by_risk_status(client, db):
+    create_audit_log(
+        "document_review",
+        "注册审核智能体",
+        "问题一",
+        AgentResult(
+            answer="回答一",
+            status="success",
+            structured_output={"highest_risk_level": "high"},
+        ),
+        batch_id="SUB-20260604-001",
+        conversation_id="conv-001",
+        product_name="产品A",
+    )
+    create_audit_log(
+        "document_review",
+        "注册审核智能体",
+        "问题二",
+        AgentResult(
+            answer="回答二",
+            status="success",
+            structured_output={"highest_risk_level": "low"},
+        ),
+        batch_id="SUB-20260604-002",
+        conversation_id="conv-002",
+        product_name="产品B",
+    )
+
+    response = client.get(reverse("audit:list"), {"risk_status": "high"})
+
+    content = response.content.decode("utf-8")
+    assert response.status_code == 200
+    assert "产品A" in content
+    assert "产品B" not in content
+
+
 def test_audit_detail_page_shows_conversation_node_results(client, db):
     Conversation.objects.create(
         conversation_id="conv-001",