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

add paid credit

Browse Source
This commit is contained in:
Yansong Zhang 2025-09-26 12:49:35 +08:00
parent e974c696f7
commit d745c2e8e3
4 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/hosting_configuration.py
View File

@ -202,7 +202,7 @@ class HostingConfiguration:
if dify_config.HOSTED_ANTHROPIC_PAID_ENABLED: if dify_config.HOSTED_ANTHROPIC_PAID_ENABLED:
paid_quota = PaidHostingQuota() paid_quota = PaidHostingQuota()
paid_models = self.parse_restrict_models_from_env("HOSTED_ANTHROPIC_PAID_MODELS") paid_models = self.parse_restrict_models_from_env("HOSTED_ANTHROPIC_PAID_MODELS")
quotas.append(paid_quota,restrict_models=paid_models) quotas.append(paid_quota, restrict_models=paid_models)
if len(quotas) > 0: if len(quotas) > 0:
credentials = { credentials = {

4
api/core/provider_manager.py
View File

@ -930,7 +930,7 @@ class ProviderManager:
raise ValueError("quota_used is None") raise ValueError("quota_used is None")
if provider_record.quota_limit is None: if provider_record.quota_limit is None:
raise ValueError("quota_limit is None") raise ValueError("quota_limit is None")
if provider_quota.quota_type == ProviderQuotaType.TRIAL and trail_pool is not None: if provider_quota.quota_type == ProviderQuotaType.TRIAL and trail_pool is not None:
quota_configuration = QuotaConfiguration( quota_configuration = QuotaConfiguration(
quota_type=provider_quota.quota_type, quota_type=provider_quota.quota_type,
quota_unit=provider_hosting_configuration.quota_unit or QuotaUnit.TOKENS, quota_unit=provider_hosting_configuration.quota_unit or QuotaUnit.TOKENS,
@ -940,7 +940,7 @@ class ProviderManager:
restrict_models=provider_quota.restrict_models, restrict_models=provider_quota.restrict_models,
) )
elif provider_quota.quota_type == ProviderQuotaType.PAID and paid_pool is not None: elif provider_quota.quota_type == ProviderQuotaType.PAID and paid_pool is not None:
quota_configuration = QuotaConfiguration( quota_configuration = QuotaConfiguration(
quota_type=provider_quota.quota_type, quota_type=provider_quota.quota_type,
quota_unit=provider_hosting_configuration.quota_unit or QuotaUnit.TOKENS, quota_unit=provider_hosting_configuration.quota_unit or QuotaUnit.TOKENS,

4
api/core/workflow/nodes/llm/llm_utils.py
View File

@ -6,7 +6,7 @@ from sqlalchemy.orm import Session
from configs import dify_config from configs import dify_config
from core.app.entities.app_invoke_entities import ModelConfigWithCredentialsEntity from core.app.entities.app_invoke_entities import ModelConfigWithCredentialsEntity
from core.entities.provider_entities import QuotaUnit from core.entities.provider_entities import ProviderQuotaType, QuotaUnit
from core.file.models import File from core.file.models import File
from core.memory.token_buffer_memory import TokenBufferMemory from core.memory.token_buffer_memory import TokenBufferMemory
from core.model_manager import ModelInstance, ModelManager from core.model_manager import ModelInstance, ModelManager
@ -23,7 +23,7 @@ from libs.datetime_utils import naive_utc_now
from models.model import Conversation from models.model import Conversation
from models.provider import Provider, ProviderType from models.provider import Provider, ProviderType
from models.provider_ids import ModelProviderID from models.provider_ids import ModelProviderID
from core.entities.provider_entities import ProviderQuotaType
from .exc import InvalidVariableTypeError, LLMModeRequiredError, ModelNotExistError from .exc import InvalidVariableTypeError, LLMModeRequiredError, ModelNotExistError

2
api/services/credit_pool_service.py
View File

@ -2,12 +2,12 @@ import logging
from typing import Optional from typing import Optional
from sqlalchemy import update from sqlalchemy import update
from sqlalchemy.orm import Session
from configs import dify_config from configs import dify_config
from core.errors.error import QuotaExceededError from core.errors.error import QuotaExceededError
from extensions.ext_database import db from extensions.ext_database import db
from models import TenantCreditPool from models import TenantCreditPool
from sqlalchemy.orm import Session
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)