50 lines
2.2 KiB
Python
50 lines
2.2 KiB
Python
"""AI worker 执行账本。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Optional
|
|
|
|
from sqlalchemy import BigInteger, ForeignKey, Integer, Text, UniqueConstraint, VARCHAR, TIMESTAMP
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from .base import Base
|
|
from .types import JSONB
|
|
|
|
|
|
class AIWorkerExecution(Base):
|
|
__tablename__ = "ai_worker_executions"
|
|
__table_args__ = (
|
|
UniqueConstraint(
|
|
"tenant_id",
|
|
"trace_id",
|
|
"worker_role_key",
|
|
name="uq_ai_worker_execution_trace_role",
|
|
),
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
project_id: Mapped[int] = mapped_column(
|
|
BigInteger, ForeignKey("projects.id"), nullable=False, index=True
|
|
)
|
|
signal_id: Mapped[Optional[int]] = mapped_column(
|
|
BigInteger, ForeignKey("signals.id"), nullable=True, index=True
|
|
)
|
|
signal_source_registry_id: Mapped[Optional[int]] = mapped_column(
|
|
BigInteger, ForeignKey("signal_source_registries.id"), nullable=True, index=True
|
|
)
|
|
operator_user_id: Mapped[Optional[int]] = mapped_column(
|
|
BigInteger, ForeignKey("users.id"), nullable=True, index=True
|
|
)
|
|
trace_id: Mapped[str] = mapped_column(VARCHAR(64), nullable=False, index=True)
|
|
operation_trace_id: Mapped[Optional[str]] = mapped_column(VARCHAR(64), nullable=True, index=True)
|
|
execution_type: Mapped[str] = mapped_column(VARCHAR(40), nullable=False, default="signal_source_onboarding", index=True)
|
|
worker_role_key: Mapped[str] = mapped_column(VARCHAR(32), nullable=False, index=True)
|
|
worker_role_name: Mapped[str] = mapped_column(VARCHAR(64), nullable=False)
|
|
status: Mapped[str] = mapped_column(VARCHAR(20), nullable=False, default="planned", index=True)
|
|
summary: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
|
input_payload: Mapped[Optional[dict]] = mapped_column(JSONB, nullable=True)
|
|
output_payload: Mapped[Optional[dict]] = mapped_column(JSONB, nullable=True)
|
|
error_message: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
|
started_at: Mapped[Optional[TIMESTAMP]] = mapped_column(TIMESTAMP, nullable=True)
|
|
completed_at: Mapped[Optional[TIMESTAMP]] = mapped_column(TIMESTAMP, nullable=True)
|