From 2078ee5a4c7c37dfec4c360d6da50432a5f8413c Mon Sep 17 00:00:00 2001
From: clavedeluna <danalitovsky+git@gmail.com>
Date: Tue, 4 Feb 2025 11:03:51 -0300
Subject: [PATCH] add tokens fields to codetf

---
 src/codemodder/codetf.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/codemodder/codetf.py b/src/codemodder/codetf.py
index 779d319b..5ccf4389 100644
--- 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):