From 5405809e901877d4b980930f5c550db9dd9ff45d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 22 Jun 2024 00:37:39 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20(custom=5Fcomponent.py):=20fix?= =?UTF-8?q?=20an=20issue=20where=20the=20code=20was=20trying=20to=20access?= =?UTF-8?q?=20model=5Fdump=20method=20on=20message=20object=20without=20ch?= =?UTF-8?q?ecking=20if=20it=20is=20an=20instance=20of=20BaseModel=20class?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../base/langflow/custom/custom_component/custom_component.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/base/langflow/custom/custom_component/custom_component.py b/src/backend/base/langflow/custom/custom_component/custom_component.py index a0a7813cd..369199148 100644 --- a/src/backend/base/langflow/custom/custom_component/custom_component.py +++ b/src/backend/base/langflow/custom/custom_component/custom_component.py @@ -489,7 +489,7 @@ class CustomComponent(BaseComponent): """ if name is None: 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() log = Log(message=message, type=get_artifact_type(message), name=name) self._logs.append(log)