Skip to content

Commit bef0620

Browse files
authored
add tokens fields to codetf (#988)
1 parent 4ff7bd7 commit bef0620

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/codemodder/codetf.py

+2
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,8 @@ class AIMetadata(BaseModel):
9191
provider: Optional[str] = None
9292
model: Optional[str] = None
9393
tokens: Optional[int] = None
94+
completion_tokens: Optional[int] = None
95+
prompt_tokens: Optional[int] = None
9496

9597

9698
class Strategy(Enum):

0 commit comments

Comments
 (0)