⚡️ Speed up method LangSmithTracer.add_trace by 18% in src/backend/base/langflow/services/tracing/langsmith.py

src/backend/base/langflow/graph/vertex/base.py

@@ -14,14 +14,16 @@
 from loguru import logger
 
 from langflow.exceptions.component import ComponentBuildError
-from langflow.graph.schema import INPUT_COMPONENTS, OUTPUT_COMPONENTS, InterfaceComponentTypes, ResultData
+from langflow.graph.schema import (INPUT_COMPONENTS, OUTPUT_COMPONENTS,
+                                   InterfaceComponentTypes, ResultData)
 from langflow.graph.utils import UnbuiltObject, UnbuiltResult, log_transaction
 from langflow.interface import initialize
 from langflow.interface.listing import lazy_load_dict
 from langflow.schema.artifact import ArtifactType
 from langflow.schema.data import Data
 from langflow.schema.message import Message
-from langflow.schema.schema import INPUT_FIELD_NAME, OutputValue, build_output_logs
+from langflow.schema.schema import (INPUT_FIELD_NAME, OutputValue,
+                                    build_output_logs)
 from langflow.services.deps import get_storage_service
 from langflow.utils.constants import DIRECT_TYPES
 from langflow.utils.schemas import ChatOutputResponse

src/backend/base/langflow/services/tracing/langsmith.py

@@ -5,9 +5,11 @@
 import types
 from datetime import datetime, timezone
 from typing import TYPE_CHECKING, Any
+from uuid import UUID
 
 from loguru import logger
 
+from langflow.graph.vertex.base import Vertex
 from langflow.schema.data import Data
 from langflow.services.tracing.base import BaseTracer
 
@@ -73,24 +75,23 @@ def add_trace(
     ) -> None:
         if not self._ready or not self._run_tree:
             return
-        processed_inputs = {}
-        if inputs:
-            processed_inputs = self._convert_to_langchain_types(inputs)
+
+        processed_inputs = self._convert_to_langchain_types(inputs) if inputs else {}
+
         child = self._run_tree.create_child(
             name=trace_name,
             run_type=trace_type,  # type: ignore[arg-type]
             inputs=processed_inputs,
         )
+
         if metadata:
             child.add_metadata(self._convert_to_langchain_types(metadata))
+
         self._children[trace_name] = child
         self._child_link: dict[str, str] = {}
 
     def _convert_to_langchain_types(self, io_dict: dict[str, Any]):
-        converted = {}
-        for key, value in io_dict.items():
-            converted[key] = self._convert_to_langchain_type(value)
-        return converted
+        return {key: self._convert_to_langchain_type(value) for key, value in io_dict.items()}
 
     def _convert_to_langchain_type(self, value):
         from langflow.schema.message import Message