From f16d3bcb92097cab5470dc1d24cb3f2586faa58d Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Thu, 6 Jun 2024 14:18:53 -0300
Subject: [PATCH] Refactor Record class to handle file inputs in to_lc_message
 method

---
 src/backend/base/langflow/schema/schema.py | 34 ++++++++--------------
 1 file changed, 12 insertions(+), 22 deletions(-)

diff --git a/src/backend/base/langflow/schema/schema.py b/src/backend/base/langflow/schema/schema.py
index e3394ec5e..df62691c3 100644
--- a/src/backend/base/langflow/schema/schema.py
+++ b/src/backend/base/langflow/schema/schema.py
@@ -126,33 +126,23 @@ class Record(BaseModel):
             raise ValueError(f"Missing required keys ('text', 'sender') in Record: {self.data}")
         sender = self.data.get("sender", "Machine")
         text = self.data.get("text", "")
-        if sender == "User":
-            return HumanMessage(content=text)
-        return AIMessage(content=text)
-
-    def to_lc_messages(self):
-        """
-        Converts the Record to a list of BaseMessage.
-
-        Returns:
-            list[BaseMessage]: The converted list of BaseMessage.
-        """
-        if not all(key in self.data for key in ["text", "sender"]):
-            raise ValueError(f"Missing required keys ('text', 'sender') in Record: {self.data}")
-        sender = self.data.get("sender", "Machine")
-        text = self.data.get("text", "")
         files = self.data.get("files", [])
         if sender == "User":
             if files:
-                human_messages = [HumanMessage(content=text)]
-                for base64_image in files:
+                contents = [{"type": "text", "text": text}]
+                for file_path in files:
                     image_template = ImagePromptTemplate()
-                    human_message = image_template.invoke(url=f"data:image/png;base64,{base64_image}")
-                    human_messages.append(human_message)
-                return human_messages
+                    image_prompt_value = image_template.invoke(input={"path": file_path})
+                    contents.append({"type": "image_url", "image_url": image_prompt_value.image_url})
+                human_message = HumanMessage(content=contents)
             else:
-                return [HumanMessage(content=text)]
-        return [AIMessage(content=text)]
+                human_message = HumanMessage(
+                    content=[{"type": "text", "text": text}],
+                )
+
+            return human_message
+
+        return AIMessage(content=text)
 
     def __getattr__(self, key):
         """