diff --git a/models/itm.py b/models/itm.py index 6b04182..8fd53e3 100644 --- a/models/itm.py +++ b/models/itm.py @@ -14,6 +14,13 @@ class ExtractModel(BaseModel): source: str = "body" +class AssertModel(BaseModel): + type: str + path: str | None = None + operator: str = "equals" + value: Any = None + + class StepModel(BaseModel): name: str request: RequestModel