feat: mypy for all type check (#10921)
This commit is contained in:
@@ -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
|
||||
|
||||
|
Reference in New Issue
Block a user