30 lines
721 B
Python
30 lines
721 B
Python
from .base import BaseMongoModel
|
|
|
|
from datetime import datetime
|
|
from pydantic import BaseModel, Field
|
|
from typing import Literal, Union, Annotated, Optional, List
|
|
|
|
|
|
class WaitAction(BaseModel):
|
|
action_type: Literal["wait"]
|
|
wait_for: int
|
|
|
|
|
|
class TextCommentAction(BaseModel):
|
|
action_type: Literal["text_comment"]
|
|
text: str
|
|
|
|
|
|
Action = Annotated[
|
|
Union[WaitAction, TextCommentAction], Field(description="action_type")
|
|
]
|
|
|
|
|
|
class Chain(BaseMongoModel):
|
|
name: Annotated[Optional[str], Field(default=None)]
|
|
actions: Annotated[Optional[List[Action]], Field(default=[])]
|
|
created_at: Annotated[datetime, Field(default_factory=datetime.utcnow)]
|
|
|
|
class Config(BaseMongoModel.Config):
|
|
pass
|