Implement FastAPI Temporal MVP pipeline
This commit is contained in:
23
app/api/schemas/asset.py
Normal file
23
app/api/schemas/asset.py
Normal file
@@ -0,0 +1,23 @@
|
||||
"""Asset API schemas."""
|
||||
|
||||
from datetime import datetime
|
||||
from typing import Any
|
||||
|
||||
from pydantic import BaseModel, ConfigDict
|
||||
|
||||
from app.domain.enums import AssetType, WorkflowStepName
|
||||
|
||||
|
||||
class AssetRead(BaseModel):
|
||||
"""Serialized asset response."""
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
id: int
|
||||
order_id: int
|
||||
asset_type: AssetType
|
||||
step_name: WorkflowStepName | None
|
||||
uri: str
|
||||
metadata_json: dict[str, Any] | None
|
||||
created_at: datetime
|
||||
|
||||
47
app/api/schemas/order.py
Normal file
47
app/api/schemas/order.py
Normal file
@@ -0,0 +1,47 @@
|
||||
"""Order API schemas."""
|
||||
|
||||
from datetime import datetime
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
from app.api.schemas.asset import AssetRead
|
||||
from app.domain.enums import CustomerLevel, OrderStatus, ServiceMode, WorkflowStepName
|
||||
|
||||
|
||||
class CreateOrderRequest(BaseModel):
|
||||
"""Request payload for creating an order."""
|
||||
|
||||
customer_level: CustomerLevel
|
||||
service_mode: ServiceMode
|
||||
model_id: int
|
||||
pose_id: int
|
||||
garment_asset_id: int
|
||||
scene_ref_asset_id: int
|
||||
|
||||
|
||||
class CreateOrderResponse(BaseModel):
|
||||
"""Response returned after an order has been created."""
|
||||
|
||||
order_id: int
|
||||
workflow_id: str
|
||||
status: OrderStatus
|
||||
|
||||
|
||||
class OrderDetailResponse(BaseModel):
|
||||
"""Order detail response."""
|
||||
|
||||
order_id: int
|
||||
customer_level: CustomerLevel
|
||||
service_mode: ServiceMode
|
||||
status: OrderStatus
|
||||
model_id: int
|
||||
pose_id: int
|
||||
garment_asset_id: int
|
||||
scene_ref_asset_id: int
|
||||
final_asset_id: int | None
|
||||
workflow_id: str | None
|
||||
current_step: WorkflowStepName | None
|
||||
final_asset: AssetRead | None
|
||||
created_at: datetime
|
||||
updated_at: datetime
|
||||
|
||||
36
app/api/schemas/review.py
Normal file
36
app/api/schemas/review.py
Normal file
@@ -0,0 +1,36 @@
|
||||
"""Review API schemas."""
|
||||
|
||||
from datetime import datetime
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
from app.domain.enums import ReviewDecision, WorkflowStepName
|
||||
|
||||
|
||||
class SubmitReviewRequest(BaseModel):
|
||||
"""Request payload for review submission."""
|
||||
|
||||
decision: ReviewDecision
|
||||
reviewer_id: int
|
||||
selected_asset_id: int | None = None
|
||||
comment: str | None = None
|
||||
|
||||
|
||||
class SubmitReviewResponse(BaseModel):
|
||||
"""Response returned after a review signal is sent."""
|
||||
|
||||
order_id: int
|
||||
workflow_id: str
|
||||
decision: ReviewDecision
|
||||
status: str
|
||||
|
||||
|
||||
class PendingReviewResponse(BaseModel):
|
||||
"""Response model for pending review items."""
|
||||
|
||||
review_task_id: int
|
||||
order_id: int
|
||||
workflow_id: str
|
||||
current_step: WorkflowStepName | None
|
||||
created_at: datetime
|
||||
|
||||
38
app/api/schemas/workflow.py
Normal file
38
app/api/schemas/workflow.py
Normal file
@@ -0,0 +1,38 @@
|
||||
"""Workflow API schemas."""
|
||||
|
||||
from datetime import datetime
|
||||
from typing import Any
|
||||
|
||||
from pydantic import BaseModel, ConfigDict
|
||||
|
||||
from app.domain.enums import OrderStatus, StepStatus, WorkflowStepName
|
||||
|
||||
|
||||
class WorkflowStepRead(BaseModel):
|
||||
"""Serialized workflow step record."""
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
id: int
|
||||
workflow_run_id: int
|
||||
step_name: WorkflowStepName
|
||||
step_status: StepStatus
|
||||
input_json: dict[str, Any] | None
|
||||
output_json: dict[str, Any] | None
|
||||
error_message: str | None
|
||||
started_at: datetime
|
||||
ended_at: datetime | None
|
||||
|
||||
|
||||
class WorkflowStatusResponse(BaseModel):
|
||||
"""Serialized workflow run details."""
|
||||
|
||||
order_id: int
|
||||
workflow_id: str
|
||||
workflow_type: str
|
||||
workflow_status: OrderStatus
|
||||
current_step: WorkflowStepName | None
|
||||
steps: list[WorkflowStepRead]
|
||||
created_at: datetime
|
||||
updated_at: datetime
|
||||
|
||||
Reference in New Issue
Block a user