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

refactor: Update Component class to use private _tracing_service variable

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-21 20:21:44 -03:00
commit 9991c6ea4c
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/base/models/model.py
View file

@ -140,7 +140,7 @@ class LCModelComponent(Component):
inputs: Union[list, dict] = messages or {}
try:
runnable = runnable.with_config(
{"run_name": self.display_name, "project_name": self.tracing_service.project_name}
{"run_name": self.display_name, "project_name": self._tracing_service.project_name}
)
if stream:
return runnable.stream(inputs)

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

@ -118,11 +118,11 @@ class Component(CustomComponent):
async def build_results(self):
inputs = self.get_trace_as_inputs()
metadata = self.get_trace_as_metadata()
with self.tracing_service.trace_context(
with self._tracing_service.trace_context(
f"{self.display_name} ({self.vertex.id})", self.trace_type, inputs, metadata
):
_results, _artifacts = await self._build_results()
self.tracing_service.set_outputs(_results)
self._tracing_service.set_outputs(_results)
return _results, _artifacts
@ -180,8 +180,8 @@ class Component(CustomComponent):
_artifacts[output.name] = artifact
self._artifacts = _artifacts
self._results = _results
if self.tracing_service:
self.tracing_service.set_outputs(_results)
if self._tracing_service:
self._tracing_service.set_outputs(_results)
return _results, _artifacts
def custom_repr(self):