jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
commit 430e100142
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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(