45 lines
951 B
Python
45 lines
951 B
Python
import uuid
|
|
from datetime import datetime
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class TaskResponse(BaseModel):
|
|
id: uuid.UUID
|
|
template_id: uuid.UUID
|
|
status: str
|
|
result_file_path: str | None = None
|
|
error_msg: str | None = None
|
|
created_at: datetime
|
|
finished_at: datetime | None = None
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class TaskDetailResponse(BaseModel):
|
|
id: uuid.UUID
|
|
template_id: uuid.UUID
|
|
template_name: str = ""
|
|
status: str
|
|
result_file_path: str | None = None
|
|
error_msg: str | None = None
|
|
created_at: datetime
|
|
finished_at: datetime | None = None
|
|
points: list[dict] = []
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class GenerateResponse(BaseModel):
|
|
task_id: uuid.UUID
|
|
status: str
|
|
|
|
|
|
class SingleTestRequest(BaseModel):
|
|
prompt: str
|
|
model_id: uuid.UUID | None = None
|
|
ref_file_path: str | None = None
|
|
|
|
|
|
class SingleTestResponse(BaseModel):
|
|
result: str
|