From 44abbabb135c9ad978fe64f720262bf4420f38ad Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Mon, 17 Jun 2024 16:34:29 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=93=9D=20(model.py):=20import=20Message?=
 =?UTF-8?q?=20class=20from=20langflow.schema.message=20to=20fix=20the=20re?=
 =?UTF-8?q?ference=20error=20=E2=99=BB=EF=B8=8F=20(utils.py):=20remove=20u?=
 =?UTF-8?q?nnecessary=20whitespace=20to=20improve=20code=20readability?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/base/langflow/base/models/model.py | 3 ++-
 src/backend/base/langflow/graph/utils.py       | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/backend/base/langflow/base/models/model.py b/src/backend/base/langflow/base/models/model.py
index a8069dcf9..eb2f4b36c 100644
--- a/src/backend/base/langflow/base/models/model.py
+++ b/src/backend/base/langflow/base/models/model.py
@@ -7,6 +7,7 @@ from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
 
 from langflow.custom import Component
 from langflow.field_typing.prompt import Prompt
+from langflow.schema.message import Message
 
 
 class LCModelComponent(Component):
@@ -92,7 +93,7 @@ class LCModelComponent(Component):
         if system_message:
             messages.append(SystemMessage(content=system_message))
         if input_value:
-            if isinstance(input_value, Prompt):
+            if isinstance(input_value, Message):
                 with warnings.catch_warnings():
                     warnings.simplefilter("ignore")
                     if "prompt" in input_value:
diff --git a/src/backend/base/langflow/graph/utils.py b/src/backend/base/langflow/graph/utils.py
index 6d1dca0cb..7183b923f 100644
--- a/src/backend/base/langflow/graph/utils.py
+++ b/src/backend/base/langflow/graph/utils.py
@@ -54,6 +54,7 @@ def flatten_list(list_of_lists: list[Union[list, Any]]) -> list:
 def serialize_field(value):
     """Unified serialization function for handling both BaseModel and Document types,
     including handling lists of these types."""
+
     if isinstance(value, (list, tuple)):
         return [serialize_field(v) for v in value]
     elif isinstance(value, Document):