From 6e7a5fa7bdef2305e8b510ac80e09bca7d2754f0 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 21 Jun 2024 20:21:13 -0300 Subject: [PATCH] refactor: Update CustomComponent to use private _tracing_service variable --- .../base/langflow/custom/custom_component/custom_component.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/custom/custom_component/custom_component.py b/src/backend/base/langflow/custom/custom_component/custom_component.py index 16a4cecc3..acdbd0803 100644 --- a/src/backend/base/langflow/custom/custom_component/custom_component.py +++ b/src/backend/base/langflow/custom/custom_component/custom_component.py @@ -85,7 +85,7 @@ class CustomComponent(BaseComponent): _flows_data: Optional[List[Data]] = None _outputs: List[OutputLog] = [] _logs: List[Log] = [] - tracing_service: Optional["TracingService"] = None + _tracing_service: "TracingService" def update_state(self, name: str, value: Any): if not self.vertex: @@ -494,4 +494,4 @@ class CustomComponent(BaseComponent): name = self.display_name log = Log(message=message, type=get_artifact_type(message), name=name) self._logs.append(log) - self.tracing_service.add_log(trace_name=self.vertex.id, log=log) + self._tracing_service.add_log(trace_name=self.vertex.id, log=log)