New features: token calculation logic and account information query logic. (#311)

* feat: Tool for calculating token consumption quantity. API for querying account balance

* feat:

* feat: update jtokkit

---------

Co-authored-by: 大名（张瑜） <[email protected]>

Author: aaronuu

api/build.gradle

@@ -4,6 +4,7 @@ apply plugin: "com.vanniktech.maven.publish"
 dependencies {
     api libs.jacksonAnnotations
     api libs.jacksonDatabind
+    api libs.jtokkit
     compileOnly libs.lombok
     annotationProcessor libs.lombok
 

api/src/main/java/com/theokanning/openai/billing/BillingUsage.java

@@ -0,0 +1,29 @@
+package com.theokanning.openai.billing;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.math.BigDecimal;
+import java.util.List;
+
+/**
+ * Amount consumption information
+ *
+ */
+@Data
+public class BillingUsage {
+
+    @JsonProperty("object")
+    private String object;
+    /**
+     * Account expenditure details
+     */
+    @JsonProperty("daily_costs")
+    private List<DailyCost> dailyCosts;
+    /**
+     * Total usage amount: cents
+     */
+    @JsonProperty("total_usage")
+    private BigDecimal totalUsage;
+
+}

api/src/main/java/com/theokanning/openai/billing/CreditGrantsResponse.java

@@ -0,0 +1,35 @@
+package com.theokanning.openai.billing;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.io.Serializable;
+import java.math.BigDecimal;
+
+/**
+ * Return value of balance inquiry interface
+ *
+ */
+@Data
+public class CreditGrantsResponse implements Serializable {
+    private String object;
+    /**
+     * Total amount: US dollars
+     */
+    @JsonProperty("total_granted")
+    private BigDecimal totalGranted;
+    /**
+     * Total usage amount: US dollars
+     */
+    @JsonProperty("total_used")
+    private BigDecimal totalUsed;
+    /**
+     * Total remaining amount: US dollars
+     */
+    @JsonProperty("total_available")
+    private BigDecimal totalAvailable;
+    /**
+     * Balance details
+     */
+    private Grants grants;
+}

api/src/main/java/com/theokanning/openai/billing/DailyCost.java

@@ -0,0 +1,24 @@
+package com.theokanning.openai.billing;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ * List of amount consumption
+ *
+ */
+@Data
+public class DailyCost {
+    /**
+     *
+     */
+    @JsonProperty("timestamp")
+    private long timestamp;
+    /**
+     * Model consumption amount details
+     */
+    @JsonProperty("line_items")
+    private List<LineItem> lineItems;
+}

api/src/main/java/com/theokanning/openai/billing/Datum.java

@@ -0,0 +1,36 @@
+package com.theokanning.openai.billing;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.math.BigDecimal;
+
+/**
+ *
+ *
+ */
+@Data
+public class Datum {
+    private String object;
+    private String id;
+    /**
+     * Gift amount: US dollars
+     */
+    @JsonProperty("grant_amount")
+    private BigDecimal grantAmount;
+    /**
+     * Usage amount: US dollars
+     */
+    @JsonProperty("used_amount")
+    private BigDecimal usedAmount;
+    /**
+     * Effective timestamp
+     */
+    @JsonProperty("effective_at")
+    private Long effectiveAt;
+    /**
+     * Expiration timestamp
+     */
+    @JsonProperty("expires_at")
+    private Long expiresAt;
+}

api/src/main/java/com/theokanning/openai/billing/Grants.java

@@ -0,0 +1,17 @@
+package com.theokanning.openai.billing;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ *
+ *
+ */
+@Data
+public class Grants {
+    private String object;
+    @JsonProperty("data")
+    private List<Datum> data;
+}

api/src/main/java/com/theokanning/openai/billing/LineItem.java

@@ -0,0 +1,21 @@
+package com.theokanning.openai.billing;
+
+import lombok.Data;
+
+import java.math.BigDecimal;
+
+/**
+ * List of amount consumption
+ *
+ */
+@Data
+public class LineItem {
+    /**
+     * model name
+     */
+    private String name;
+    /**
+     * Expenditure amount
+     */
+    private BigDecimal cost;
+}

api/src/main/java/com/theokanning/openai/billing/Plan.java

@@ -0,0 +1,13 @@
+package com.theokanning.openai.billing;
+
+import lombok.Data;
+
+/**
+ *
+ *
+ */
+@Data
+public class Plan {
+    private String title;
+    private String id;
+}

api/src/main/java/com/theokanning/openai/billing/Subscription.java

@@ -0,0 +1,51 @@
+package com.theokanning.openai.billing;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+/**
+ * Account information
+ *
+ *
+ */
+@Data
+public class Subscription {
+    @JsonProperty("object")
+    private String object;
+    @JsonProperty("has_payment_method")
+    private boolean hasPaymentMethod;
+    @JsonProperty("canceled")
+    private boolean canceled;
+    @JsonProperty("canceled_at")
+    private Object canceledAt;
+    @JsonProperty("delinquent")
+    private Object delinquent;
+    @JsonProperty("access_until")
+    private long accessUntil;
+    @JsonProperty("soft_limit")
+    private long softLimit;
+    @JsonProperty("hard_limit")
+    private long hardLimit;
+    @JsonProperty("system_hard_limit")
+    private long systemHardLimit;
+    @JsonProperty("soft_limit_usd")
+    private double softLimitUsd;
+    @JsonProperty("hard_limit_usd")
+    private double hardLimitUsd;
+    @JsonProperty("system_hard_limit_usd")
+    private double systemHardLimitUsd;
+    @JsonProperty("plan")
+    private Plan plan;
+    @JsonProperty("account_name")
+    private String accountName;
+    @JsonProperty("po_number")
+    private Object poNumber;
+    @JsonProperty("billing_email")
+    private Object billingEmail;
+    @JsonProperty("tax_ids")
+    private Object taxIds;
+    @JsonProperty("billing_address")
+    private Object billingAddress;
+    @JsonProperty("business_address")
+    private Object businessAddress;
+}