Files
state-hub/api/schemas/execution.py
2026-05-23 19:11:30 +02:00

92 lines
2.9 KiB
Python

import uuid
from datetime import datetime
from typing import Literal
from pydantic import BaseModel, ConfigDict, Field
ExecutionState = Literal["manual", "queued", "scheduled", "launching", "paused", "completed", "cancelled"]
LaunchMode = Literal["manual", "queued", "scheduled", "immediate"]
ConcurrencyMode = Literal["sequential", "parallel"]
LaunchRequestStatus = Literal["requested", "accepted", "completed", "cancelled"]
class ExecutionIntentUpdate(BaseModel):
execution_state: ExecutionState | None = None
launch_mode: LaunchMode | None = None
concurrency_mode: ConcurrencyMode | None = None
queue_rank: int | None = None
execution_group: str | None = None
scheduled_for: datetime | None = None
class ExecutionIntentRead(BaseModel):
workstream_id: uuid.UUID
execution_state: ExecutionState
launch_mode: LaunchMode
concurrency_mode: ConcurrencyMode
queue_rank: int | None = None
execution_group: str | None = None
scheduled_for: datetime | None = None
class WorkplanQueueItem(BaseModel):
workstream_id: uuid.UUID
slug: str
title: str
status: str
repo_id: uuid.UUID | None = None
planning_priority: str | None = None
planning_order: int | None = None
execution_state: ExecutionState
launch_mode: LaunchMode
concurrency_mode: ConcurrencyMode
queue_rank: int | None = None
execution_group: str | None = None
scheduled_for: datetime | None = None
eligible: bool
blocked_by_workstream_ids: list[uuid.UUID] = Field(default_factory=list)
blocked_by_task_ids: list[uuid.UUID] = Field(default_factory=list)
sort_key: list[str | int] = Field(default_factory=list)
class LaunchRequestCreate(BaseModel):
workstream_id: uuid.UUID
requested_by: str = "dashboard"
requested_actor: str | None = None
launch_mode: LaunchMode = "queued"
concurrency_mode: ConcurrencyMode = "sequential"
priority: str | None = None
repo_id: uuid.UUID | None = None
branch_preference: str | None = None
immediate_pickup: bool = False
notes: str | None = None
request_metadata: dict = Field(default_factory=dict)
class LaunchRequestRead(BaseModel):
model_config = ConfigDict(from_attributes=True)
id: uuid.UUID
workstream_id: uuid.UUID
requested_by: str
requested_actor: str | None = None
launch_mode: LaunchMode
concurrency_mode: ConcurrencyMode
priority: str | None = None
repo_id: uuid.UUID | None = None
branch_preference: str | None = None
immediate_pickup: bool
status: LaunchRequestStatus
notes: str | None = None
request_metadata: dict = Field(default_factory=dict)
created_at: datetime
updated_at: datetime
class ExecutionSemantics(BaseModel):
execution_states: dict[str, str]
launch_modes: dict[str, str]
concurrency_modes: dict[str, str]
state_hub_responsibility: list[str]
activity_core_responsibility: list[str]