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

🐛 (custom_component.py): fix issue where name was not being set correctly when name is None and self.display_name is True

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-27 15:20:42 -03:00
commit 293adb0417
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/base/langflow/custom/custom_component/custom_component.py
View file

@ -488,11 +488,14 @@ class CustomComponent(BaseComponent):
Args:
message (LoggableType | list[LoggableType]): The message to log.
"""
if name is None:
if name is None and self.display_name:
name = self.display_name
else:
name = self.__class__.__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)
if self.tracing_service and self.vertex:
self.tracing_service.add_log(trace_name=self.vertex.id, log=log)
def post_code_processing(self, new_build_config: dict, current_build_config: dict):
"""