hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-10 11:26:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: server multi models support (#799)

Browse Source
This commit is contained in:
takatost
2023-08-12 00:57:00 +08:00
committed by GitHub
parent d8b712b325
commit 5fa2161b05
213 changed files with 10556 additions and 2579 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
api/services/workspace_service.py
View File

@@ -1,6 +1,6 @@
from extensions.ext_database import db
from models.account import Tenant
from models.provider import Provider, ProviderType, ProviderName
from models.provider import Provider
class WorkspaceService:
@@ -13,8 +13,8 @@ class WorkspaceService:
'status': tenant.status,
'created_at': tenant.created_at,
'providers': [],
'in_trail': False,
'trial_end_reason': 'using_custom'
'in_trial': True,
'trial_end_reason': None
}
# Get providers
@@ -25,25 +25,4 @@ class WorkspaceService:
# Add providers to the tenant info
tenant_info['providers'] = providers
custom_provider = None
system_provider = None
for provider in providers:
if provider.provider_type == ProviderType.CUSTOM.value:
if provider.is_valid and provider.encrypted_config:
custom_provider = provider
elif provider.provider_type == ProviderType.SYSTEM.value:
if provider.provider_name == ProviderName.OPENAI.value and provider.is_valid:
system_provider = provider
if system_provider and not custom_provider:
quota_used = system_provider.quota_used if system_provider.quota_used is not None else 0
quota_limit = system_provider.quota_limit if system_provider.quota_limit is not None else 0
if quota_used >= quota_limit:
tenant_info['trial_end_reason'] = 'trial_exceeded'
else:
tenant_info['in_trail'] = True
tenant_info['trial_end_reason'] = None
return tenant_info