Fixed prompt tokens not being accounted in CodeTF v2 to v3 conversion (#1087)

andrecsilva · web-flow · commit f0a3f1bb03b5 · 2025-08-06T10:19:09.000-03:00
Fixed prompt tokens not being accounted
diff --git 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,
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -158,6 +158,7 @@ def from_v2_aimetadata(ai_metadata: AIMetadatav2) -> AIMetadata:`
`158`	`158`	`models=[ai_metadata.model] if ai_metadata.model else None,`
`159`	`159`	`total_tokens=ai_metadata.tokens,`
`160`	`160`	`completion_tokens=ai_metadata.completion_tokens,`
	`161`	`+ prompt_tokens=ai_metadata.prompt_tokens,`
`161`	`162`	`)`
`162`	`163`
`163`	`164`