356 lines
10 KiB
Python
356 lines
10 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class OutputTypeCreate(BaseModel):
|
|
name: str
|
|
description: str | None = None
|
|
renderer: str = "blender"
|
|
render_settings: dict = {}
|
|
output_format: str = "png"
|
|
sort_order: int = 0
|
|
is_active: bool = True
|
|
compatible_categories: list[str] = []
|
|
render_backend: str = "celery"
|
|
is_animation: bool = False
|
|
transparent_bg: bool = False
|
|
pricing_tier_id: int | None = None
|
|
cycles_device: str | None = None
|
|
workflow_family: str = "order_line"
|
|
artifact_kind: str | None = None
|
|
invocation_overrides: dict = {}
|
|
workflow_definition_id: uuid.UUID | None = None
|
|
workflow_rollout_mode: str = "legacy_only"
|
|
material_override: str | None = None
|
|
|
|
|
|
class OutputTypePatch(BaseModel):
|
|
name: str | None = None
|
|
description: str | None = None
|
|
renderer: str | None = None
|
|
render_settings: dict | None = None
|
|
output_format: str | None = None
|
|
sort_order: int | None = None
|
|
is_active: bool | None = None
|
|
compatible_categories: list[str] | None = None
|
|
render_backend: str | None = None
|
|
is_animation: bool | None = None
|
|
transparent_bg: bool | None = None
|
|
workflow_family: str | None = None
|
|
artifact_kind: str | None = None
|
|
invocation_overrides: dict | None = None
|
|
pricing_tier_id: int | None = None
|
|
cycles_device: str | None = None
|
|
workflow_definition_id: uuid.UUID | None = None
|
|
workflow_rollout_mode: str | None = None
|
|
material_override: str | None = None
|
|
|
|
|
|
class OutputTypeInvocationProfileOut(BaseModel):
|
|
renderer: str
|
|
render_backend: str
|
|
workflow_family: str
|
|
artifact_kind: str
|
|
output_format: str
|
|
is_animation: bool
|
|
workflow_definition_id: uuid.UUID | None = None
|
|
workflow_rollout_mode: str = "legacy_only"
|
|
transparent_bg: bool
|
|
cycles_device: str | None = None
|
|
material_override: str | None = None
|
|
allowed_override_keys: list[str] = Field(default_factory=list)
|
|
invocation_overrides: dict = Field(default_factory=dict)
|
|
|
|
|
|
class OutputTypeContractCatalogOut(BaseModel):
|
|
workflow_families: list[str] = Field(default_factory=list)
|
|
workflow_rollout_modes: list[str] = Field(default_factory=list)
|
|
artifact_kinds: list[str] = Field(default_factory=list)
|
|
allowed_artifact_kinds_by_family: dict[str, list[str]] = Field(default_factory=dict)
|
|
allowed_output_formats_by_family: dict[str, list[str]] = Field(default_factory=dict)
|
|
allowed_invocation_override_keys_by_artifact_kind: dict[str, list[str]] = Field(default_factory=dict)
|
|
default_output_format_by_artifact_kind: dict[str, str] = Field(default_factory=dict)
|
|
parameter_ownership: dict[str, dict | list[str]] = Field(default_factory=dict)
|
|
|
|
|
|
class OutputTypeOut(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
description: str | None
|
|
renderer: str
|
|
render_settings: dict
|
|
output_format: str
|
|
sort_order: int
|
|
compatible_categories: list[str]
|
|
render_backend: str
|
|
is_animation: bool
|
|
transparent_bg: bool
|
|
workflow_family: str
|
|
artifact_kind: str
|
|
invocation_overrides: dict
|
|
cycles_device: str | None = None
|
|
pricing_tier_id: int | None = None
|
|
pricing_tier_name: str | None = None
|
|
price_per_item: float | None = None
|
|
workflow_definition_id: uuid.UUID | None = None
|
|
workflow_rollout_mode: str
|
|
workflow_name: str | None = None
|
|
material_override: str | None = None
|
|
invocation_profile: OutputTypeInvocationProfileOut | None = None
|
|
is_active: bool
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class RenderPositionCreate(BaseModel):
|
|
name: str
|
|
rotation_x: float = 0.0
|
|
rotation_y: float = 0.0
|
|
rotation_z: float = 0.0
|
|
is_default: bool = False
|
|
sort_order: int = 0
|
|
focal_length_mm: float | None = None
|
|
sensor_width_mm: float | None = None
|
|
|
|
|
|
class RenderPositionPatch(BaseModel):
|
|
name: str | None = None
|
|
rotation_x: float | None = None
|
|
rotation_y: float | None = None
|
|
rotation_z: float | None = None
|
|
is_default: bool | None = None
|
|
sort_order: int | None = None
|
|
focal_length_mm: float | None = None
|
|
sensor_width_mm: float | None = None
|
|
|
|
|
|
class RenderPositionOut(BaseModel):
|
|
id: uuid.UUID
|
|
product_id: uuid.UUID
|
|
name: str
|
|
rotation_x: float
|
|
rotation_y: float
|
|
rotation_z: float
|
|
is_default: bool
|
|
sort_order: int
|
|
focal_length_mm: float | None = None
|
|
sensor_width_mm: float | None = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class GlobalRenderPositionCreate(BaseModel):
|
|
name: str
|
|
rotation_x: float = 0.0
|
|
rotation_y: float = 0.0
|
|
rotation_z: float = 0.0
|
|
is_default: bool = False
|
|
sort_order: int = 0
|
|
focal_length_mm: float | None = None
|
|
sensor_width_mm: float | None = None
|
|
|
|
|
|
class GlobalRenderPositionPatch(BaseModel):
|
|
name: str | None = None
|
|
rotation_x: float | None = None
|
|
rotation_y: float | None = None
|
|
rotation_z: float | None = None
|
|
is_default: bool | None = None
|
|
sort_order: int | None = None
|
|
focal_length_mm: float | None = None
|
|
sensor_width_mm: float | None = None
|
|
|
|
|
|
class GlobalRenderPositionOut(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
rotation_x: float
|
|
rotation_y: float
|
|
rotation_z: float
|
|
is_default: bool
|
|
sort_order: int
|
|
focal_length_mm: float | None = None
|
|
sensor_width_mm: float | None = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class WorkflowDefinitionCreate(BaseModel):
|
|
name: str
|
|
output_type_id: uuid.UUID | None = None
|
|
config: dict
|
|
is_active: bool = True
|
|
|
|
|
|
class WorkflowDefinitionUpdate(BaseModel):
|
|
name: str | None = None
|
|
config: dict | None = None
|
|
is_active: bool | None = None
|
|
|
|
|
|
class WorkflowDefinitionOut(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
output_type_id: uuid.UUID | None
|
|
config: dict
|
|
family: str | None = None
|
|
supported_artifact_kinds: list[str] = Field(default_factory=list)
|
|
rollout_summary: "WorkflowRolloutSummaryOut" = Field(
|
|
default_factory=lambda: WorkflowRolloutSummaryOut()
|
|
)
|
|
is_active: bool
|
|
created_at: datetime
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class WorkflowDraftPreflightRequest(BaseModel):
|
|
context_id: str
|
|
config: dict
|
|
workflow_id: uuid.UUID | None = None
|
|
|
|
|
|
class WorkflowDraftDispatchRequest(BaseModel):
|
|
context_id: str
|
|
config: dict
|
|
workflow_id: uuid.UUID | None = None
|
|
|
|
|
|
class WorkflowNodeResultOut(BaseModel):
|
|
id: uuid.UUID
|
|
node_name: str
|
|
status: str
|
|
output: dict | None
|
|
log: str | None
|
|
duration_s: float | None
|
|
created_at: datetime
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class WorkflowRunOut(BaseModel):
|
|
id: uuid.UUID
|
|
workflow_def_id: uuid.UUID | None
|
|
order_line_id: uuid.UUID | None
|
|
celery_task_id: str | None
|
|
execution_mode: str
|
|
status: str
|
|
started_at: datetime | None
|
|
completed_at: datetime | None
|
|
error_message: str | None
|
|
created_at: datetime
|
|
node_results: list[WorkflowNodeResultOut] = []
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class WorkflowRolloutLatestRunOut(BaseModel):
|
|
workflow_run_id: uuid.UUID
|
|
execution_mode: str
|
|
status: str
|
|
created_at: datetime
|
|
completed_at: datetime | None = None
|
|
|
|
|
|
class WorkflowRolloutLinkedOutputTypeOut(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
is_active: bool
|
|
artifact_kind: str
|
|
workflow_rollout_mode: str
|
|
|
|
|
|
class WorkflowRolloutSummaryOut(BaseModel):
|
|
linked_output_type_count: int = 0
|
|
active_output_type_count: int = 0
|
|
linked_output_type_names: list[str] = Field(default_factory=list)
|
|
linked_output_types: list[WorkflowRolloutLinkedOutputTypeOut] = Field(default_factory=list)
|
|
rollout_modes: list[str] = Field(default_factory=list)
|
|
has_blocking_contracts: bool = False
|
|
blocking_reasons: list[str] = Field(default_factory=list)
|
|
latest_run: WorkflowRolloutLatestRunOut | None = None
|
|
latest_shadow_run: WorkflowRolloutLatestRunOut | None = None
|
|
latest_rollout_gate_verdict: str | None = None
|
|
latest_rollout_ready: bool | None = None
|
|
latest_rollout_status: str | None = None
|
|
latest_rollout_reasons: list[str] = Field(default_factory=list)
|
|
|
|
|
|
class WorkflowComparisonArtifactOut(BaseModel):
|
|
path: str | None
|
|
storage_key: str | None
|
|
exists: bool
|
|
file_size_bytes: int | None
|
|
sha256: str | None
|
|
mime_type: str | None
|
|
image_width: int | None
|
|
image_height: int | None
|
|
|
|
|
|
class WorkflowRunComparisonOut(BaseModel):
|
|
workflow_run_id: uuid.UUID
|
|
workflow_def_id: uuid.UUID | None
|
|
order_line_id: uuid.UUID | None
|
|
execution_mode: str
|
|
status: str
|
|
summary: str
|
|
rollout_gate_verdict: str
|
|
workflow_rollout_ready: bool
|
|
workflow_rollout_status: str
|
|
rollout_reasons: list[str] = []
|
|
rollout_thresholds: dict[str, float] = Field(default_factory=dict)
|
|
authoritative_output: WorkflowComparisonArtifactOut
|
|
observer_output: WorkflowComparisonArtifactOut
|
|
exact_match: bool | None
|
|
dimensions_match: bool | None
|
|
mean_pixel_delta: float | None
|
|
|
|
|
|
class WorkflowPreflightIssueOut(BaseModel):
|
|
severity: str
|
|
code: str
|
|
message: str
|
|
node_id: str | None = None
|
|
step: str | None = None
|
|
|
|
|
|
class WorkflowPreflightNodeOut(BaseModel):
|
|
node_id: str
|
|
step: str
|
|
label: str | None = None
|
|
execution_kind: str
|
|
supported: bool
|
|
status: str
|
|
issues: list[WorkflowPreflightIssueOut] = []
|
|
|
|
|
|
class WorkflowPreflightOut(BaseModel):
|
|
workflow_id: uuid.UUID | None = None
|
|
context_id: str
|
|
context_kind: str | None = None
|
|
expected_context_kind: str
|
|
execution_mode: str
|
|
graph_dispatch_allowed: bool
|
|
summary: str
|
|
resolved_order_line_id: uuid.UUID | None = None
|
|
resolved_cad_file_id: uuid.UUID | None = None
|
|
unsupported_node_ids: list[str] = []
|
|
issues: list[WorkflowPreflightIssueOut] = []
|
|
nodes: list[WorkflowPreflightNodeOut] = []
|
|
|
|
|
|
class WorkflowOrderLineContextOptionOut(BaseModel):
|
|
value: uuid.UUID
|
|
label: str
|
|
meta: str
|
|
|
|
|
|
class WorkflowOrderLineContextGroupOut(BaseModel):
|
|
order_id: uuid.UUID
|
|
order_label: str
|
|
options: list[WorkflowOrderLineContextOptionOut] = []
|