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

fix: create paid provider auto

Browse Source
This commit is contained in:
Yansong Zhang 2025-09-26 14:31:26 +08:00
parent 560fe8a0f6
commit 0360f0b33b
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/core/provider_manager.py
View File

@ -618,9 +618,9 @@ class ProviderManager:
) )
for quota in configuration.quotas: for quota in configuration.quotas:
if quota.quota_type == ProviderQuotaType.TRIAL: if quota.quota_type in (ProviderQuotaType.TRIAL, ProviderQuotaType.PAID):
# Init trial provider records if not exists # Init trial provider records if not exists
if ProviderQuotaType.TRIAL not in provider_quota_to_provider_record_dict: if quota.quota_type not in provider_quota_to_provider_record_dict:
try: try:
# FIXME ignore the type error, only TrialHostingQuota has limit need to change the logic # FIXME ignore the type error, only TrialHostingQuota has limit need to change the logic
new_provider_record = Provider( new_provider_record = Provider(
@ -628,7 +628,7 @@ class ProviderManager:
# TODO: Use provider name with prefix after the data migration. # TODO: Use provider name with prefix after the data migration.
provider_name=ModelProviderID(provider_name).provider_name, provider_name=ModelProviderID(provider_name).provider_name,
provider_type=ProviderType.SYSTEM.value, provider_type=ProviderType.SYSTEM.value,
quota_type=ProviderQuotaType.TRIAL.value, quota_type=quota.quota_type,
quota_limit=0, # type: ignore quota_limit=0, # type: ignore
quota_used=0, quota_used=0,
is_valid=True, is_valid=True,
@ -642,7 +642,7 @@ class ProviderManager:
Provider.tenant_id == tenant_id, Provider.tenant_id == tenant_id,
Provider.provider_name == ModelProviderID(provider_name).provider_name, Provider.provider_name == ModelProviderID(provider_name).provider_name,
Provider.provider_type == ProviderType.SYSTEM.value, Provider.provider_type == ProviderType.SYSTEM.value,
Provider.quota_type == ProviderQuotaType.TRIAL.value, Provider.quota_type == quota.quota_type,
) )
existed_provider_record = db.session.scalar(stmt) existed_provider_record = db.session.scalar(stmt)
if not existed_provider_record: if not existed_provider_record:
@ -918,11 +918,11 @@ class ProviderManager:
trail_pool = CreditPoolService.get_pool( trail_pool = CreditPoolService.get_pool(
tenant_id=tenant_id, tenant_id=tenant_id,
pool_type="trial", pool_type=ProviderQuotaType.TRIAL.value,
) )
paid_pool = CreditPoolService.get_pool( paid_pool = CreditPoolService.get_pool(
tenant_id=tenant_id, tenant_id=tenant_id,
pool_type="paid", pool_type=ProviderQuotaType.PAID.value,
) )
else: else:
trail_pool = None trail_pool = None