jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: add trace_name parameter to method call

Browse source
This commit is contained in:
italojohnny 2024-06-22 11:26:03 -03:00 • committed by Gabriel Luiz Freitas Almeida
commit da5eec8fc7
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/custom/custom_component/component.py
View file

@ -122,7 +122,8 @@ class Component(CustomComponent):
f"{self.display_name} ({self.vertex.id})", self.trace_type, inputs, metadata f"{self.display_name} ({self.vertex.id})", self.trace_type, inputs, metadata
): ):
_results, _artifacts = await self._build_results() _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 return _results, _artifacts
@ -181,7 +182,8 @@ class Component(CustomComponent):
self._artifacts = _artifacts self._artifacts = _artifacts
self._results = _results self._results = _results
if self._tracing_service: 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 return _results, _artifacts
def custom_repr(self): def custom_repr(self):