feat: mypy for all type check (#10921)

This commit is contained in:
yihong
2024-12-24 18:38:51 +08:00
committed by GitHub
parent c91e8b1737
commit 56e15d09a9
584 changed files with 3975 additions and 2826 deletions

View File

@@ -1,5 +1,6 @@
import logging
from typing import Optional
from collections.abc import Mapping
from typing import Any, Optional
from core.app.app_config.entities import AppConfig
from core.moderation.base import ModerationAction, ModerationError
@@ -17,11 +18,11 @@ class InputModeration:
app_id: str,
tenant_id: str,
app_config: AppConfig,
inputs: dict,
inputs: Mapping[str, Any],
query: str,
message_id: str,
trace_manager: Optional[TraceQueueManager] = None,
) -> tuple[bool, dict, str]:
) -> tuple[bool, Mapping[str, Any], str]:
"""
Process sensitive_word_avoidance.
:param app_id: app id
@@ -33,6 +34,7 @@ class InputModeration:
:param trace_manager: trace manager
:return:
"""
inputs = dict(inputs)
if not app_config.sensitive_word_avoidance:
return False, inputs, query