chore: simplify SectionsData definition

robusta-dev · Jan 30, 2025 · b244594 · b244594
1 parent 3d8c877
commit b244594
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/holmes/core/investigation_structured_output.py b/holmes/core/investigation_structured_output.py
@@ -7,8 +7,7 @@
 
 OutputSectionsDataType = Optional[Dict[str, Union[str, None]]]
 
-class SectionsData(RootModel):
-    root: OutputSectionsDataType
+SectionsData = RootModel[OutputSectionsDataType]
 
 DEFAULT_SECTIONS:InputSectionsDataType = {
     "Alert Explanation": "1-2 sentences explaining the alert itself - note don't say \"The alert indicates a warning event related to a Kubernetes pod doing blah\" rather just say \"The pod XYZ did blah\" because that is what the user actually cares about",