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

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

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