726 lines
35 KiB
Python
726 lines
35 KiB
Python
"""雅文内容创造工作台 - 数据模型（SQLModel / SQLite）"""
|
||
from datetime import datetime
|
||
from typing import Optional
|
||
from sqlalchemy import UniqueConstraint
|
||
from sqlmodel import SQLModel, Field, JSON, Column
|
||
|
||
def now() -> str:
|
||
return datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||
|
||
# ---------- 组织 ----------
|
||
class Organization(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
name: str = Field(index=True, unique=True)
|
||
slug: str = Field(index=True, unique=True)
|
||
enabled: bool = True
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 内容标签 ----------
|
||
class ContentTag(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
name: str = Field(index=True, unique=True) # 标签名：心法/技法…
|
||
category: str = "内容" # 大类：结构/内容/技法/产品/表现/辅助
|
||
definition: str = "" # 一句话定义
|
||
usage: str = "" # 典型用法
|
||
is_core: bool = False # ★核心
|
||
enabled: bool = True # 停用后不再出现在选项里
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "global" # private|shared|global
|
||
|
||
# ---------- 模型 Provider ----------
|
||
class ModelProvider(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
name: str
|
||
base_url: str
|
||
api_key: str = "" # 本地 SQLite 明文存储（本机单用户）
|
||
models: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
default_model: str = ""
|
||
builtin: bool = False
|
||
note: str = ""
|
||
enabled: bool = True
|
||
# Provider 默认按组织共享；保留 0 代表系统级 Provider。
|
||
organization_id: int = Field(default=1, index=True)
|
||
|
||
class ModelCatalog(SQLModel, table=True):
|
||
"""面向用户的模型目录；Provider 凭据和目录能力完全分离。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
provider_id: int = Field(index=True)
|
||
model_name: str = Field(index=True)
|
||
display_name: str = ""
|
||
capabilities: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
capability_status: str = "untested" # untested|supported|partial|blocked
|
||
rate_version: str = "v1"
|
||
input_rate: float = 0.0
|
||
output_rate: float = 0.0
|
||
enabled: bool = True
|
||
builtin: bool = False
|
||
created_at: str = Field(default_factory=now)
|
||
updated_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 系统设置 ----------
|
||
class AppSetting(SQLModel, table=True):
|
||
key: str = Field(primary_key=True)
|
||
value: str = ""
|
||
|
||
class AuditLog(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
actor_user_id: int = Field(default=0, index=True)
|
||
actor_display_name: str = ""
|
||
action: str = Field(index=True)
|
||
resource: str = Field(index=True)
|
||
result: str = "success"
|
||
created_at: str = Field(default_factory=now, index=True)
|
||
|
||
class SystemLog(SQLModel, table=True):
|
||
"""应用运行日志，与管理员操作审计严格分离。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
level: str = Field(default="INFO", index=True)
|
||
source: str = Field(default="app", index=True)
|
||
message: str = ""
|
||
trace_id: str = Field(default="", index=True)
|
||
status: str = Field(default="ok", index=True)
|
||
created_at: str = Field(default_factory=now, index=True)
|
||
|
||
# ---------- 知识库 ----------
|
||
class KnowledgeBase(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
name: str = Field(index=True, unique=True)
|
||
desc: str = ""
|
||
source_dir: str = "" # KB_DIR/<id>_<name>/
|
||
index_status: str = "empty" # empty|indexing|ready|needs_reindex|error
|
||
file_count: int = 0
|
||
chunk_count: int = 0
|
||
chunk_target: int = 0 # 分段目标长度：0=自动(500)；2200=长文聚合；自定义任意 120~8000
|
||
chunk_mode: str = "auto" # auto=按长度聚合 | heading=按标题切分
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "global" # private|shared|global
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
class KnowledgeBaseGrant(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
kb_id: int = Field(index=True)
|
||
user_id: int = Field(index=True)
|
||
permission: str = "use" # read|use|manage|export
|
||
enabled: bool = True
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
class KnowledgeCategory(SQLModel, table=True):
|
||
"""知识库内的无限层级分类树。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
kb_id: int = Field(index=True)
|
||
parent_id: Optional[int] = Field(default=None, index=True)
|
||
name: str
|
||
desc: str = ""
|
||
sort_order: int = 0
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
class KnowledgeFile(SQLModel, table=True):
|
||
"""源文件与分类的关系；物理文件仍保存在知识库目录内。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
kb_id: int = Field(index=True)
|
||
file_name: str = Field(index=True)
|
||
category_id: Optional[int] = Field(default=None, index=True)
|
||
ai_category_id: Optional[int] = Field(default=None, index=True)
|
||
classification_status: str = "unclassified" # unclassified|ai_suggested|human_confirmed
|
||
ai_confidence: float = 0.0
|
||
ai_reason: str = ""
|
||
content_hash: str = Field(default="", index=True) # 源文件内容指纹，重建索引后保持可追溯
|
||
revision: int = 1 # 同名文件内容变化时递增
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 知识片段消费记录 ----------
|
||
class KnowledgeUsage(SQLModel, table=True):
|
||
"""同一个知识片段仅对消费它的 Agent 锁定，其他 Agent 不受影响。"""
|
||
__table_args__ = (
|
||
UniqueConstraint("agent_id", "kb_id", "chunk_id", name="uq_knowledge_usage_agent_chunk"),
|
||
)
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
agent_id: int = Field(index=True)
|
||
kb_id: int = Field(index=True)
|
||
chunk_id: str = Field(index=True)
|
||
file_name: str = ""
|
||
heading: str = ""
|
||
excerpt: str = ""
|
||
campaign_id: int = 0
|
||
segment_id: int = 0
|
||
status: str = "consumed" # reserved|consumed|available
|
||
banned: bool = False # 人工审查不通过且选择"不再使用"→ 检索永久排除
|
||
used_at: str = Field(default_factory=now)
|
||
released_at: str = ""
|
||
|
||
# ---------- 技能库 ----------
|
||
class AgentSkill(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
name: str = Field(index=True, unique=True)
|
||
desc: str = ""
|
||
instructions: str = ""
|
||
enabled: bool = True
|
||
builtin: bool = False
|
||
version: int = 1
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "global"
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
class AgentSkillBinding(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
agent_id: int = Field(index=True)
|
||
skill_id: int = Field(index=True)
|
||
|
||
class AgentSkillGrant(SQLModel, table=True):
|
||
"""技能的成员授权；技能挂接到 Agent 后仍遵循技能自身的访问边界。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
skill_id: int = Field(index=True)
|
||
user_id: int = Field(index=True)
|
||
permission: str = "use" # read|use|manage|export
|
||
enabled: bool = True
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 提示模板 ----------
|
||
class AgentPromptTemplate(SQLModel, table=True):
|
||
"""可复用、可版本化的提示模板；任务创建时会冻结完整快照。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
name: str = Field(index=True, unique=True)
|
||
desc: str = ""
|
||
role_prompt: str = ""
|
||
task_template: str = ""
|
||
constraints: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
examples: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
output_schema: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
style: str = ""
|
||
version: int = 1
|
||
enabled: bool = True
|
||
builtin: bool = False
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "global"
|
||
created_at: str = Field(default_factory=now)
|
||
updated_at: str = Field(default_factory=now)
|
||
|
||
# ---------- Agent ----------
|
||
class Agent(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
name: str
|
||
role: str = "production" # production|training_map|qc|service|coordinator|composer
|
||
desc: str = ""
|
||
content_tags: list = Field(default_factory=list, sa_column=Column(JSON)) # 生产 Agent 擅长标签
|
||
kb_ids: list = Field(default_factory=list, sa_column=Column(JSON)) # 知识库挂接
|
||
kb_category_ids: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
fusion_agent_ids: list = Field(default_factory=list, sa_column=Column(JSON)) # 融合写法的生产 Agent
|
||
model_provider_id: Optional[int] = None # None=跟随全局默认
|
||
model_name: str = ""
|
||
model_catalog_id: Optional[int] = Field(default=None, index=True)
|
||
prompt_template_id: Optional[int] = Field(default=None, index=True)
|
||
# 允许该 Agent 使用的 Runtime 工具名称；空列表按系统内置工具兼容处理。
|
||
allowed_tools: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
rules: str = "" # rules.md 内容
|
||
rules_version: int = 1
|
||
persona_json: dict = Field(default_factory=dict, sa_column=Column(JSON)) # Agent 独立人格档案（优先级低于绑定的数字人）
|
||
digital_human_id: Optional[int] = Field(default=None, index=True) # 绑定数字人：绑定时以数字人身份生成
|
||
runtime_snapshot: dict = Field(default_factory=dict, sa_column=Column(JSON)) # 发布冻结的运行配置（借鉴 Coze 草稿/发布分离）
|
||
status: str = "active" # active|disabled
|
||
builtin: bool = False
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "global" # private|shared|global
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
class AgentGrant(SQLModel, table=True):
|
||
"""Agent 本身的成员授权；与 Agent 内部技能授权独立。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
agent_id: int = Field(index=True)
|
||
user_id: int = Field(index=True)
|
||
permission: str = "use" # read|use|manage|export
|
||
enabled: bool = True
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 数字人（蒸馏人格档案，可绑定到 Agent） ----------
|
||
class DigitalHuman(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
name: str = Field(index=True)
|
||
desc: str = ""
|
||
persona: dict = Field(default_factory=dict, sa_column=Column(JSON)) # 十一维人格档案
|
||
# 蒸馏质量元数据（覆盖范围、证据来源、不确定性、情境规则）；不混入 persona 字段，兼容旧档案。
|
||
persona_meta: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
status: str = "distilling" # distilling 蒸馏中 | testing 待人工测试 | active 已生效
|
||
version: int = 1 # 每次重新蒸馏 +1
|
||
source_files: list = Field(default_factory=list, sa_column=Column(JSON)) # 蒸馏来源文件名
|
||
last_feedback: str = "" # 最近一次人工测试意见
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "private" # private=仅本人 | shared=授权成员 | global=全员可用
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
class DigitalHumanGrant(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
dh_id: int = Field(index=True)
|
||
user_id: int = Field(index=True)
|
||
permission: str = "use" # read|use
|
||
enabled: bool = True
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
class WeChatBotAccount(SQLModel, table=True):
|
||
"""微信 Bot 账号连接；凭证只由独立适配器持有，数据库不保存明文。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
organization_id: int = Field(default=1, index=True)
|
||
account_name: str = ""
|
||
provider: str = "ilink" # ilink 官方通道 | wechaty 开发通道
|
||
status: str = "disconnected" # pending_qr|connected|disconnected|error
|
||
qr_code: str = "" # 短期二维码 URL/内容，不保存长期凭证
|
||
external_account_id: str = "" # 适配器返回的脱敏账号标识
|
||
agent_id: Optional[int] = Field(default=None, index=True)
|
||
credential_ref: str = "" # 适配器凭证引用（不含密钥）
|
||
bot_secret_hash: str = "" # 每个账号独立的 Bot 密钥哈希
|
||
bot_secret_hint: str = "" # 仅显示末四位
|
||
last_error: str = ""
|
||
created_at: str = Field(default_factory=now)
|
||
updated_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 画布（M2 完整实现，M1 存简版） ----------
|
||
class Canvas(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
name: str
|
||
nodes: list = Field(default_factory=list, sa_column=Column(JSON)) # [{node_id,agent_id,enabled}]
|
||
edges: list = Field(default_factory=list, sa_column=Column(JSON)) # [{from_node,to_node}]
|
||
supervisor_agent_id: Optional[int] = None # 全局总监 Agent：负责各段不跑偏，None=不启用
|
||
supervisor_mode: str = "on_drift" # always|on_drift|off —— 每段必审/疑似跑偏才审/关闭
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "global"
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
class CanvasGrant(SQLModel, table=True):
|
||
"""画布授权是 Agent、关联知识库和记忆的统一使用边界。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
canvas_id: int = Field(index=True)
|
||
user_id: int = Field(index=True)
|
||
permission: str = "use" # read|use|manage|export
|
||
enabled: bool = True
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
|
||
# ---------- DSH Planner：工作流模板与计划版本 ----------
|
||
class WorkflowTemplate(SQLModel, table=True):
|
||
"""工作台管理的工作流模板；DSH Planner 只能在模板约束内生成 DAG。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
key: str = Field(index=True, unique=True)
|
||
name: str
|
||
description: str = ""
|
||
# {mode, required_tags, optional_tags, max_segments, allow_auto_replan, risk_rules, default_kb_categories}
|
||
config: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
enabled: bool = True
|
||
builtin: bool = False
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "global" # private|shared|global
|
||
created_at: str = Field(default_factory=now)
|
||
updated_at: str = Field(default_factory=now)
|
||
|
||
|
||
class CampaignPlanVersion(SQLModel, table=True):
|
||
"""DSH Planner 产出的不可变计划版本；批准后才允许 Worker 执行。"""
|
||
__table_args__ = (
|
||
UniqueConstraint("campaign_id", "version", name="uq_campaign_plan_version"),
|
||
)
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
campaign_id: int = Field(index=True)
|
||
version: int = 1
|
||
source: str = "dsh" # dsh|manual
|
||
template_id: int = Field(default=0, index=True)
|
||
template_key: str = "default-live-course"
|
||
status: str = Field(default="draft", index=True) # draft|waiting_review|approved|rejected|superseded|failed
|
||
plan: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
validation: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
summary: str = ""
|
||
risks: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
advice: str = ""
|
||
created_by: int = 0
|
||
reviewed_by: int = 0
|
||
review_note: str = ""
|
||
created_at: str = Field(default_factory=now, index=True)
|
||
reviewed_at: str = ""
|
||
|
||
|
||
class ResourceRevision(SQLModel, table=True):
|
||
"""用户对全局资源的个人修改提案。
|
||
|
||
全局资源本体保持不变；普通用户编辑/删除时写入 pending 修订，管理员
|
||
审核后才会把内容合并到全局资源。这样同一资源可被多人使用而互不覆盖。
|
||
"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
resource_type: str = Field(index=True) # kb|agent|skill|canvas|memory|digital_human
|
||
resource_id: int = Field(index=True)
|
||
fork_resource_id: int = Field(default=0, index=True)
|
||
owner_user_id: int = Field(index=True)
|
||
organization_id: int = Field(default=1, index=True)
|
||
operation: str = "update" # update|delete
|
||
base_version: int = 1
|
||
base_fingerprint: str = ""
|
||
version: int = 1
|
||
payload: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
status: str = Field(default="pending", index=True) # pending|approved|rejected
|
||
reviewer_user_id: int = 0
|
||
review_note: str = ""
|
||
created_at: str = Field(default_factory=now)
|
||
reviewed_at: str = ""
|
||
|
||
|
||
class ResourceFork(SQLModel, table=True):
|
||
"""个人副本与全局源资源的关系，用于版本提交和冲突检测。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
resource_type: str = Field(index=True)
|
||
source_id: int = Field(index=True)
|
||
copy_id: int = Field(index=True)
|
||
owner_user_id: int = Field(index=True)
|
||
organization_id: int = Field(default=1, index=True)
|
||
base_version: int = 1
|
||
base_fingerprint: str = ""
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 生产任务 ----------
|
||
class Campaign(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
name: str # 如 2026-08-26-day5
|
||
mindmap_text: str = "" # 用户输入的导图/框架原文
|
||
objective: str = "" # 本次任务唯一业务目标
|
||
success_criteria: str = "" # 人工验收标准
|
||
canvas_snapshot: dict = Field(default_factory=dict, sa_column=Column(JSON)) # {nodes:[{node_id,agent_id,enabled}], edges:[{from,to}]}
|
||
canvas_name: str = "" # 画布名（快照来源）
|
||
duration_minutes: int = 0 # 直播总时长（分钟）：0=未指定；拆解按此分配各段时间
|
||
qc_mode: str = "full" # 质检模式：full=出稿同步质检 | lazy=先出稿后台质检 | off=不质检
|
||
previous_campaign_id: Optional[int] = None # 关联上一天任务（跨稿对账）
|
||
task_spec: list = Field(default_factory=list, sa_column=Column(JSON)) # 内容段列表
|
||
pending_confirms: list = Field(default_factory=list, sa_column=Column(JSON)) # 待确认清单
|
||
notes: list = Field(default_factory=list, sa_column=Column(JSON)) # 拆解说明/压缩决策
|
||
approved_plan_version_id: Optional[int] = None # 已人工批准的 DSH Planner 计划版本；空=旧流程兼容
|
||
status: str = "draft" # draft|planning|plan_review|plan_failed|parsed|running|qc|human_review|composing|done|failed
|
||
has_failure: bool = False # ⚠️ 有待处理失败
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "private"
|
||
created_at: str = Field(default_factory=now)
|
||
finished_at: str = ""
|
||
|
||
# ---------- 内容段 ----------
|
||
class Segment(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
campaign_id: int = Field(index=True)
|
||
node_id: str = "" # 画布节点 id（M1 用 seg-N）
|
||
agent_id: int = 0
|
||
agent_assignment_mode: str = "auto" # manual 后标签修改不再覆盖用户选择
|
||
generation_history: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
fusion_agent_ids: list = Field(default_factory=list, sa_column=Column(JSON)) # 创建任务时冻结的融合 Agent 快照
|
||
runtime_snapshot: dict = Field(default_factory=dict, sa_column=Column(JSON)) # 任务拆解时冻结的规则/技能/模型配置
|
||
content_tag: str = "" # 内容标签
|
||
title: str = ""
|
||
time_range: str = "" # 00:00-00:08
|
||
mindmap_node: str = "" # 导图节点原文
|
||
product_ref: str = "" # 产品节点（如有）
|
||
status: str = "pending" # pending|running|qc|human_review|approved|rework|failed|skipped
|
||
round: int = 1 # 第几轮
|
||
artifact_path: str = "" # 产物文件（md）
|
||
anchor_card: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
qc_reports: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
failure: dict = Field(default_factory=dict, sa_column=Column(JSON)) # 失败详情
|
||
references: list = Field(default_factory=list, sa_column=Column(JSON)) # RAG 引用来源（溯源核对）
|
||
kb_category_ids: list = Field(default_factory=list, sa_column=Column(JSON)) # 段级知识分类限定：空=用整个知识库
|
||
kb_category_recommendation: dict = Field(default_factory=dict, sa_column=Column(JSON)) # AI 推荐的真实分类及理由，需人工采用
|
||
updated_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 质检报告 ----------
|
||
class QCReport(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
segment_id: int = Field(index=True)
|
||
gate: str = "" # structure|content|style|cross
|
||
checks: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
verdict: str = "PASS" # PASS|FAIL
|
||
run_id: int = Field(default=0, index=True)
|
||
trace_id: str = Field(default="", index=True)
|
||
runtime_snapshot: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 调用日志 ----------
|
||
class LLMCall(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
run_id: int = Field(default=0, index=True)
|
||
campaign_id: int = 0
|
||
segment_id: int = 0
|
||
agent_id: int = 0
|
||
agent_name: str = ""
|
||
provider: str = ""
|
||
model: str = ""
|
||
status: str = "ok" # ok|error|rejected|timeout
|
||
error: str = ""
|
||
trace_id: str = Field(default="", index=True)
|
||
span_id: str = ""
|
||
attempt: int = 1
|
||
prompt_hash: str = ""
|
||
output_hash: str = ""
|
||
prompt_tokens: int = 0
|
||
completion_tokens: int = 0
|
||
duration_ms: int = 0
|
||
user_id: int = Field(default=0, index=True)
|
||
organization_id: int = Field(default=1, index=True)
|
||
usage_event_id: Optional[int] = Field(default=None, index=True)
|
||
billing_status: str = "unbilled" # unbilled|measured|estimated|settled|void
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 不可变用量事件与配额 ----------
|
||
class UsageEvent(SQLModel, table=True):
|
||
"""每一次模型、Embedding、OCR 或导出行为的追加式计量记录。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
user_id: int = Field(default=0, index=True)
|
||
organization_id: int = Field(default=1, index=True)
|
||
campaign_id: int = Field(default=0, index=True)
|
||
run_id: int = Field(default=0, index=True)
|
||
segment_id: int = Field(default=0, index=True)
|
||
trace_id: str = Field(default="", index=True)
|
||
llm_call_id: Optional[int] = Field(default=None, index=True)
|
||
agent_id: int = Field(default=0, index=True)
|
||
provider_id: int = Field(default=0, index=True)
|
||
provider: str = ""
|
||
model: str = ""
|
||
service_type: str = "llm" # llm|embedding|ocr|storage|export|rerank
|
||
unit: str = "token" # token|image|byte|gb_month|request
|
||
quantity: float = 0.0
|
||
input_quantity: int = 0
|
||
output_quantity: int = 0
|
||
input_rate: float = 0.0
|
||
output_rate: float = 0.0
|
||
amount_minor: int = 0 # CNY 分，避免浮点金额作为账本依据
|
||
currency: str = "CNY"
|
||
pricing_version: str = "v1"
|
||
status: str = "measured" # measured|estimated|settled|void
|
||
idempotency_key: str = Field(index=True, unique=True)
|
||
usage_metadata: dict = Field(default_factory=dict, sa_column=Column("metadata", JSON))
|
||
created_at: str = Field(default_factory=now, index=True)
|
||
|
||
|
||
class PointAccount(SQLModel, table=True):
|
||
"""用户点数余额；10 点数折算 1 Token。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
user_id: int = Field(index=True)
|
||
organization_id: int = Field(default=1, index=True)
|
||
balance_points: int = 0
|
||
updated_at: str = Field(default_factory=now)
|
||
|
||
|
||
class PointTransaction(SQLModel, table=True):
|
||
"""点数不可变流水：管理员充值为 credit，模型消耗为 debit。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
user_id: int = Field(index=True)
|
||
organization_id: int = Field(default=1, index=True)
|
||
delta_points: int
|
||
kind: str = "credit" # credit|debit|adjustment
|
||
reason: str = ""
|
||
source: str = "admin"
|
||
operator_user_id: int = 0
|
||
idempotency_key: str = Field(index=True, unique=True)
|
||
transaction_metadata: dict = Field(default_factory=dict, sa_column=Column("metadata", JSON))
|
||
created_at: str = Field(default_factory=now, index=True)
|
||
|
||
|
||
class UsageQuota(SQLModel, table=True):
|
||
"""用户级配额；0 表示该项不设上限，策略由管理员配置。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
user_id: int = Field(index=True)
|
||
organization_id: int = Field(default=1, index=True)
|
||
period: str = "monthly" # daily|monthly
|
||
token_limit: int = 0
|
||
amount_limit_minor: int = 0
|
||
storage_limit_bytes: int = 0
|
||
concurrent_limit: int = 0
|
||
warning_ratio: float = 0.8
|
||
overage_policy: str = "warn" # warn|soft_limit|hard_limit
|
||
enabled: bool = True
|
||
updated_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 通知 ----------
|
||
class Notification(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
type: str = "failure" # failure|pending_review|memory_pending|tag_pending|info
|
||
ref_id: int = 0
|
||
message: str = ""
|
||
status: str = "unread"
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
class ProgressEvent(SQLModel, table=True):
|
||
"""Cross-process progress events shared by API workbench and DSH worker."""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
user_id: int = Field(default=0, index=True)
|
||
event: str = Field(default="info", index=True)
|
||
payload: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
created_at: str = Field(default_factory=now, index=True)
|
||
|
||
# ---------- 记忆待确认池 ----------
|
||
class PendingMemory(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
target_agent_id: int
|
||
type: str = "L3" # L1规则|L2纠错|L3偏好|L4成功
|
||
content: str = ""
|
||
source_ref: str = "" # 来源：哪次打回/哪场
|
||
status: str = "pending" # pending|approved|rejected|blocked
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "private"
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 已确认记忆 ----------
|
||
class Memory(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
agent_id: int = Field(index=True)
|
||
layer: str = "L3"
|
||
title: str = ""
|
||
description: str = ""
|
||
content: str = ""
|
||
source_ref: str = ""
|
||
status: str = "active" # active|expired
|
||
privacy: str = "internal" # internal|public
|
||
tags: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
timestamp: str = Field(default_factory=now) # 记忆沉淀日期
|
||
index_status: str = "indexed" # indexed|needs_reindex
|
||
usage_count: int = 0
|
||
last_used_at: str = ""
|
||
supersedes_id: int = 0 # 人工确认替代的旧记忆；正文保留为历史版本
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "private"
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
class MemoryGrant(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
memory_id: int = Field(index=True)
|
||
user_id: int = Field(index=True)
|
||
permission: str = "use" # read|use|manage|export
|
||
enabled: bool = True
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 组合产物 ----------
|
||
class Artifact(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
campaign_id: int = Field(index=True)
|
||
docx_path: str = ""
|
||
compose_report: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
final_status: str = "composed"
|
||
run_id: int = Field(default=0, index=True)
|
||
trace_id: str = Field(default="", index=True)
|
||
runtime_snapshot: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
organization_id: int = Field(default=1, index=True)
|
||
owner_user_id: int = Field(default=0, index=True)
|
||
visibility: str = "private"
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 本地账号与会话 ----------
|
||
class AppUser(SQLModel, table=True):
|
||
"""本地工作台账号；密码只保存 PBKDF2 派生摘要。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
username: str = Field(index=True, unique=True)
|
||
display_name: str = ""
|
||
password_hash: str = ""
|
||
role: str = "viewer" # admin|editor|reviewer|viewer
|
||
organization_id: int = Field(default=1, index=True)
|
||
enabled: bool = True
|
||
created_at: str = Field(default_factory=now)
|
||
updated_at: str = Field(default_factory=now)
|
||
|
||
class RoleDefinition(SQLModel, table=True):
|
||
"""管理员可维护的角色权限集合；AppUser.role 保存角色标识。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
role: str = Field(index=True, unique=True)
|
||
label: str = ""
|
||
description: str = ""
|
||
permissions: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
builtin: bool = True
|
||
organization_id: int = Field(default=1, index=True)
|
||
updated_at: str = Field(default_factory=now)
|
||
|
||
class AuthSession(SQLModel, table=True):
|
||
"""不透明令牌的可撤销会话，数据库只保存令牌哈希。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
user_id: int = Field(index=True)
|
||
token_hash: str = Field(index=True, unique=True)
|
||
expires_at: str = ""
|
||
revoked_at: str = ""
|
||
created_at: str = Field(default_factory=now)
|
||
last_seen_at: str = Field(default_factory=now)
|
||
|
||
# ---------- 可恢复执行记录 ----------
|
||
class CampaignRun(SQLModel, table=True):
|
||
"""一次任务运行的持久化状态；服务重启后可从检查点恢复。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
campaign_id: int = Field(index=True)
|
||
user_id: int = Field(default=0, index=True)
|
||
organization_id: int = Field(default=1, index=True)
|
||
status: str = "queued" # queued|running|paused|awaiting_review|completed|failed|cancelled
|
||
cancel_requested: bool = False
|
||
pause_requested: bool = False
|
||
checkpoint: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
config_snapshot: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
trace_id: str = Field(default="", index=True)
|
||
runtime_version: str = "v1"
|
||
error: str = ""
|
||
created_at: str = Field(default_factory=now)
|
||
started_at: str = ""
|
||
finished_at: str = ""
|
||
|
||
class RunEvent(SQLModel, table=True):
|
||
"""运行审计轨迹，供前端展示生成进度与诊断失败。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
run_id: int = Field(index=True)
|
||
campaign_id: int = Field(index=True)
|
||
segment_id: int = 0
|
||
node_id: str = ""
|
||
trace_id: str = Field(default="", index=True)
|
||
span_id: str = ""
|
||
parent_span_id: str = ""
|
||
actor_agent_id: int = 0
|
||
actor_name: str = ""
|
||
tool_name: str = ""
|
||
event_type: str = "info"
|
||
status: str = ""
|
||
payload: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
created_at: str = Field(default_factory=now)
|
||
|
||
# ---------- Agent Runtime 会话 ----------
|
||
class AgentSession(SQLModel, table=True):
|
||
"""单个 Agent 的可恢复会话；messages 保存 Agent loop 的上下文和工具结果。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
run_id: int = Field(default=0, index=True)
|
||
campaign_id: int = Field(default=0, index=True)
|
||
segment_id: int = Field(default=0, index=True)
|
||
agent_id: int = Field(default=0, index=True)
|
||
agent_name: str = ""
|
||
parent_session_id: Optional[int] = Field(default=None, index=True)
|
||
phase: str = "production" # coordinator|production|qc|composer|handoff
|
||
trace_id: str = Field(default="", index=True)
|
||
status: str = "running" # running|waiting_approval|interrupted|completed|failed|cancelled
|
||
turn: int = 0
|
||
max_turns: int = 6
|
||
messages: list = Field(default_factory=list, sa_column=Column(JSON))
|
||
tool_state: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
result_text: str = ""
|
||
result_meta: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
error: str = ""
|
||
created_at: str = Field(default_factory=now)
|
||
updated_at: str = Field(default_factory=now)
|
||
finished_at: str = ""
|
||
|
||
# ---------- 持久任务队列（P0-3）----------
|
||
class TaskQueueItem(SQLModel, table=True):
|
||
"""可靠任务队列：租约 + 重试退避 + 死信。task_key 幂等防重复入队。"""
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
task_key: str = Field(index=True) # 幂等键（如 campaign-run:12）
|
||
payload: dict = Field(default_factory=dict, sa_column=Column(JSON))
|
||
status: str = Field(default="pending", index=True) # pending|running|done|dead|cancelled
|
||
attempt: int = 0
|
||
max_attempts: int = 3
|
||
worker_id: str = ""
|
||
lease_until: str = "" # 租约到期时间（ISO）
|
||
next_run_at: str = Field(default_factory=now) # 退避后可领取时间
|
||
error: str = ""
|
||
created_at: str = Field(default_factory=now)
|
||
updated_at: str = Field(default_factory=now)
|