File size: 613 Bytes
3d833be |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from pydantic import BaseModel, Field
from typing import List, Optional, Literal
class FileImpact(BaseModel):
files_impacted: str
impact_details: str
class ImpactAnalysis(BaseModel):
files_impacted: List[FileImpact]
number_of_files_impacted: int
severity_level: Optional[Literal["Low", "Medium", "High"]] = Field(description="possible values: Low, Medium, High based on the 'number_of_files_impacted' impacted. E.g. if 'number_of_files_impacted' > 0 but < 3 then LOW, if 'number_of_files_impacted' > 3 but < 8 then MEDIUM, if 'number_of_files_impacted' > 8 then HIGH.")
|