add tokens fields to codetf (#988)

pixee · Feb 4, 2025 · bef0620 · bef0620
1 parent 4ff7bd7
commit bef0620
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/codemodder/codetf.py b/src/codemodder/codetf.py
@@ -91,6 +91,8 @@ class AIMetadata(BaseModel):
     provider: Optional[str] = None
     model: Optional[str] = None
     tokens: Optional[int] = None
+    completion_tokens: Optional[int] = None
+    prompt_tokens: Optional[int] = None
 
 
 class Strategy(Enum):