opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-28 11:56:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

add credit next_credit_reset_date

Browse Source
This commit is contained in:
Yansong Zhang 2025-10-11 11:45:18 +08:00
parent 09998612e7
commit c0ed353c10
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/services/feature_service.py
View File

@ -131,6 +131,7 @@ class FeatureModel(BaseModel):
# pydantic configs # pydantic configs
model_config = ConfigDict(protected_namespaces=()) model_config = ConfigDict(protected_namespaces=())
knowledge_pipeline: KnowledgePipeline = KnowledgePipeline() knowledge_pipeline: KnowledgePipeline = KnowledgePipeline()
next_credit_reset_date: int = 0
class KnowledgeRateLimitModel(BaseModel): class KnowledgeRateLimitModel(BaseModel):
@ -278,6 +279,9 @@ class FeatureService:
if "knowledge_pipeline_publish_enabled" in billing_info: if "knowledge_pipeline_publish_enabled" in billing_info:
features.knowledge_pipeline.publish_enabled = billing_info["knowledge_pipeline_publish_enabled"] features.knowledge_pipeline.publish_enabled = billing_info["knowledge_pipeline_publish_enabled"]
if "next_credit_reset_date" in billing_info:
features.next_credit_reset_date = billing_info["next_credit_reset_date"]
@classmethod @classmethod
def _fulfill_params_from_enterprise(cls, features: SystemFeatureModel): def _fulfill_params_from_enterprise(cls, features: SystemFeatureModel):

7
api/services/workspace_service.py
View File

@ -1,3 +1,4 @@
from dify.api.configs import feature
from flask_login import current_user from flask_login import current_user
from configs import dify_config from configs import dify_config
@ -31,7 +32,8 @@ class WorkspaceService:
assert tenant_account_join is not None, "TenantAccountJoin not found" assert tenant_account_join is not None, "TenantAccountJoin not found"
tenant_info["role"] = tenant_account_join.role tenant_info["role"] = tenant_account_join.role
can_replace_logo = FeatureService.get_features(tenant.id).can_replace_logo feature = FeatureService.get_features(tenant.id)
can_replace_logo = feature.can_replace_logo
if can_replace_logo and TenantService.has_roles(tenant, [TenantAccountRole.OWNER, TenantAccountRole.ADMIN]): if can_replace_logo and TenantService.has_roles(tenant, [TenantAccountRole.OWNER, TenantAccountRole.ADMIN]):
base_url = dify_config.FILES_URL base_url = dify_config.FILES_URL
@ -47,6 +49,9 @@ class WorkspaceService:
"replace_webapp_logo": replace_webapp_logo, "replace_webapp_logo": replace_webapp_logo,
} }
if dify_config.EDITION == "CLOUD": if dify_config.EDITION == "CLOUD":
tenant_info["next_credit_reset_date"] = feature.next_credit_reset_date
from services.credit_pool_service import CreditPoolService from services.credit_pool_service import CreditPoolService
paid_pool = CreditPoolService.get_pool(tenant_id=tenant.id, pool_type="paid") paid_pool = CreditPoolService.get_pool(tenant_id=tenant.id, pool_type="paid")