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 {} inputs: Union[list, dict] = messages or {}
try: try:
runnable = runnable.with_config( 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: if stream:
return runnable.stream(inputs) 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): async def build_results(self):
inputs = self.get_trace_as_inputs() inputs = self.get_trace_as_inputs()
metadata = self.get_trace_as_metadata() 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 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) self._tracing_service.set_outputs(_results)
return _results, _artifacts return _results, _artifacts
@ -180,8 +180,8 @@ class Component(CustomComponent):
_artifacts[output.name] = artifact _artifacts[output.name] = artifact
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) self._tracing_service.set_outputs(_results)
return _results, _artifacts return _results, _artifacts
def custom_repr(self): def custom_repr(self):