AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-02 06:10:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(billing_service): change retry condition to handle specific reque… (#12091)

Browse Source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
Co-authored-by: Garfield Dai <dai.hai@foxmail.com>
This commit is contained in:
-LAN- 2024-12-25 18:40:27 +08:00 committed by GitHub
parent a1c78ad9d4
commit 017d7538ae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/services/billing_service.py
View File

@ -2,7 +2,7 @@ import os
from typing import Optional from typing import Optional
import httpx import httpx
from tenacity import retry, retry_if_not_exception_type, stop_before_delay, wait_fixed from tenacity import retry, retry_if_exception_type, stop_before_delay, wait_fixed
from extensions.ext_database import db from extensions.ext_database import db
from models.account import TenantAccountJoin, TenantAccountRole from models.account import TenantAccountJoin, TenantAccountRole
@ -44,7 +44,7 @@ class BillingService:
@retry( @retry(
wait=wait_fixed(2), wait=wait_fixed(2),
stop=stop_before_delay(10), stop=stop_before_delay(10),
retry=retry_if_not_exception_type(httpx.RequestError), retry=retry_if_exception_type(httpx.RequestError),
reraise=True, reraise=True,
) )
def _send_request(cls, method, endpoint, json=None, params=None): def _send_request(cls, method, endpoint, json=None, params=None):

2
api/services/feature_service.py
View File

@ -74,7 +74,7 @@ class FeatureService:
cls._fulfill_params_from_env(features) cls._fulfill_params_from_env(features)
if dify_config.BILLING_ENABLED: if dify_config.BILLING_ENABLED and tenant_id:
cls._fulfill_params_from_billing_api(features, tenant_id) cls._fulfill_params_from_billing_api(features, tenant_id)
return features return features