feat: simplify adder

Author: WJPBProjects

src/agents/usage.py

@@ -1,34 +1,6 @@
 from dataclasses import dataclass, field
-from typing import TypeVar
 
 from openai.types.responses.response_usage import InputTokensDetails, OutputTokensDetails
-from pydantic import BaseModel
-
-T = TypeVar("T", bound=BaseModel)
-
-
-def add_numeric_fields(current: T, other: T) -> T:
-    """
-    Add numeric fields from other to current.
-    """
-    clone = current.model_copy()
-    for key, v1 in current.model_dump().items():
-        v2 = getattr(other, key, 0)
-        if isinstance(v1, (int, float)) and isinstance(v2, (int, float)):
-            setattr(clone, key, (v1 or 0) + (v2 or 0))
-    return clone
-
-
-def add_input_tokens_details(
-    current: InputTokensDetails, other: InputTokensDetails
-) -> InputTokensDetails:
-    return add_numeric_fields(current, other)
-
-
-def add_output_tokens_details(
-    current: OutputTokensDetails, other: OutputTokensDetails
-) -> OutputTokensDetails:
-    return add_numeric_fields(current, other)
 
 
 @dataclass
@@ -59,9 +31,12 @@ def add(self, other: "Usage") -> None:
         self.input_tokens += other.input_tokens if other.input_tokens else 0
         self.output_tokens += other.output_tokens if other.output_tokens else 0
         self.total_tokens += other.total_tokens if other.total_tokens else 0
-        self.input_tokens_details = add_input_tokens_details(
-            self.input_tokens_details, other.input_tokens_details
+        self.input_tokens_details = InputTokensDetails(
+            cached_tokens=self.input_tokens_details.cached_tokens
+            + other.input_tokens_details.cached_tokens
         )
-        self.output_tokens_details = add_output_tokens_details(
-            self.output_tokens_details, other.output_tokens_details
+
+        self.output_tokens_details = OutputTokensDetails(
+            reasoning_tokens=self.output_tokens_details.reasoning_tokens
+            + other.output_tokens_details.reasoning_tokens
         )