Add new CodeTF metadata for "strategy" and "provisional" (#914)

pixee · Nov 14, 2024 · 4878ff0 · 4878ff0
1 parent 68b1653
commit 4878ff0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/codemodder/codetf.py b/src/codemodder/codetf.py
@@ -83,13 +83,21 @@ class AIMetadata(BaseModel):
     tokens: Optional[int] = None
 
 
+class Strategy(Enum):
+    ai = "ai"
+    hybrid = "hybrid"
+    deterministic = "deterministic"
+
+
 class ChangeSet(BaseModel):
     """A set of changes made to a file at `path`"""
 
     path: str
     diff: str
     changes: list[Change] = []
     ai: Optional[AIMetadata] = None
+    strategy: Optional[Strategy] = None
+    provisional: Optional[bool] = False
 
 
 class Reference(BaseModel):