bizyengine 1.1.2__tar.gz → 1.2.1__tar.gz

{bizyengine-1.1.2 → bizyengine-1.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: bizyengine
-Version: 1.1.2
+Version: 1.2.1
 Summary: [a/BizyAir](https://github.com/siliconflow/BizyAir) Comfy Nodes that can run in any environment.
 Author-email: SiliconFlow <yaochi@siliconflow.cn>
 Project-URL: Repository, https://github.com/siliconflow/BizyAir

{bizyengine-1.1.2 → bizyengine-1.2.1}/bizyengine/bizy_server/api_client.py

@@ -6,15 +6,18 @@ import bizyengine.core as core
 from bizyengine.core.common import get_api_key
 from bizyengine.core.common.env_var import (
     BIZYAIR_PRODUCTION_TEST,
-    BIZYAIR_SERVER_ADDRESS,
+    BIZYAIR_X_SERVER,
+    BIZYAIR_Y_SERVER,
 )
 
 from .errno import ErrorNo, errnos
 from .error_handler import ErrorHandler
+from .profile import user_profile
 from .utils import is_string_valid
 
-with open(os.path.join(os.path.dirname(__file__), "..", "version.txt"), "r") as file:
-    CLIENT_VERSION = file.read()
+version_path = os.path.join(os.path.dirname(__file__), "..", "version.txt")
+with open(version_path, "r") as file:
+    CLIENT_VERSION = file.read().strip()
 
 
 class APIClient:
@@ -34,9 +37,13 @@ class APIClient:
                 "content-type": "application/json",
                 "authorization": auth,
                 "x-bizyair-client-version": CLIENT_VERSION,
+                "X-Client-Type": "bizyair",
             }
             if BIZYAIR_PRODUCTION_TEST != None:
                 headers["x-bizyair-production-test"] = BIZYAIR_PRODUCTION_TEST
+            if user_profile.getLang():
+                headers["lang"] = user_profile.getLang()
+
             return headers, None
         except ValueError as e:
             error_message = e.args[0] if e.args else "Invalid API key"
@@ -55,7 +62,11 @@ class APIClient:
                         response.status,
                         resp_json.get("code", response.status),
                         None,
-                        resp_json.get("message", await response.text()),
+                        {
+                            user_profile.getLang(): resp_json.get(
+                                "message", await response.text()
+                            )
+                        },
                     )
                 return resp_json, None
 
@@ -68,7 +79,11 @@ class APIClient:
                         response.status,
                         resp_json.get("code", response.status),
                         None,
-                        resp_json.get("message", await response.text()),
+                        {
+                            user_profile.getLang(): resp_json.get(
+                                "message", await response.text()
+                            )
+                        },
                     )
                 return resp_json, None
 
@@ -81,7 +96,11 @@ class APIClient:
                         response.status,
                         resp_json.get("code", response.status),
                         None,
-                        resp_json.get("message", await response.text()),
+                        {
+                            user_profile.getLang(): resp_json.get(
+                                "message", await response.text()
+                            )
+                        },
                     )
                 return resp_json, None
 
@@ -94,7 +113,11 @@ class APIClient:
                         response.status,
                         resp_json.get("code", response.status),
                         None,
-                        resp_json.get("message", await response.text()),
+                        {
+                            user_profile.getLang(): resp_json.get(
+                                "message", await response.text()
+                            )
+                        },
                     )
                 return resp_json, None
 
@@ -103,7 +126,7 @@ class APIClient:
         if err is not None:
             return None, err
 
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/user/info"
+        server_url = f"{BIZYAIR_X_SERVER}/user/info"
         try:
             ret, err = await self.do_get(server_url, headers=headers)
             if err is not None:
@@ -117,7 +140,7 @@ class APIClient:
     async def sign(
         self, signature: str, type: str
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/files/{signature}"
+        server_url = f"{BIZYAIR_X_SERVER}/files/{signature}"
         params = None
         if is_string_valid(type):
             params = {"type": type}
@@ -140,7 +163,7 @@ class APIClient:
     async def commit_file(
         self, signature: str, object_key: str, md5_hash: str, type: str
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/files"
+        server_url = f"{BIZYAIR_X_SERVER}/files"
 
         payload = {
             "sign": signature,
@@ -163,7 +186,7 @@ class APIClient:
             return None, errnos.COMMIT_FILE
 
     async def commit_bizy_model(self, payload) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/bizy_models"
+        server_url = f"{BIZYAIR_X_SERVER}/bizy_models"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -182,7 +205,7 @@ class APIClient:
     async def delete_bizy_model(
         self, model_id: int
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/{model_id}"
+        server_url = f"{BIZYAIR_X_SERVER}/bizy_models/{model_id}"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -207,7 +230,7 @@ class APIClient:
         base_models: list[str] = None,
         sort: str = None,
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/community"
+        server_url = f"{BIZYAIR_X_SERVER}/bizy_models/community"
         params = {"current": current, "page_size": page_size}
         if keyword:
             params["keyword"] = keyword
@@ -241,7 +264,7 @@ class APIClient:
         base_models: list[str] = None,
         sort: str = None,
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/official"
+        server_url = f"{BIZYAIR_X_SERVER}/bizy_models/official"
         params = {"current": current, "page_size": page_size}
         if keyword:
             params["keyword"] = keyword
@@ -276,7 +299,7 @@ class APIClient:
         base_models: list[str] = None,
         sort: str = None,
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/{mode}"
+        server_url = f"{BIZYAIR_X_SERVER}/bizy_models/{mode}"
         params = {"current": current, "page_size": page_size}
         if keyword:
             params["keyword"] = keyword
@@ -304,9 +327,7 @@ class APIClient:
     async def get_model_detail(
         self, model_id: int, source: str
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = (
-            f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/{model_id}/detail?source={source}"
-        )
+        server_url = f"{BIZYAIR_X_SERVER}/bizy_models/{model_id}/detail?source={source}"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -325,7 +346,7 @@ class APIClient:
     async def get_model_version_detail(
         self, version_id: int
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/versions/{version_id}"
+        server_url = f"{BIZYAIR_X_SERVER}/bizy_models/versions/{version_id}"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -344,7 +365,7 @@ class APIClient:
             return None, errnos.GET_MODEL_VERSION_DETAIL
 
     async def fork_model_version(self, version_id: int) -> tuple[None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/versions/{version_id}/fork"
+        server_url = f"{BIZYAIR_X_SERVER}/bizy_models/versions/{version_id}/fork"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -363,7 +384,7 @@ class APIClient:
     async def unfork_model_version(
         self, version_id: int
     ) -> tuple[None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/versions/{version_id}/fork"
+        server_url = f"{BIZYAIR_X_SERVER}/bizy_models/versions/{version_id}/fork"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -382,7 +403,7 @@ class APIClient:
     async def update_model(
         self, model_id: int, name: str, type_: str, versions: list[dict]
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/{model_id}"
+        server_url = f"{BIZYAIR_X_SERVER}/bizy_models/{model_id}"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -403,7 +424,7 @@ class APIClient:
     async def get_upload_token(
         self, filename: str
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/upload/token?file_name={filename}"
+        server_url = f"{BIZYAIR_X_SERVER}/upload/token?file_name={filename}"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -423,9 +444,7 @@ class APIClient:
         self, like_type: str, object_id: str
     ) -> tuple[dict | None, ErrorNo | None]:
         if like_type == "model_version":
-            server_url = (
-                f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/versions/{object_id}/like"
-            )
+            server_url = f"{BIZYAIR_X_SERVER}/bizy_models/versions/{object_id}/like"
         else:
             return None, errnos.UNSUPPORT_LIKE_TYPE
 
@@ -446,7 +465,7 @@ class APIClient:
     async def get_download_url(
         self, sign: str, model_version_id: int
     ) -> tuple[str | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/files/temp-download/{sign}?version_id={model_version_id}"
+        server_url = f"{BIZYAIR_X_SERVER}/files/temp-download/{sign}?version_id={model_version_id}"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -463,12 +482,14 @@ class APIClient:
             return None, errnos.GET_DOWNLOAD_URL
 
     async def get_share_model_files(self, shareId, payload) -> (dict, ErrorNo):
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/{shareId}/models/files"
+        server_url = f"{BIZYAIR_X_SERVER}/{shareId}/models/files"
         try:
 
             def callback(ret: dict):
                 if ret["code"] != errnos.OK.code:
-                    return [], ErrorNo(500, ret["code"], None, f"{ret}")
+                    return [], ErrorNo(
+                        500, ret["code"], None, {user_profile.getLang(): ret["message"]}
+                    )
                 if not ret or "data" not in ret or ret["data"] is None:
                     return [], None
 
@@ -492,7 +513,7 @@ class APIClient:
             return [], errnos.LIST_SHARE_MODEL_FILE_ERR
 
     async def commit_dataset(self, payload) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/datasets"
+        server_url = f"{BIZYAIR_X_SERVER}/datasets"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -511,7 +532,7 @@ class APIClient:
     async def update_dataset(
         self, dataset_id: int, name: str, versions: list[dict]
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/datasets/{dataset_id}"
+        server_url = f"{BIZYAIR_X_SERVER}/datasets/{dataset_id}"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -532,7 +553,7 @@ class APIClient:
     async def get_dataset_version_detail(
         self, version_id: int
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/datasets/versions/{version_id}"
+        server_url = f"{BIZYAIR_X_SERVER}/datasets/versions/{version_id}"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -553,7 +574,7 @@ class APIClient:
     async def delete_dataset(
         self, dataset_id: int
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/datasets/{dataset_id}"
+        server_url = f"{BIZYAIR_X_SERVER}/datasets/{dataset_id}"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -576,7 +597,7 @@ class APIClient:
         keyword: str = None,
         annotated: str = None,
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/datasets"
+        server_url = f"{BIZYAIR_X_SERVER}/datasets"
         params = {"current": current, "page_size": page_size}
         if keyword:
             params["keyword"] = keyword
@@ -600,7 +621,7 @@ class APIClient:
     async def get_dataset_detail(
         self, dataset_id: int
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/datasets/{dataset_id}/detail"
+        server_url = f"{BIZYAIR_X_SERVER}/datasets/{dataset_id}/detail"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -617,7 +638,7 @@ class APIClient:
             return None, errnos.GET_DATASET_DETAIL
 
     async def create_share(self, payload) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/share"
+        server_url = f"{BIZYAIR_X_SERVER}/share"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -633,7 +654,7 @@ class APIClient:
             return None, errnos.CREATE_SHARE
 
     async def get_share_detail(self, code: str) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/share/{code}"
+        server_url = f"{BIZYAIR_X_SERVER}/share/{code}"
 
         headers, err = self.auth_header()
         if err is not None:
@@ -650,7 +671,7 @@ class APIClient:
             return None, errnos.GET_SHARE_DETAIL
 
     async def get_data_dict(self) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/dict"
+        server_url = f"{BIZYAIR_X_SERVER}/dict"
         headers, err = self.auth_header()
         if err is not None:
             return None, err
@@ -666,7 +687,7 @@ class APIClient:
             return None, errnos.GET_DATA_DICT
 
     async def get_notification_unread_count(self) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/notifications/unread_count"
+        server_url = f"{BIZYAIR_X_SERVER}/notifications/unread_count"
         headers, err = self.auth_header()
         if err is not None:
             return None, err
@@ -691,7 +712,7 @@ class APIClient:
         types: list | None,
         read_status: bool | None,
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/notifications"
+        server_url = f"{BIZYAIR_X_SERVER}/notifications"
         headers, err = self.auth_header()
         if err is not None:
             return None, err
@@ -719,7 +740,7 @@ class APIClient:
     async def read_notifications(
         self, notif_ids: list[int]
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/notifications/read"
+        server_url = f"{BIZYAIR_X_SERVER}/notifications/read"
         headers, err = self.auth_header()
         if err is not None:
             return None, err
@@ -744,7 +765,7 @@ class APIClient:
     async def read_all_notifications(
         self, type: int
     ) -> tuple[dict | None, ErrorNo | None]:
-        server_url = f"{BIZYAIR_SERVER_ADDRESS}/notifications/read_all"
+        server_url = f"{BIZYAIR_X_SERVER}/notifications/read_all"
         headers, err = self.auth_header()
         if err is not None:
             return None, err
@@ -764,3 +785,289 @@ class APIClient:
                 f"\033[31m[BizyAir]\033[0m Fail to mark all notifications as read: {str(e)}"
             )
             return None, errnos.READ_ALL_NOTIF
+
+    async def get_wallet(self) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/wallet"
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_get(server_url, headers=headers)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to get wallet info: {str(e)}")
+            return None, errnos.GET_WALLET_INFO
+
+    async def query_coins(
+        self, current: int, page_size: int, coin_type: int = 0, expire_days: int = 0
+    ) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/coins"
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        params = {"current": current, "page_size": page_size}
+        if coin_type > 0:
+            params["coin_type"] = coin_type
+        if expire_days > 0:
+            params["expire_days"] = expire_days
+
+        try:
+            ret, err = await self.do_get(server_url, headers=headers, params=params)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to query coins: {str(e)}")
+            return None, errnos.QUERY_COINS
+
+    async def get_user_metadata(self) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_X_SERVER}/user/metadata"
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_get(server_url, headers=headers)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to get user metadata: {str(e)}")
+            return None, errnos.GET_USER_METADATA
+
+    async def update_user_info(
+        self, name: str = None, avatar: str = None, introduction: str = None
+    ) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_X_SERVER}/user/metadata"
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        payload = {}
+        if name is not None:
+            payload["name"] = name
+        if avatar is not None:
+            payload["avatar"] = avatar
+        if introduction is not None:
+            payload["introduction"] = introduction
+
+        try:
+            ret, err = await self.do_put(server_url, headers=headers, data=payload)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to update user info: {str(e)}")
+            return None, errnos.UPDATE_USER_INFO
+
+    async def user_real_name(self) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_X_SERVER}/user/real_name"
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_post(server_url, headers=headers)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to verify real name: {str(e)}")
+            return None, errnos.USER_REAL_NAME
+
+    async def buy_product(
+        self, product_id: int, platform: str
+    ) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/pay/product/{product_id}"
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        payload = {}
+        if platform is not None:
+            payload["platform"] = platform
+
+        try:
+            ret, err = await self.do_post(server_url, headers=headers, data=payload)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to buy product: {str(e)}")
+            return None, errnos.BUY_PRODUCT
+
+    async def get_pay_status(self, orderNum: str) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/pay/orders/{orderNum}"
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_get(server_url, headers=headers)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to get pay status: {str(e)}")
+            return None, errnos.PAY_STATUS
+
+    async def cancel_pay_order(
+        self, orderNum: str
+    ) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/pay/orders/{orderNum}"
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_delete(server_url, headers=headers)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to cancel payment: {str(e)}")
+            return None, errnos.PAY_CANCEL
+
+    async def list_pay_orders(
+        self, current: int, page_size: int, status: str | None
+    ) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/pay/page"
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        params = {"current": current, "page_size": page_size}
+        if status is not None:
+            params["status"] = status
+
+        try:
+            ret, err = await self.do_get(server_url, headers=headers, params=params)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to list pay orders: {str(e)}")
+            return None, errnos.LIST_PAY_ORDER
+
+    async def list_products(self) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/products"
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_get(server_url, headers=headers)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to list products: {str(e)}")
+            return None, errnos.LIST_PRODUCTS
+
+    async def get_year_cost(
+        self, year: str = None, api_key: str = None
+    ) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/invoices/year_cost"
+        params = {}
+        if year:
+            params["year"] = year
+        if api_key:
+            params["api_key"] = api_key
+
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_get(server_url, params=params, headers=headers)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to get year cost: {str(e)}")
+            return None, errnos.GET_YEAR_COST
+
+    async def get_month_cost(
+        self, month: str = None, api_key: str = None
+    ) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/invoices/month_cost"
+        params = {}
+        if month:
+            params["month"] = month
+        if api_key:
+            params["api_key"] = api_key
+
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_get(server_url, params=params, headers=headers)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to get month cost: {str(e)}")
+            return None, errnos.GET_MONTH_COST
+
+    async def get_day_cost(
+        self, day: str = None, api_key: str = None
+    ) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/invoices/day_cost"
+        params = {}
+        if day:
+            params["date"] = day
+        if api_key:
+            params["api_key"] = api_key
+
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_get(server_url, params=params, headers=headers)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to get day cost: {str(e)}")
+            return None, errnos.GET_DAY_COST
+
+    async def get_recent_cost(
+        self, api_key: str = None
+    ) -> tuple[dict | None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_Y_SERVER}/invoices/recent_cost"
+        params = {}
+        if api_key:
+            params["api_key"] = api_key
+
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_get(server_url, params=params, headers=headers)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to get recent cost: {str(e)}")
+            return None, errnos.GET_RECENT_COST