Add new CodeTF metadata for "strategy" and "provisional"

pixee · Nov 14, 2024 · fddf156 · fddf156
1 parent 68b1653
commit fddf156
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):