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

chore(api/core): apply ruff reformatting (#7624)

Browse source
This commit is contained in:
Bowen Liang 2024-09-10 17:00:20 +08:00 • committed by GitHub
commit 2cf1187b32
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
724 changed files with 21180 additions and 21123 deletions
Download patch file Download diff file Expand all files Collapse all files

12
api/core/moderation/input_moderation.py
View file

@ -13,13 +13,14 @@ logger = logging.getLogger(__name__)
class InputModeration:
def check(
self, app_id: str,
self,
app_id: str,
tenant_id: str,
app_config: AppConfig,
inputs: dict,
query: str,
message_id: str,
trace_manager: Optional[TraceQueueManager] = None
trace_manager: Optional[TraceQueueManager] = None,
) -> tuple[bool, dict, str]:
"""
Process sensitive_word_avoidance.
@ -39,10 +40,7 @@ class InputModeration:
moderation_type = sensitive_word_avoidance_config.type
moderation_factory = ModerationFactory(
name=moderation_type,
app_id=app_id,
tenant_id=tenant_id,
config=sensitive_word_avoidance_config.config
name=moderation_type, app_id=app_id, tenant_id=tenant_id, config=sensitive_word_avoidance_config.config
)
with measure_time() as timer:
@ -55,7 +53,7 @@ class InputModeration:
message_id=message_id,
moderation_result=moderation_result,
inputs=inputs,
timer=timer
timer=timer,
)
)