estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

refactor: Add @staticmethod decorator in api/core (#7652)

Browse Source
This commit is contained in:
Shota Totsuka
2024-08-26 20:45:03 +09:00
committed by GitHub
parent 1473083a41
commit 430e100142
4 changed files with 52 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
api/core/hosting_configuration.py
View File

@@ -58,7 +58,8 @@ class HostingConfiguration:
self.moderation_config = self.init_moderation_config(config)
def init_azure_openai(self, app_config: Config) -> HostingProvider:
@staticmethod
def init_azure_openai(app_config: Config) -> HostingProvider:
quota_unit = QuotaUnit.TIMES
if app_config.get("HOSTED_AZURE_OPENAI_ENABLED"):
credentials = {
@@ -145,7 +146,8 @@ class HostingConfiguration:
quota_unit=quota_unit,
)
def init_anthropic(self, app_config: Config) -> HostingProvider:
@staticmethod
def init_anthropic(app_config: Config) -> HostingProvider:
quota_unit = QuotaUnit.TOKENS
quotas = []
@@ -180,7 +182,8 @@ class HostingConfiguration:
quota_unit=quota_unit,
)
def init_minimax(self, app_config: Config) -> HostingProvider:
@staticmethod
def init_minimax(app_config: Config) -> HostingProvider:
quota_unit = QuotaUnit.TOKENS
if app_config.get("HOSTED_MINIMAX_ENABLED"):
quotas = [FreeHostingQuota()]
@@ -197,7 +200,8 @@ class HostingConfiguration:
quota_unit=quota_unit,
)
def init_spark(self, app_config: Config) -> HostingProvider:
@staticmethod
def init_spark(app_config: Config) -> HostingProvider:
quota_unit = QuotaUnit.TOKENS
if app_config.get("HOSTED_SPARK_ENABLED"):
quotas = [FreeHostingQuota()]
@@ -214,7 +218,8 @@ class HostingConfiguration:
quota_unit=quota_unit,
)
def init_zhipuai(self, app_config: Config) -> HostingProvider:
@staticmethod
def init_zhipuai(app_config: Config) -> HostingProvider:
quota_unit = QuotaUnit.TOKENS
if app_config.get("HOSTED_ZHIPUAI_ENABLED"):
quotas = [FreeHostingQuota()]
@@ -231,7 +236,8 @@ class HostingConfiguration:
quota_unit=quota_unit,
)
def init_moderation_config(self, app_config: Config) -> HostedModerationConfig:
@staticmethod
def init_moderation_config(app_config: Config) -> HostedModerationConfig:
if app_config.get("HOSTED_MODERATION_ENABLED") \
and app_config.get("HOSTED_MODERATION_PROVIDERS"):
return HostedModerationConfig(