From 73aee622c79b1ec436cf29f5f7237a730672c548 Mon Sep 17 00:00:00 2001 From: caoxiaozhu Date: Fri, 3 Jul 2026 14:38:14 +0800 Subject: [PATCH] =?UTF-8?q?feat(flywheel):=20=E6=96=B0=E5=A2=9E=20GoldenCa?= =?UTF-8?q?se=20=E6=A8=A1=E5=9E=8B=E7=94=A8=E4=BA=8E=E8=A7=84=E5=88=99?= =?UTF-8?q?=E5=9B=9E=E5=BD=92=E9=97=A8=E7=A6=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 新增 golden_cases 表,承载版本化的风险规则黄金用例 (case_key/rule_code/values_json/expected_hit/expected_severity/status) - 注册到 db/base.py 和 models/__init__.py,进入 Base.metadata --- server/src/app/db/base.py | 2 ++ server/src/app/models/__init__.py | 2 ++ server/src/app/models/golden_case.py | 48 ++++++++++++++++++++++++++++ 3 files changed, 52 insertions(+) create mode 100644 server/src/app/models/golden_case.py diff --git a/server/src/app/db/base.py b/server/src/app/db/base.py index b9ee3c7..c1a9861 100644 --- a/server/src/app/db/base.py +++ b/server/src/app/db/base.py @@ -21,6 +21,7 @@ from app.models.financial_record import ( ExpenseClaim, ExpenseClaimItem, ) +from app.models.golden_case import GoldenCase from app.models.hermes_config import HermesTaskConfig, HermesTaskExecutionLog from app.models.hermes_report import HermesRiskReport from app.models.notification_state import NotificationState @@ -58,6 +59,7 @@ __all__ = [ "EmployeeChangeLog", "ExpenseClaim", "ExpenseClaimItem", + "GoldenCase", "HermesTaskConfig", "HermesTaskExecutionLog", "HermesRiskReport", diff --git a/server/src/app/models/__init__.py b/server/src/app/models/__init__.py index b3549eb..135aa59 100644 --- a/server/src/app/models/__init__.py +++ b/server/src/app/models/__init__.py @@ -14,6 +14,7 @@ from app.models.financial_record import ( ExpenseClaim, ExpenseClaimItem, ) +from app.models.golden_case import GoldenCase from app.models.hermes_config import HermesTaskConfig, HermesTaskExecutionLog from app.models.hermes_report import HermesRiskReport from app.models.notification_state import NotificationState @@ -49,6 +50,7 @@ __all__ = [ "EmployeeChangeLog", "ExpenseClaim", "ExpenseClaimItem", + "GoldenCase", "HermesTaskConfig", "HermesTaskExecutionLog", "HermesRiskReport", diff --git a/server/src/app/models/golden_case.py b/server/src/app/models/golden_case.py new file mode 100644 index 0000000..4a06348 --- /dev/null +++ b/server/src/app/models/golden_case.py @@ -0,0 +1,48 @@ +from __future__ import annotations + +import uuid +from datetime import datetime +from typing import Any + +from sqlalchemy import Boolean, DateTime, Index, String, Text, func +from sqlalchemy.orm import Mapped, mapped_column +from sqlalchemy.types import JSON + +from app.db.base_class import Base + + +class GoldenCase(Base): + """风险规则回归门禁用的黄金用例。 + + 由运营手动维护(或从已确认风险观测导入),在规则发布前作为回归集执行, + 100% 通过才放行。``values_json`` 复用 ``AgentAssetRiskRuleSampleCase.values`` + 的扁平字典格式,``expected_hit`` / ``expected_severity`` 作为 ground truth。 + """ + + __tablename__ = "golden_cases" + __table_args__ = ( + Index("ix_golden_cases_rule_code_status", "rule_code", "status"), + Index("ix_golden_cases_scene_status", "scene", "status"), + ) + + id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4())) + case_key: Mapped[str] = mapped_column(String(160), unique=True, index=True) + rule_code: Mapped[str | None] = mapped_column(String(120), nullable=True, index=True) + scene: Mapped[str] = mapped_column(String(50), default="", index=True) + + name: Mapped[str] = mapped_column(String(120), default="") + values_json: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict) + expected_hit: Mapped[bool] = mapped_column(Boolean, default=True) + expected_severity: Mapped[str | None] = mapped_column(String(20), nullable=True) + note: Mapped[str | None] = mapped_column(Text(), nullable=True) + + status: Mapped[str] = mapped_column(String(20), default="active", index=True) + source: Mapped[str] = mapped_column(String(30), default="manual") + + created_at: Mapped[datetime] = mapped_column(DateTime, default=func.now(), server_default=func.now()) + updated_at: Mapped[datetime] = mapped_column( + DateTime, + default=func.now(), + onupdate=func.now(), + server_default=func.now(), + )