From fddf156384840a0fe8514c295c34d5cc0715825c Mon Sep 17 00:00:00 2001 From: Daniel D'Avella Date: Thu, 14 Nov 2024 12:38:32 -0500 Subject: [PATCH] Add new CodeTF metadata for "strategy" and "provisional" --- src/codemodder/codetf.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/codemodder/codetf.py b/src/codemodder/codetf.py index 331cb06a..0fc9ea07 100644 --- a/src/codemodder/codetf.py +++ b/src/codemodder/codetf.py @@ -83,6 +83,12 @@ 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`""" @@ -90,6 +96,8 @@ class ChangeSet(BaseModel): diff: str changes: list[Change] = [] ai: Optional[AIMetadata] = None + strategy: Optional[Strategy] = None + provisional: Optional[bool] = False class Reference(BaseModel):