From 7d6b99cf03daee83c47bf1f763204435b20d04c9 Mon Sep 17 00:00:00 2001 From: "renovate[bot]" <29139614+renovate[bot]@users.noreply.github.com> Date: Wed, 9 Jul 2025 07:26:50 -0300 Subject: [PATCH 1/2] Update dependency types-defusedxml to v0.7.0.20250708 (#1080) Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com> --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 40172f4f..28f84fbf 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -58,7 +58,7 @@ test = [ "coverage>=7.9,<7.10", "coverage-threshold~=0.4", "defusedxml==0.7.1", - "types-defusedxml==0.7.0.20250516", + "types-defusedxml==0.7.0.20250708", "flask-wtf==1.2.2", "types-WTForms==3.2.1.20250602", "Flask<4", From 287477ed21ab4e8f77b25510fa1e86ba48ee93f6 Mon Sep 17 00:00:00 2001 From: Dani Alcala <112832187+clavedeluna@users.noreply.github.com> Date: Thu, 10 Jul 2025 09:32:29 -0300 Subject: [PATCH 2/2] Report token_usage in FixResult (#1081) report token usage --- src/codemodder/codetf/v3/codetf.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/src/codemodder/codetf/v3/codetf.py b/src/codemodder/codetf/v3/codetf.py index f5ec09b8..b4036472 100644 --- a/src/codemodder/codetf/v3/codetf.py +++ b/src/codemodder/codetf/v3/codetf.py @@ -5,6 +5,7 @@ from pydantic import BaseModel, model_validator +from codemodder.llm import TokenUsage from codemodder.logging import logger from ..common import Change, CodeTFWriter, Finding, FixQuality @@ -119,6 +120,27 @@ def validate_fixMetadata(self): raise ValueError("fixMetadata must be provided for fixed results") return self + def store_token_data(self, token_usage: TokenUsage): + if token_usage == TokenUsage() or not self.fixMetadata: + return + + if not self.fixMetadata.generation.ai: + self.fixMetadata.generation.ai = AIMetadata( + total_tokens=token_usage.total, + completion_tokens=token_usage.completion_tokens, + prompt_tokens=token_usage.prompt_tokens, + ) + else: + self.fixMetadata.generation.ai.completion_tokens = ( + self.fixMetadata.generation.ai.completion_tokens or 0 + ) + token_usage.completion_tokens + self.fixMetadata.generation.ai.prompt_tokens = ( + self.fixMetadata.generation.ai.prompt_tokens or 0 + ) + token_usage.prompt_tokens + self.fixMetadata.generation.ai.total_tokens = ( + self.fixMetadata.generation.ai.total_tokens or 0 + ) + token_usage.total + class CodeTF(CodeTFWriter, BaseModel): run: Run