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,