198 lines
9.8 KiB
Python
198 lines
9.8 KiB
Python
"""司命台治理对象模型（九枢治理视图，api-spec §19.4）。
|
||
|
||
目标（太一）/ 感知通道（烛照）/ 世界实体（河图）/ 计划（洛书）/ 天工阁角色
|
||
为可增长的开放集合；生成偏好（模型网关 §4.3）为本地单用户单行配置。
|
||
这些对象当前由司命台治理视图维护，对应枢的自动运行时暂未启用（见
|
||
docs/operations/capability-matrix.md §8）。
|
||
"""
|
||
|
||
from datetime import datetime
|
||
|
||
from sqlalchemy import (
|
||
Boolean,
|
||
DateTime,
|
||
Float,
|
||
ForeignKey,
|
||
Integer,
|
||
String,
|
||
Text,
|
||
)
|
||
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
||
from app.core.config import settings
|
||
from app.core.ids import new_goal_id, new_plan_id, new_world_entity_id
|
||
from app.core.time import utc_now
|
||
from app.db.base import Base
|
||
|
||
|
||
class Goal(Base):
|
||
"""太一枢目标节点（契约字典 §5.11 / §5.12）。"""
|
||
|
||
__tablename__ = "goals"
|
||
|
||
def __init__(self, **kwargs: object) -> None:
|
||
# Python 端默认值：mapped_column(default=...) 仅在 ORM flush 时生效，
|
||
# __init__ 覆盖使测试在不侵入 session 时就能验证默认值。
|
||
kwargs.setdefault("type", "session")
|
||
kwargs.setdefault("progress", 0.0)
|
||
super().__init__(**kwargs)
|
||
|
||
id: Mapped[str] = mapped_column(String(80), primary_key=True, default=new_goal_id)
|
||
character_id: Mapped[str | None] = mapped_column(
|
||
String(80), ForeignKey("characters.id"), nullable=True, index=True
|
||
)
|
||
title: Mapped[str] = mapped_column(String(300), nullable=False)
|
||
description: Mapped[str] = mapped_column(Text, nullable=False, default="")
|
||
level: Mapped[str] = mapped_column(String(40), nullable=False, default="task", index=True)
|
||
type: Mapped[str] = mapped_column(String(40), nullable=False, default="session")
|
||
status: Mapped[str] = mapped_column(String(40), nullable=False, default="pending", index=True)
|
||
priority: Mapped[int] = mapped_column(Integer, nullable=False, default=50)
|
||
progress: Mapped[float] = mapped_column(Float, nullable=False, default=0.0)
|
||
constraints_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
|
||
parent_id: Mapped[str | None] = mapped_column(String(80), ForeignKey("goals.id"), nullable=True)
|
||
success_criteria_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
|
||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
||
updated_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True), default=utc_now, onupdate=utc_now
|
||
)
|
||
|
||
|
||
class PerceptionChannel(Base):
|
||
"""烛照枢感知通道授权（契约字典 §5.7 频道级）。channel 为主键，默认六类可自定义。"""
|
||
|
||
__tablename__ = "perception_channels"
|
||
|
||
channel: Mapped[str] = mapped_column(String(80), primary_key=True)
|
||
name: Mapped[str] = mapped_column(String(120), nullable=False)
|
||
description: Mapped[str] = mapped_column(Text, nullable=False, default="")
|
||
enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||
consent_status: Mapped[str] = mapped_column(String(40), nullable=False, default="unknown")
|
||
privacy_level: Mapped[str] = mapped_column(String(40), nullable=False, default="normal")
|
||
risk_level: Mapped[str] = mapped_column(String(20), nullable=False, default="L2")
|
||
last_active_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
||
updated_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True), default=utc_now, onupdate=utc_now
|
||
)
|
||
|
||
|
||
class WorldEntity(Base):
|
||
"""河图枢世界实体（契约字典 §11.1 / §11.2）。"""
|
||
|
||
__tablename__ = "world_entities"
|
||
|
||
id: Mapped[str] = mapped_column(String(80), primary_key=True, default=new_world_entity_id)
|
||
type: Mapped[str] = mapped_column(String(40), nullable=False, default="concept", index=True)
|
||
name: Mapped[str] = mapped_column(String(200), nullable=False)
|
||
attributes_json: Mapped[str] = mapped_column(Text, nullable=False, default="{}")
|
||
state_type: Mapped[str] = mapped_column(String(40), nullable=False, default="inference")
|
||
confidence: Mapped[float] = mapped_column(Float, nullable=False, default=0.5)
|
||
sources_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
|
||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
||
updated_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True), default=utc_now, onupdate=utc_now
|
||
)
|
||
|
||
|
||
class Plan(Base):
|
||
"""洛书枢计划（契约字典 §5.13）。goal_id 关联太一枢目标，可空。"""
|
||
|
||
__tablename__ = "plans"
|
||
|
||
id: Mapped[str] = mapped_column(String(80), primary_key=True, default=new_plan_id)
|
||
goal_id: Mapped[str | None] = mapped_column(String(80), ForeignKey("goals.id"), index=True)
|
||
title: Mapped[str] = mapped_column(String(300), nullable=False)
|
||
status: Mapped[str] = mapped_column(String(40), nullable=False, default="draft", index=True)
|
||
assumptions_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
|
||
risks_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
|
||
required_tools_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
|
||
success_criteria_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
|
||
step_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
completed_steps: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
||
updated_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True), default=utc_now, onupdate=utc_now
|
||
)
|
||
|
||
|
||
class AgentRosterEntry(Base):
|
||
"""天工阁枢九角色在岗（契约字典 §5.14）。role 为主键，默认九角色可自定义。"""
|
||
|
||
__tablename__ = "agent_roster"
|
||
|
||
role: Mapped[str] = mapped_column(String(80), primary_key=True)
|
||
name: Mapped[str] = mapped_column(String(120), nullable=False)
|
||
responsibility: Mapped[str] = mapped_column(Text, nullable=False, default="")
|
||
enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
||
last_active_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
||
updated_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True), default=utc_now, onupdate=utc_now
|
||
)
|
||
|
||
|
||
class GenerationPreferences(Base):
|
||
"""生成参数偏好（模型网关 §4.3）。本地单用户单行，按 owner_scope 固定。"""
|
||
|
||
__tablename__ = "generation_preferences"
|
||
|
||
owner_scope: Mapped[str] = mapped_column(
|
||
String(80), primary_key=True, default=settings.default_owner_scope
|
||
)
|
||
default_model_hint: Mapped[str] = mapped_column(String(120), nullable=False)
|
||
temperature: Mapped[float] = mapped_column(Float, nullable=False, default=0.7)
|
||
top_p: Mapped[float] = mapped_column(Float, nullable=False, default=1.0)
|
||
max_output_tokens: Mapped[int] = mapped_column(Integer, nullable=False, default=4096)
|
||
streaming: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
||
updated_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True), default=utc_now, onupdate=utc_now
|
||
)
|
||
|
||
|
||
class ToolPolicy(Base):
|
||
"""百工枢工具策略治理（api-spec §15 / §16）。
|
||
|
||
工具的风险/权限策略持久化层；``name`` 为主键。builtin 工具启动时由种子 upsert
|
||
（``source=builtin``），自定义工具经 API 注册（``source=custom``）。司命决策器
|
||
（SafetyGovernor）从本表加载覆盖策略；``/tools`` 与 ``/safety/policies`` 两组 API
|
||
共享本表——前者返回完整元数据，后者只返回策略子集。
|
||
"""
|
||
|
||
__tablename__ = "tool_policies"
|
||
|
||
name: Mapped[str] = mapped_column(String(80), primary_key=True)
|
||
description: Mapped[str] = mapped_column(Text, nullable=False, default="")
|
||
risk_level: Mapped[str] = mapped_column(String(20), nullable=False, default="L2")
|
||
permission_policy: Mapped[str] = mapped_column(String(20), nullable=False, default="ask")
|
||
side_effect: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
||
audit_required: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
||
source: Mapped[str] = mapped_column(String(20), nullable=False, default="custom")
|
||
enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
||
updated_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True), default=utc_now, onupdate=utc_now
|
||
)
|
||
|
||
|
||
class ModelEndpointOverride(Base):
|
||
"""模型网关 endpoint 运行时覆盖（DB 优先于 .env / yaml，api-spec §4.4）。
|
||
|
||
每个 (logical_model_name, endpoint_name) 一行；字段非空才覆盖同名 endpoint 配置，
|
||
None = 不覆盖、回退 yaml/env。base_url/api_key 经 resolved_* 优先 endpoint 字段生效，
|
||
故 DB 写入后即「DB 优先 env」。本地单用户边界下 API key 明文存储，GET 脱敏不回值。
|
||
"""
|
||
|
||
__tablename__ = "model_endpoint_overrides"
|
||
|
||
logical_model_name: Mapped[str] = mapped_column(String(120), primary_key=True)
|
||
endpoint_name: Mapped[str] = mapped_column(String(120), primary_key=True)
|
||
base_url: Mapped[str | None] = mapped_column(String(500))
|
||
api_key: Mapped[str | None] = mapped_column(String(500))
|
||
status: Mapped[str | None] = mapped_column(String(40))
|
||
weight: Mapped[int | None] = mapped_column(Integer)
|
||
priority: Mapped[int | None] = mapped_column(Integer)
|
||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
||
updated_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True), default=utc_now, onupdate=utc_now
|
||
)
|