feat: add ops trace (#5483)

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

api/core/callback_handler/agent_tool_callback_handler.py

@@ -3,6 +3,8 @@ from typing import Any, Optional, TextIO, Union
 
 from pydantic import BaseModel
 
+from core.ops.ops_trace_manager import TraceQueueManager, TraceTask, TraceTaskName
+
 _TEXT_COLOR_MAPPING = {
     "blue": "36;1",
     "yellow": "33;1",
@@ -51,6 +53,9 @@ class DifyAgentCallbackHandler(BaseModel):
         tool_name: str,
         tool_inputs: dict[str, Any],
         tool_outputs: str,
+        message_id: Optional[str] = None,
+        timer: Optional[Any] = None,
+        trace_manager: Optional[TraceQueueManager] = None
     ) -> None:
         """If not the final action, print out observation."""
         print_text("\n[on_tool_end]\n", color=self.color)
@@ -59,6 +64,18 @@ class DifyAgentCallbackHandler(BaseModel):
         print_text("Outputs: " + str(tool_outputs)[:1000] + "\n", color=self.color)
         print_text("\n")
 
+        if trace_manager:
+            trace_manager.add_trace_task(
+                TraceTask(
+                    TraceTaskName.TOOL_TRACE,
+                    message_id=message_id,
+                    tool_name=tool_name,
+                    tool_inputs=tool_inputs,
+                    tool_outputs=tool_outputs,
+                    timer=timer,
+                )
+            )
+
     def on_tool_error(
         self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
     ) -> None: