diff --git a/api/controllers/service_api/app/audio.py b/api/controllers/service_api/app/audio.py index dd27c99ee..fe2b72542 100644 --- a/api/controllers/service_api/app/audio.py +++ b/api/controllers/service_api/app/audio.py @@ -106,7 +106,7 @@ class TextApi(Resource): } ) @validate_app_token(fetch_user_arg=FetchUserArg(fetch_from=WhereisUserArg.JSON)) - def post(self, app_model: App, end_user: EndUser): + def post(self, app_model: App, end_user: EndUser, api_token: ApiToken): # 二开部分End - 密钥额度限制,新增api_token """Convert text to audio using text-to-speech. Converts the provided text to audio using the specified voice. diff --git a/api/controllers/service_api/app/completion.py b/api/controllers/service_api/app/completion.py index 7aaa7a32c..3b27d318a 100644 --- a/api/controllers/service_api/app/completion.py +++ b/api/controllers/service_api/app/completion.py @@ -30,6 +30,7 @@ from libs import helper from libs.helper import uuid_value from models.model import ApiToken, App, AppMode, EndUser # 二开部分End - 密钥额度限制,新增ApiToken from services.app_generate_service import AppGenerateService +from services.app_generate_service_extend import AppGenerateServiceExtend from services.errors.app import IsDraftWorkflowError, WorkflowIdFormatError, WorkflowNotFoundError from services.errors.llm import InvokeRateLimitError @@ -204,6 +205,13 @@ class ChatApi(Resource): # # ------------------- 二开部分End - 密钥额度限制 ------------------- try: + # ------------------- 二开部分Begin - 添加使用统计 ------------------- + AppGenerateServiceExtend.calculate_cumulative_usage( + app_model=app_model, + args=args, + ) # Extend: App + # ------------------- 二开部分End - 添加使用统计 ------------------- + response = AppGenerateService.generate( app_model=app_model, user=end_user, args=args, invoke_from=InvokeFrom.SERVICE_API, streaming=streaming ) diff --git a/web/app/components/header/account-money-extend/index.tsx b/web/app/components/header/account-money-extend/index.tsx index 3d8c4da4f..214e99bf7 100644 --- a/web/app/components/header/account-money-extend/index.tsx +++ b/web/app/components/header/account-money-extend/index.tsx @@ -5,7 +5,7 @@ import type { UserMoney } from '@/models/common-extend' import cn from 'classnames' const AccountMoneyExtend = () => { - const [userMoney, setUserMoney] = useState({ used_quota: 0, total_quota: 15 }) + const [userMoney, setUserMoney] = useState({ used_quota: 0, total_quota: 0 }) const [isFetched, setIsFetched] = useState(false) const exchangeRate = 7.26 // 美元转人民币固定汇率 @@ -24,9 +24,13 @@ const AccountMoneyExtend = () => { // 计算额度(确保使用数字类型) const usedQuota = Number(userMoney.used_quota) || 0 - const totalQuota = Number(userMoney.total_quota) || 15 + const totalQuota = Number(userMoney.total_quota) || 0 const remainingQuota = totalQuota - usedQuota + // 当总额度为0时不显示 + if (totalQuota === 0) + return null + // 转换为人民币并保留2位小数 const usedRMB = (usedQuota * exchangeRate).toFixed(2) const totalRMB = (totalQuota * exchangeRate).toFixed(2)