From 0806626f5181776ccf5ef69f7b80cb01e9efde31 Mon Sep 17 00:00:00 2001 From: Robert Date: Fri, 8 Mar 2024 22:45:29 +0700 Subject: [PATCH] refactored config --- chain_service/database/models/chain.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/chain_service/database/models/chain.py b/chain_service/database/models/chain.py index aef11b4..9c6b1bf 100644 --- a/chain_service/database/models/chain.py +++ b/chain_service/database/models/chain.py @@ -8,23 +8,20 @@ from typing import Literal, Union, Annotated, Optional, List class BaseAction(BaseModel): action_type: Literal["wait", "comment"] + class Config(BaseConfig): + pass + class WaitAction(BaseAction): action_type: Literal["wait"] wait_for: int - class Config(BaseConfig): - pass - class CommentAction(BaseAction): action_type: Literal["comment"] text: Annotated[Optional[str], Field(default=None)] file_urls: Annotated[Optional[List[str]], Field(default=[])] - class Config(BaseConfig): - pass - Action = Annotated[Union[WaitAction, CommentAction], Field(description="action_type")]