30 lines
954 B
Python
30 lines
954 B
Python
from .base import BaseModel
|
||
from . import db
|
||
|
||
|
||
class Tool(BaseModel):
|
||
__tablename__ = "tools"
|
||
|
||
name = db.Column(db.String(100), unique=True, nullable=False, comment="工具名称")
|
||
type = db.Column(
|
||
db.String(50), nullable=False, comment="类型（builtin, mcp, custom）"
|
||
)
|
||
description = db.Column(db.Text, nullable=True, comment="描述")
|
||
config = db.Column(db.JSON, nullable=True, comment="配置（JSON）")
|
||
is_active = db.Column(db.Boolean, default=True, nullable=False, comment="是否激活")
|
||
|
||
agent_tools = db.relationship(
|
||
"AgentTool", backref="tool", lazy=True, cascade="all, delete-orphan"
|
||
)
|
||
|
||
|
||
class AgentTool(BaseModel):
|
||
__tablename__ = "agent_tools"
|
||
|
||
agent_id = db.Column(
|
||
db.BigInteger, db.ForeignKey("agents.id"), nullable=False, comment="Agent ID"
|
||
)
|
||
tool_id = db.Column(
|
||
db.BigInteger, db.ForeignKey("tools.id"), nullable=False, comment="工具ID"
|
||
)
|