fix(api): schema validation problem in mapping rule in dto (#1995)

keephq · Sep 25, 2024 · b512179 · b512179
1 parent da3398c
commit b512179
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/keep/api/models/db/mapping.py b/keep/api/models/db/mapping.py
@@ -60,8 +60,8 @@ class MappingRuleDtoOut(MappRuleDtoBase, extra="ignore"):
 class MappingRuleDtoIn(MappRuleDtoBase):
     rows: Optional[list[dict]] = None
 
-    @validator("type", pre=True, always=True)
-    def validate_type(cls, _type, values):
-        if _type == "csv" and not values.get("rows"):
+    @validator("rows", pre=True, always=True)
+    def validate_rows(cls, rows, values):
+        if not rows and values.get("type") == "csv":
             raise ValueError("Mapping of type CSV cannot have empty rows")
-        return _type
+        return rows