From da5eec8fc70250c81ab8d9f8aa301b4279605019 Mon Sep 17 00:00:00 2001 From: italojohnny Date: Sat, 22 Jun 2024 11:26:03 -0300 Subject: [PATCH] fix: add trace_name parameter to method call --- .../base/langflow/custom/custom_component/component.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/custom/custom_component/component.py b/src/backend/base/langflow/custom/custom_component/component.py index d6d6a5af1..73938fb6c 100644 --- a/src/backend/base/langflow/custom/custom_component/component.py +++ b/src/backend/base/langflow/custom/custom_component/component.py @@ -122,7 +122,8 @@ class Component(CustomComponent): f"{self.display_name} ({self.vertex.id})", self.trace_type, inputs, metadata ): _results, _artifacts = await self._build_results() - self._tracing_service.set_outputs(_results) + trace_name = self._tracing_service.run_name + self._tracing_service.set_outputs(trace_name, _results) return _results, _artifacts @@ -181,7 +182,8 @@ class Component(CustomComponent): self._artifacts = _artifacts self._results = _results if self._tracing_service: - self._tracing_service.set_outputs(_results) + trace_name = self._tracing_service.run_name + self._tracing_service.set_outputs(trace_name, _results) return _results, _artifacts def custom_repr(self):