Skip to content

Commit 4878ff0

Browse files
authored
Add new CodeTF metadata for "strategy" and "provisional" (#914)
1 parent 68b1653 commit 4878ff0

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/codemodder/codetf.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,13 +83,21 @@ class AIMetadata(BaseModel):
8383
tokens: Optional[int] = None
8484

8585

86+
class Strategy(Enum):
87+
ai = "ai"
88+
hybrid = "hybrid"
89+
deterministic = "deterministic"
90+
91+
8692
class ChangeSet(BaseModel):
8793
"""A set of changes made to a file at `path`"""
8894

8995
path: str
9096
diff: str
9197
changes: list[Change] = []
9298
ai: Optional[AIMetadata] = None
99+
strategy: Optional[Strategy] = None
100+
provisional: Optional[bool] = False
93101

94102

95103
class Reference(BaseModel):

0 commit comments

Comments
 (0)