From a3852ce4767ec3f985e5074788e780c96fcf7402 Mon Sep 17 00:00:00 2001
From: andrecs <12188364+andrecsilva@users.noreply.github.com>
Date: Wed, 6 Aug 2025 09:32:03 -0300
Subject: [PATCH] Fixed prompt tokens not being accounted

---
 src/codemodder/codetf/v3/codetf.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/codemodder/codetf/v3/codetf.py b/src/codemodder/codetf/v3/codetf.py
index d60e140c..f06a3fbd 100644
--- a/src/codemodder/codetf/v3/codetf.py
+++ b/src/codemodder/codetf/v3/codetf.py
@@ -158,6 +158,7 @@ def from_v2_aimetadata(ai_metadata: AIMetadatav2) -> AIMetadata:
         models=[ai_metadata.model] if ai_metadata.model else None,
         total_tokens=ai_metadata.tokens,
         completion_tokens=ai_metadata.completion_tokens,
+        prompt_tokens=ai_metadata.prompt_tokens,
     )