zwt13703
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class GenerationPointCreate(BaseModel):
|
|
template_id: uuid.UUID
|
|
position: dict = Field(..., description="选区位置信息")
|
|
prompt: str
|
|
model_id: uuid.UUID | None = None
|
|
order: int = 0
|
|
selected_text: str | None = None
|
|
need_ref_file: bool = False
|
|
remark: str | None = None
|
|
|
|
|
|
class GenerationPointUpdate(BaseModel):
|
|
position: dict | None = None
|
|
prompt: str | None = None
|
|
model_id: uuid.UUID | None = None
|
|
order: int | None = None
|
|
selected_text: str | None = None
|
|
need_ref_file: bool | None = None
|
|
remark: str | None = None
|
|
|
|
|
|
class GenerationPointResponse(BaseModel):
|
|
id: uuid.UUID
|
|
template_id: uuid.UUID
|
|
position: dict
|
|
prompt: str
|
|
model_id: uuid.UUID | None = None
|
|
ref_file_path: str | None = None
|
|
need_ref_file: bool = False
|
|
remark: str | None = None
|
|
order: int
|
|
selected_text: str | None = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class BatchOrderUpdate(BaseModel):
|
|
points: list[dict] = Field(..., description="[{id: uuid, order: int}, ...]")
|