diff --git a/src/codemodder/codetf.py b/src/codemodder/codetf.py index 779d319b..8466727f 100644 --- a/src/codemodder/codetf.py +++ b/src/codemodder/codetf.py @@ -99,6 +99,17 @@ class Strategy(Enum): deterministic = "deterministic" +class Rating(BaseModel): + score: float + description: Optional[str] = None + + +class FixQuality(BaseModel): + safetyRating: Rating + effectivenessRating: Rating + cleanlinessRating: Rating + + class ChangeSet(BaseModel): """A set of changes made to a file at `path`""" @@ -110,6 +121,7 @@ class ChangeSet(BaseModel): provisional: Optional[bool] = False # For fixed findings that are not associated with a specific change fixedFindings: Optional[list[Finding]] = None + fixQuality: Optional[FixQuality] = None def with_changes(self, changes: list[Change]) -> ChangeSet: return ChangeSet(