50b1c3f9a9
后端新增规则资产版本管理和规则文件 CRUD 接口，优化风险 规则生成模板执行和员工数据模型字段，知识库 RAG 增强本 地回退和文档提取能力，清理旧风险规则文件统一由生成引擎 管理，前端审计页面增加运行时调试面板和规则资产编辑交互， 补充单元测试覆盖。
106 lines
4.8 KiB
Python
106 lines
4.8 KiB
Python
from __future__ import annotations
|
|
|
|
import uuid
|
|
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from sqlalchemy import Boolean, DateTime, ForeignKey, String, Text, UniqueConstraint, func
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
from sqlalchemy.types import JSON
|
|
|
|
from app.db.base_class import Base
|
|
|
|
|
|
class AgentAsset(Base):
|
|
__tablename__ = "agent_assets"
|
|
|
|
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
|
|
asset_type: Mapped[str] = mapped_column(String(20), index=True)
|
|
code: Mapped[str] = mapped_column(String(100), unique=True, index=True)
|
|
name: Mapped[str] = mapped_column(String(200))
|
|
description: Mapped[str] = mapped_column(Text(), default="")
|
|
domain: Mapped[str] = mapped_column(String(50), index=True)
|
|
scenario_json: Mapped[list[Any]] = mapped_column(JSON, default=list)
|
|
owner: Mapped[str] = mapped_column(String(100))
|
|
reviewer: Mapped[str | None] = mapped_column(String(100), nullable=True)
|
|
status: Mapped[str] = mapped_column(String(20), index=True, default="draft")
|
|
current_version: Mapped[str | None] = mapped_column(String(30), nullable=True)
|
|
published_version: Mapped[str | None] = mapped_column(String(30), nullable=True)
|
|
working_version: Mapped[str | None] = mapped_column(String(30), nullable=True)
|
|
config_json: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
|
|
)
|
|
|
|
versions = relationship(
|
|
"AgentAssetVersion",
|
|
back_populates="asset",
|
|
cascade="all, delete-orphan",
|
|
order_by="desc(AgentAssetVersion.created_at)",
|
|
)
|
|
reviews = relationship(
|
|
"AgentAssetReview",
|
|
back_populates="asset",
|
|
cascade="all, delete-orphan",
|
|
order_by="desc(AgentAssetReview.created_at)",
|
|
)
|
|
scheduled_runs = relationship("AgentRun", back_populates="task_asset")
|
|
test_runs = relationship(
|
|
"AgentAssetTestRun",
|
|
back_populates="asset",
|
|
cascade="all, delete-orphan",
|
|
order_by="desc(AgentAssetTestRun.created_at)",
|
|
)
|
|
|
|
|
|
class AgentAssetVersion(Base):
|
|
__tablename__ = "agent_asset_versions"
|
|
__table_args__ = (
|
|
UniqueConstraint("asset_id", "version", name="uq_agent_asset_versions_asset_version"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
|
|
asset_id: Mapped[str] = mapped_column(ForeignKey("agent_assets.id"), index=True)
|
|
version: Mapped[str] = mapped_column(String(30))
|
|
content: Mapped[str] = mapped_column(Text())
|
|
content_type: Mapped[str] = mapped_column(String(20))
|
|
change_note: Mapped[str | None] = mapped_column(Text(), nullable=True)
|
|
created_by: Mapped[str] = mapped_column(String(100))
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
asset = relationship("AgentAsset", back_populates="versions")
|
|
|
|
|
|
class AgentAssetReview(Base):
|
|
__tablename__ = "agent_asset_reviews"
|
|
|
|
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
|
|
asset_id: Mapped[str] = mapped_column(ForeignKey("agent_assets.id"), index=True)
|
|
version: Mapped[str] = mapped_column(String(30))
|
|
reviewer: Mapped[str] = mapped_column(String(100))
|
|
review_status: Mapped[str] = mapped_column(String(20), index=True)
|
|
review_note: Mapped[str | None] = mapped_column(Text(), nullable=True)
|
|
reviewed_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
asset = relationship("AgentAsset", back_populates="reviews")
|
|
|
|
|
|
class AgentAssetTestRun(Base):
|
|
__tablename__ = "agent_asset_test_runs"
|
|
|
|
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
|
|
asset_id: Mapped[str] = mapped_column(ForeignKey("agent_assets.id"), index=True)
|
|
version: Mapped[str] = mapped_column(String(30), index=True)
|
|
test_type: Mapped[str] = mapped_column(String(30), index=True)
|
|
status: Mapped[str] = mapped_column(String(20), index=True)
|
|
passed: Mapped[bool] = mapped_column(Boolean, default=False, index=True)
|
|
summary: Mapped[str] = mapped_column(Text(), default="")
|
|
input_json: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict)
|
|
result_json: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict)
|
|
created_by: Mapped[str] = mapped_column(String(100))
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
asset = relationship("AgentAsset", back_populates="test_runs")
|