Coverage for backend/common/src/common/models/idaes/payloads/solve_request_schema.py: 100%

1from enum import StrEnum 

2from typing import Optional 

3from pydantic import BaseModel 

4from common.models import CloudEventsPayload 

5from ahuora_builder_types import FlowsheetSchema 

6from ahuora_builder_types.flowsheet_schema import SolvedFlowsheetSchema 

7from ahuora_builder_types.payloads.solve_request_schema import IdaesSolveRequestPayload, IdaesSolveCompletionPayload, MultiSolvePayload 

8 

9 

10IdaesSolveEvent = CloudEventsPayload[IdaesSolveRequestPayload] 

11IdaesSolveCompletionEvent = CloudEventsPayload[IdaesSolveCompletionPayload] 

12DispatchMultiSolveEvent = CloudEventsPayload[MultiSolvePayload] 

13 

14 

15class CompletionStatus(StrEnum): 

16 SUCCESS = "success" 

17 ERROR = "error"