From e30b9b3c142da2ca4bc02f78c7447a578ab5f126 Mon Sep 17 00:00:00 2001 From: Robert Date: Fri, 22 Mar 2024 03:49:39 +0700 Subject: [PATCH] aborted status --- chain_service/database/models/progress_chain.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/chain_service/database/models/progress_chain.py b/chain_service/database/models/progress_chain.py index 6128ac5..77ec348 100644 --- a/chain_service/database/models/progress_chain.py +++ b/chain_service/database/models/progress_chain.py @@ -11,6 +11,7 @@ class ProgressActionStatusEnum(str, Enum): PENDING = "pending" DONE = "done" FAILED = "failed" + ABORTED = "aborted" class BaseProgressAction(BaseModel): @@ -50,6 +51,7 @@ Action = Annotated[ class ProgressChain(BaseMongoModel): task_id: int + chain_id: str namespace_id: str variables: Annotated[Optional[Dict], Field(default={})] recipients: Annotated[Optional[List[int]], Field(default=[])] @@ -62,7 +64,6 @@ class ProgressChain(BaseMongoModel): cls, chain: Chain, task_id: int, - namespace_id: str, variables: Optional[Dict] = {}, recipients: Optional[List[int]] = [], ): @@ -78,7 +79,8 @@ class ProgressChain(BaseMongoModel): return ProgressChain( task_id=task_id, - namespace_id=namespace_id, + chain_id=str(chain.id), + namespace_id=chain.namespace_id, variables=variables, recipients=recipients, name=chain.name,