more fields to IterationCohort

Author: Karthikeyannhs

src/eligibility_signposting_api/model/rules.py

@@ -29,6 +29,8 @@
 StartDate = NewType("StartDate", date)
 EndDate = NewType("EndDate", date)
 CohortLabel = NewType("CohortLabel", str)
+CohortGroup = NewType("CohortGroup", str)
+Description = NewType("Description", str)
 RuleStop = NewType("RuleStop", bool)
 
 
@@ -84,6 +86,9 @@ class RuleAttributeLevel(StrEnum):
 
 class IterationCohort(BaseModel):
     cohort_label: CohortLabel | None = Field(None, alias="CohortLabel")
+    cohort_group: CohortGroup | None = Field(None, alias="CohortGroup")
+    positive_description: Description | None = Field(None, alias="PositiveDescription")
+    negative_description: Description | None = Field(None, alias="NegativeDescription")
     priority: int | None = Field(None, alias="Priority")
 
     model_config = {"populate_by_name": True, "extra": "ignore"}