feat: add ops trace (#5483)

Co-authored-by: takatost <takatost@gmail.com>

api/core/moderation/input_moderation.py

@@ -1,18 +1,25 @@
 import logging
+from typing import Optional
 
 from core.app.app_config.entities import AppConfig
 from core.moderation.base import ModerationAction, ModerationException
 from core.moderation.factory import ModerationFactory
+from core.ops.ops_trace_manager import TraceQueueManager, TraceTask, TraceTaskName
+from core.ops.utils import measure_time
 
 logger = logging.getLogger(__name__)
 
 
 class InputModeration:
-    def check(self, app_id: str,
-              tenant_id: str,
-              app_config: AppConfig,
-              inputs: dict,
-              query: str) -> tuple[bool, dict, str]:
+    def check(
+        self, app_id: str,
+        tenant_id: str,
+        app_config: AppConfig,
+        inputs: dict,
+        query: str,
+        message_id: str,
+        trace_manager: Optional[TraceQueueManager] = None
+    ) -> tuple[bool, dict, str]:
         """
         Process sensitive_word_avoidance.
         :param app_id: app id
@@ -20,6 +27,8 @@ class InputModeration:
         :param app_config: app config
         :param inputs: inputs
         :param query: query
+        :param message_id: message id
+        :param trace_manager: trace manager
         :return:
         """
         if not app_config.sensitive_word_avoidance:
@@ -35,8 +44,20 @@ class InputModeration:
             config=sensitive_word_avoidance_config.config
         )
 
-        moderation_result = moderation_factory.moderation_for_inputs(inputs, query)
+        with measure_time() as timer:
+            moderation_result = moderation_factory.moderation_for_inputs(inputs, query)
 
+        if trace_manager:
+            trace_manager.add_trace_task(
+                TraceTask(
+                    TraceTaskName.MODERATION_TRACE,
+                    message_id=message_id,
+                    moderation_result=moderation_result,
+                    inputs=inputs,
+                    timer=timer
+                )
+            )
+        
         if not moderation_result.flagged:
             return False, inputs, query