youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-22 11:47:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

[autofix.ci] apply automated fixes

Browse Source
This commit is contained in:
autofix-ci[bot]
2026-03-20 05:45:30 +00:00
committed by GitHub
parent 658ac15589
commit 3888969af3
2 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
api/enums/quota_type.py
View File

@ -55,7 +55,10 @@ class QuotaCharge:
self._committed = True
logger.debug(
"Committed %s quota for tenant %s, reservation_id: %s, amount: %d",
self._quota_type.value, self._tenant_id, self.charge_id, amount,
self._quota_type.value,
self._tenant_id,
self.charge_id,
amount,
)
except Exception:
logger.exception("Failed to commit quota, reservation_id: %s", self.charge_id)
@ -169,13 +172,18 @@ class QuotaType(StrEnum):
if not reservation_id:
logger.warning(
"Reserve returned no reservation_id for %s, feature %s, response: %s",
tenant_id, self.value, reserve_resp,
tenant_id,
self.value,
reserve_resp,
)
raise QuotaExceededError(feature=self.value, tenant_id=tenant_id, required=amount)
logger.debug(
"Reserved %d %s quota for tenant %s, reservation_id: %s",
amount, self.value, tenant_id, reservation_id,
amount,
self.value,
tenant_id,
reservation_id,
)
return QuotaCharge(
success=True,

14
api/services/billing_service.py
View File

@ -95,11 +95,15 @@ class BillingService:
Returns:
{"available": int, "reserved": int, "released": int}
"""
return cls._send_request("POST", "/quota/release", json={
"tenant_id": tenant_id,
"feature_key": feature_key,
"reservation_id": reservation_id,
})
return cls._send_request(
"POST",
"/quota/release",
json={
"tenant_id": tenant_id,
"feature_key": feature_key,
"reservation_id": reservation_id,
},
)
@classmethod
def get_knowledge_rate_limit(cls, tenant_id: str):