68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
|
"""提示词模板相关的Pydantic模型"""
|
||
|
|
from pydantic import BaseModel, Field, ConfigDict
|
||
|
|
from typing import Optional, List
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
|
||
|
|
class PromptTemplateBase(BaseModel):
|
||
|
|
"""提示词模板基础模型"""
|
||
|
|
template_key: str = Field(..., description="模板键名")
|
||
|
|
template_name: str = Field(..., description="模板显示名称")
|
||
|
|
template_content: str = Field(..., description="模板内容")
|
||
|
|
description: Optional[str] = Field(None, description="模板描述")
|
||
|
|
category: Optional[str] = Field(None, description="模板分类")
|
||
|
|
parameters: Optional[str] = Field(None, description="模板参数定义(JSON)")
|
||
|
|
is_active: bool = Field(True, description="是否启用")
|
||
|
|
|
||
|
|
|
||
|
|
class PromptTemplateCreate(PromptTemplateBase):
|
||
|
|
"""创建提示词模板请求模型"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class PromptTemplateUpdate(BaseModel):
|
||
|
|
"""更新提示词模板请求模型"""
|
||
|
|
template_name: Optional[str] = Field(None, description="模板显示名称")
|
||
|
|
template_content: Optional[str] = Field(None, description="模板内容")
|
||
|
|
description: Optional[str] = Field(None, description="模板描述")
|
||
|
|
category: Optional[str] = Field(None, description="模板分类")
|
||
|
|
parameters: Optional[str] = Field(None, description="模板参数定义(JSON)")
|
||
|
|
is_active: Optional[bool] = Field(None, description="是否启用")
|
||
|
|
|
||
|
|
|
||
|
|
class PromptTemplateResponse(PromptTemplateBase):
|
||
|
|
"""提示词模板响应模型"""
|
||
|
|
model_config = ConfigDict(from_attributes=True)
|
||
|
|
|
||
|
|
id: str
|
||
|
|
user_id: str
|
||
|
|
is_system_default: bool
|
||
|
|
created_at: datetime
|
||
|
|
updated_at: datetime
|
||
|
|
|
||
|
|
|
||
|
|
class PromptTemplateListResponse(BaseModel):
|
||
|
|
"""提示词模板列表响应"""
|
||
|
|
templates: List[PromptTemplateResponse]
|
||
|
|
total: int
|
||
|
|
categories: List[str]
|
||
|
|
|
||
|
|
|
||
|
|
class PromptTemplateCategoryResponse(BaseModel):
|
||
|
|
"""提示词模板分类响应"""
|
||
|
|
category: str
|
||
|
|
count: int
|
||
|
|
templates: List[PromptTemplateResponse]
|
||
|
|
|
||
|
|
|
||
|
|
class PromptTemplateExport(BaseModel):
|
||
|
|
"""提示词模板导出模型"""
|
||
|
|
templates: List[PromptTemplateBase]
|
||
|
|
export_time: datetime
|
||
|
|
version: str = "1.0"
|
||
|
|
|
||
|
|
|
||
|
|
class PromptTemplatePreviewRequest(BaseModel):
|
||
|
|
"""提示词模板预览请求"""
|
||
|
|
template_content: str = Field(..., description="模板内容")
|
||
|
|
parameters: dict = Field(..., description="参数字典")
|