init: 初始化项目

backend/app/schemas/generation_task.py

@@ -0,0 +1,44 @@
+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