67 lines
2.9 KiB
Python
67 lines
2.9 KiB
Python
|
"""
|
|||
|
|
节点模板模型
|
|||
|
|
用于管理和复用LLM节点的提示词模板
|
|||
|
|
"""
|
|||
|
|
from sqlalchemy import Column, String, Text, Integer, DateTime, JSON, ForeignKey, Boolean, func
|
|||
|
|
from sqlalchemy.dialects.mysql import CHAR
|
|||
|
|
from sqlalchemy.orm import relationship
|
|||
|
|
from app.core.database import Base
|
|||
|
|
import uuid
|
|||
|
|
|
|||
|
|
|
|||
|
|
class NodeTemplate(Base):
|
|||
|
|
"""节点模板表"""
|
|||
|
|
__tablename__ = "node_templates"
|
|||
|
|
|
|||
|
|
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="模板描述")
|
|||
|
|
category = Column(String(50), comment="分类: text_generation/data_analysis/code_generation/translation/summarization/qa/other")
|
|||
|
|
tags = Column(JSON, comment="标签列表")
|
|||
|
|
|
|||
|
|
# 模板内容
|
|||
|
|
prompt = Column(Text, nullable=False, comment="提示词模板（支持变量占位符，如 {{variable}}）")
|
|||
|
|
variables = Column(JSON, comment="变量定义列表，格式: [{\"name\": \"var1\", \"type\": \"text\", \"required\": true, \"description\": \"变量描述\"}]")
|
|||
|
|
|
|||
|
|
# 默认配置
|
|||
|
|
provider = Column(String(50), default="deepseek", comment="默认LLM提供商")
|
|||
|
|
model = Column(String(100), default="deepseek-chat", comment="默认模型")
|
|||
|
|
temperature = Column(String(10), default="0.7", comment="默认温度参数")
|
|||
|
|
max_tokens = Column(Integer, default=1500, comment="默认最大token数")
|
|||
|
|
|
|||
|
|
# 元数据
|
|||
|
|
is_public = Column(Boolean, default=False, comment="是否公开")
|
|||
|
|
is_featured = Column(Boolean, default=False, comment="是否精选")
|
|||
|
|
use_count = Column(Integer, default=0, comment="使用次数")
|
|||
|
|
user_id = Column(CHAR(36), ForeignKey("users.id"), nullable=False, 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="node_templates")
|
|||
|
|
|
|||
|
|
def __repr__(self):
|
|||
|
|
return f"<NodeTemplate(id={self.id}, name={self.name})>"
|
|||
|
|
|
|||
|
|
def to_dict(self):
|
|||
|
|
"""转换为字典"""
|
|||
|
|
return {
|
|||
|
|
"id": self.id,
|
|||
|
|
"name": self.name,
|
|||
|
|
"description": self.description,
|
|||
|
|
"category": self.category,
|
|||
|
|
"tags": self.tags or [],
|
|||
|
|
"prompt": self.prompt,
|
|||
|
|
"variables": self.variables or [],
|
|||
|
|
"provider": self.provider,
|
|||
|
|
"model": self.model,
|
|||
|
|
"temperature": self.temperature,
|
|||
|
|
"max_tokens": self.max_tokens,
|
|||
|
|
"is_public": self.is_public,
|
|||
|
|
"is_featured": self.is_featured,
|
|||
|
|
"use_count": self.use_count,
|
|||
|
|
"user_id": self.user_id,
|
|||
|
|
"created_at": self.created_at.isoformat() if self.created_at else None,
|
|||
|
|
"updated_at": self.updated_at.isoformat() if self.updated_at else None
|
|||
|
|
}
|