31 lines
611 B
Python
31 lines
611 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 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
|