aborted status

This commit is contained in:
Robert 2024-03-22 03:49:39 +07:00
parent 8d7fd0e36c
commit e30b9b3c14
No known key found for this signature in database
GPG Key ID: F631C7FD957D5F22

View File

@ -11,6 +11,7 @@ class ProgressActionStatusEnum(str, Enum):
PENDING = "pending" PENDING = "pending"
DONE = "done" DONE = "done"
FAILED = "failed" FAILED = "failed"
ABORTED = "aborted"
class BaseProgressAction(BaseModel): class BaseProgressAction(BaseModel):
@ -50,6 +51,7 @@ Action = Annotated[
class ProgressChain(BaseMongoModel): class ProgressChain(BaseMongoModel):
task_id: int task_id: int
chain_id: str
namespace_id: str namespace_id: str
variables: Annotated[Optional[Dict], Field(default={})] variables: Annotated[Optional[Dict], Field(default={})]
recipients: Annotated[Optional[List[int]], Field(default=[])] recipients: Annotated[Optional[List[int]], Field(default=[])]
@ -62,7 +64,6 @@ class ProgressChain(BaseMongoModel):
cls, cls,
chain: Chain, chain: Chain,
task_id: int, task_id: int,
namespace_id: str,
variables: Optional[Dict] = {}, variables: Optional[Dict] = {},
recipients: Optional[List[int]] = [], recipients: Optional[List[int]] = [],
): ):
@ -78,7 +79,8 @@ class ProgressChain(BaseMongoModel):
return ProgressChain( return ProgressChain(
task_id=task_id, task_id=task_id,
namespace_id=namespace_id, chain_id=str(chain.id),
namespace_id=chain.namespace_id,
variables=variables, variables=variables,
recipients=recipients, recipients=recipients,
name=chain.name, name=chain.name,