Usage add method

Author: szl97

api/pom.xml

@@ -6,7 +6,7 @@
     <parent>
         <groupId>top.bella</groupId>
         <artifactId>openai-java</artifactId>
-        <version>0.23.25</version>
+        <version>0.23.26</version>
     </parent>
     <packaging>jar</packaging>
     <artifactId>openai-api</artifactId>

api/src/main/java/com/theokanning/openai/CompletionTokensDetails.java

@@ -20,4 +20,14 @@ public class CompletionTokensDetails {
 
     @JsonProperty("rejected_prediction_tokens")
     long rejectedPredictionTokens;
+
+    public void add(CompletionTokensDetails details) {
+        if(details == null) {
+            return;
+        }
+        this.reasoningTokens += details.getReasoningTokens();
+        this.audioTokens += details.getAudioTokens();
+        this.acceptedPredictionTokens += details.getAcceptedPredictionTokens();
+        this.rejectedPredictionTokens += details.getRejectedPredictionTokens();
+    }
 }

api/src/main/java/com/theokanning/openai/PromptTokensDetails.java

@@ -19,4 +19,12 @@ public class PromptTokensDetails {
      */
     @JsonProperty("audio_tokens")
     long audioTokens;
+
+    public void add(PromptTokensDetails details) {
+        if(details == null) {
+            return;
+        }
+        this.cachedTokens += details.getCachedTokens();
+        this.audioTokens += details.getAudioTokens();
+    }
 }

api/src/main/java/com/theokanning/openai/Usage.java

@@ -37,4 +37,27 @@ public class Usage {
      */
     @JsonProperty("completion_tokens_details")
     CompletionTokensDetails completionTokensDetails;
+
+    public void add(Usage usage) {
+        if(usage == null) {
+            return;
+        }
+        this.promptTokens += usage.getPromptTokens();
+        this.completionTokens += usage.getCompletionTokens();
+        this.totalTokens += usage.getTotalTokens();
+        if(usage.getPromptTokensDetails() != null) {
+            if(this.promptTokensDetails == null) {
+                this.promptTokensDetails = usage.getPromptTokensDetails();
+            } else {
+                this.promptTokensDetails.add(usage.getPromptTokensDetails());
+            }
+        }
+        if(usage.getCompletionTokensDetails() != null) {
+            if(this.completionTokensDetails == null) {
+                this.completionTokensDetails = usage.getCompletionTokensDetails();
+            } else {
+                this.completionTokensDetails.add(usage.getCompletionTokensDetails());
+            }
+        }
+    }
 }

client/pom.xml

@@ -6,7 +6,7 @@
     <parent>
         <groupId>top.bella</groupId>
         <artifactId>openai-java</artifactId>
-        <version>0.23.25</version>
+        <version>0.23.26</version>
     </parent>
     <packaging>jar</packaging>
 

pom.xml

@@ -5,7 +5,7 @@
 
     <groupId>top.bella</groupId>
     <artifactId>openai-java</artifactId>
-    <version>0.23.25</version>
+    <version>0.23.26</version>
     <packaging>pom</packaging>
     <description>openai java 版本</description>
     <name>openai-java</name>

service/pom.xml

@@ -6,7 +6,7 @@
     <parent>
         <groupId>top.bella</groupId>
         <artifactId>openai-java</artifactId>
-        <version>0.23.25</version>
+        <version>0.23.26</version>
     </parent>
     <packaging>jar</packaging>