99 lines
4.2 KiB
Python
99 lines
4.2 KiB
Python
|
"""
|
|||
|
|
场景契约模型 — 统一 DSL 输入契约（目标/约束/产物/验收）
|
|||
|
|
让不同模板复用统一输入格式，实现"场景可编程输入"
|
|||
|
|
"""
|
|||
|
|
from sqlalchemy import Column, String, Text, Integer, DateTime, JSON, ForeignKey, Index, func
|
|||
|
|
from sqlalchemy.dialects.mysql import CHAR
|
|||
|
|
from sqlalchemy.orm import relationship
|
|||
|
|
from app.core.database import Base
|
|||
|
|
import uuid
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SceneContract(Base):
|
|||
|
|
"""场景契约表 — 定义标准输入契约"""
|
|||
|
|
__tablename__ = "scene_contracts"
|
|||
|
|
|
|||
|
|
id = Column(CHAR(36), primary_key=True, default=lambda: str(uuid.uuid4()), comment="契约ID")
|
|||
|
|
name = Column(String(100), nullable=False, comment="契约名称")
|
|||
|
|
description = Column(Text, comment="契约描述")
|
|||
|
|
|
|||
|
|
# ── 核心 DSL 字段 ──
|
|||
|
|
goal = Column(Text, nullable=False, comment="场景目标（一句话描述要达成什么）")
|
|||
|
|
role = Column(Text, nullable=True, comment="Agent 扮演的角色/人设")
|
|||
|
|
|
|||
|
|
# 输入规范
|
|||
|
|
input_description = Column(Text, nullable=True, comment="期望的输入描述")
|
|||
|
|
input_schema = Column(JSON, nullable=True, comment="输入 JSON Schema 约束")
|
|||
|
|
|
|||
|
|
# 约束条件
|
|||
|
|
constraints = Column(JSON, nullable=True, comment="行为约束列表")
|
|||
|
|
forbidden_actions = Column(JSON, nullable=True, comment="禁止的操作列表")
|
|||
|
|
required_tools = Column(JSON, nullable=True, comment="必需的工具列表")
|
|||
|
|
|
|||
|
|
# 产物定义
|
|||
|
|
deliverables = Column(JSON, nullable=True, comment="期望产出物列表 [{name, format, description}]")
|
|||
|
|
|
|||
|
|
# 验收标准
|
|||
|
|
acceptance_criteria = Column(JSON, nullable=True, comment="验收条件列表")
|
|||
|
|
|
|||
|
|
# 输出规范
|
|||
|
|
output_schema = Column(JSON, nullable=True, comment="输出 JSON Schema 约束")
|
|||
|
|
|
|||
|
|
# Few-shot 示例
|
|||
|
|
examples = Column(JSON, nullable=True, comment="示例列表 [{input, output}]")
|
|||
|
|
|
|||
|
|
# 元数据
|
|||
|
|
category = Column(String(50), nullable=True, comment="分类")
|
|||
|
|
tags = Column(JSON, nullable=True, comment="标签列表")
|
|||
|
|
version = Column(Integer, default=1, comment="版本号")
|
|||
|
|
is_public = Column(Integer, default=0, comment="是否公开: 0=私有 1=公开")
|
|||
|
|
use_count = Column(Integer, default=0, comment="使用次数")
|
|||
|
|
|
|||
|
|
# 关联
|
|||
|
|
user_id = Column(CHAR(36), ForeignKey("users.id"), nullable=True, comment="创建者ID")
|
|||
|
|
workspace_id = Column(CHAR(36), ForeignKey("workspaces.id"), nullable=True, comment="所属工作区ID")
|
|||
|
|
template_binding = Column(String(100), nullable=True, comment="绑定的场景模板ID（可选）")
|
|||
|
|
|
|||
|
|
created_at = Column(DateTime, default=func.now(), comment="创建时间")
|
|||
|
|
updated_at = Column(DateTime, default=func.now(), onupdate=func.now(), comment="更新时间")
|
|||
|
|
|
|||
|
|
__table_args__ = (
|
|||
|
|
Index("idx_sc_category", "category"),
|
|||
|
|
Index("idx_sc_user_id", "user_id"),
|
|||
|
|
Index("idx_sc_is_public", "is_public"),
|
|||
|
|
Index("idx_sc_template_binding", "template_binding"),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
user = relationship("User", backref="scene_contracts")
|
|||
|
|
|
|||
|
|
def __repr__(self):
|
|||
|
|
return f"<SceneContract(id={self.id}, name={self.name})>"
|
|||
|
|
|
|||
|
|
def to_dict(self):
|
|||
|
|
return {
|
|||
|
|
"id": self.id,
|
|||
|
|
"name": self.name,
|
|||
|
|
"description": self.description,
|
|||
|
|
"goal": self.goal,
|
|||
|
|
"role": self.role,
|
|||
|
|
"input_description": self.input_description,
|
|||
|
|
"input_schema": self.input_schema,
|
|||
|
|
"constraints": self.constraints or [],
|
|||
|
|
"forbidden_actions": self.forbidden_actions or [],
|
|||
|
|
"required_tools": self.required_tools or [],
|
|||
|
|
"deliverables": self.deliverables or [],
|
|||
|
|
"acceptance_criteria": self.acceptance_criteria or [],
|
|||
|
|
"output_schema": self.output_schema,
|
|||
|
|
"examples": self.examples or [],
|
|||
|
|
"category": self.category,
|
|||
|
|
"tags": self.tags or [],
|
|||
|
|
"version": self.version,
|
|||
|
|
"is_public": self.is_public,
|
|||
|
|
"use_count": self.use_count,
|
|||
|
|
"user_id": self.user_id,
|
|||
|
|
"workspace_id": self.workspace_id,
|
|||
|
|
"template_binding": self.template_binding,
|
|||
|
|
"created_at": self.created_at.isoformat() if self.created_at else None,
|
|||
|
|
"updated_at": self.updated_at.isoformat() if self.updated_at else None,
|
|||
|
|
}
|