b8b01254ca
- Add enterprise_review tool (35th builtin) for LLM-based quality assessment - Add evaluator workflow node type for quality gating in DAG - Add AgentRuntime built-in self-review with auto-correction loop - Rewrite error_handler node from stub to real retry mechanism - Add engine-level per-node retry with configurable max_retries/delay/on_exhausted - Add AgentExtension model for extension tracking - Enhance validation in agent_create_tool and tool_register_tool - Update 全能助手 system prompt with self-evolution workflow - Docs: 缺失能力.md and 解决缺失能力计划.md Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
53 lines
2.3 KiB
Python
53 lines
2.3 KiB
Python
"""
|
||
智能体模型
|
||
"""
|
||
from sqlalchemy import Column, String, Text, Integer, DateTime, JSON, ForeignKey, func
|
||
from sqlalchemy.dialects.mysql import CHAR
|
||
from sqlalchemy.orm import relationship
|
||
from app.core.database import Base
|
||
import uuid
|
||
|
||
|
||
class Agent(Base):
|
||
"""智能体表"""
|
||
__tablename__ = "agents"
|
||
|
||
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="描述")
|
||
workflow_config = Column(JSON, nullable=False, comment="工作流配置")
|
||
budget_config = Column(
|
||
JSON,
|
||
nullable=True,
|
||
comment="执行预算：max_steps/max_llm_invocations/max_tool_calls（可选，覆盖全局默认）",
|
||
)
|
||
version = Column(Integer, default=1, comment="版本号")
|
||
status = Column(String(20), default="draft", comment="状态: draft/published/running/stopped")
|
||
user_id = Column(CHAR(36), ForeignKey("users.id"), comment="创建者ID")
|
||
created_at = Column(DateTime, default=func.now(), comment="创建时间")
|
||
updated_at = Column(DateTime, default=func.now(), onupdate=func.now(), comment="更新时间")
|
||
|
||
# 关系
|
||
user = relationship("User", backref="agents")
|
||
|
||
def __repr__(self):
|
||
return f"<Agent(id={self.id}, name={self.name})>"
|
||
|
||
|
||
class AgentExtension(Base):
|
||
"""Agent 自主扩展记录表 — 用于 extension_log 工具"""
|
||
__tablename__ = "agent_extensions"
|
||
|
||
id = Column(CHAR(36), primary_key=True, default=lambda: str(uuid.uuid4()), comment="扩展记录ID")
|
||
extension_type = Column(String(50), nullable=False, comment="扩展类型: agent_created/tool_registered/code_tool_created")
|
||
name = Column(String(100), nullable=False, comment="创建的Agent或工具名称")
|
||
reason = Column(Text, comment="创建原因（为什么需要扩展）")
|
||
detail = Column(JSON, comment="扩展详情（参数、结果等）")
|
||
success_rating = Column(String(20), comment="效果评价: success/partial/failed")
|
||
note = Column(Text, comment="反馈备注")
|
||
user_id = Column(CHAR(36), ForeignKey("users.id"), nullable=True, comment="创建者ID")
|
||
created_at = Column(DateTime, default=func.now(), comment="创建时间")
|
||
|
||
def __repr__(self):
|
||
return f"<AgentExtension(id={self.id}, type={self.extension_type}, name={self.name})>"
|