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

🐛 (custom_component.py): fix an issue where the code was trying to access model_dump method on message object without checking if it is an instance of BaseModel class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-22 00:37:39 -03:00
commit 5405809e90
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -489,7 +489,7 @@ class CustomComponent(BaseComponent):
""" """
if name is None: if name is None:
name = self.display_name if self.display_name else self.__class__.__name__ name = self.display_name if self.display_name else self.__class__.__name__
if hasattr(message, "model_dump"): if hasattr(message, "model_dump") and isinstance(message, BaseModel):
message = message.model_dump() message = message.model_dump()
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)