From 6eac507eb9bb64e26931bd5dfe3447b9fff34f56 Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Mon, 10 Jun 2024 14:11:43 -0300
Subject: [PATCH] feat: Update RecordsOutput build method to set status

---
 .../base/langflow/components/outputs/RecordsOutput.py       | 1 +
 .../outputModal/components/switchOutputView/index.tsx       | 2 +-
 .../src/modals/IOModal/components/IOFieldView/index.tsx     | 6 +++++-
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/backend/base/langflow/components/outputs/RecordsOutput.py b/src/backend/base/langflow/components/outputs/RecordsOutput.py
index c750e675b..cbd268ea6 100644
--- a/src/backend/base/langflow/components/outputs/RecordsOutput.py
+++ b/src/backend/base/langflow/components/outputs/RecordsOutput.py
@@ -16,4 +16,5 @@ class RecordOutput(CustomComponent):
         }
 
     def build(self, input_value: Record) -> Record:
+        self.status = input_value
         return input_value
diff --git a/src/frontend/src/CustomNodes/GenericNode/components/outputModal/components/switchOutputView/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/outputModal/components/switchOutputView/index.tsx
index 307c77c1a..437adb511 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/outputModal/components/switchOutputView/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/outputModal/components/switchOutputView/index.tsx
@@ -28,7 +28,7 @@ export default function SwitchOutputView(nodeId): JSX.Element {
   if (resultMessage.raw) {
     resultMessage = resultMessage.raw;
   }
-  console.log("resultType", results);
+
   return (
     <>
       <Case condition={!resultType || resultType === "unknown"}>
diff --git a/src/frontend/src/modals/IOModal/components/IOFieldView/index.tsx b/src/frontend/src/modals/IOModal/components/IOFieldView/index.tsx
index 20cf9f691..4de8d0687 100644
--- a/src/frontend/src/modals/IOModal/components/IOFieldView/index.tsx
+++ b/src/frontend/src/modals/IOModal/components/IOFieldView/index.tsx
@@ -254,7 +254,11 @@ export default function IOFieldView({
               <div className={left ? "h-56" : "h-full"}>
                 <RecordsOutputComponent
                   pagination={!left}
-                  rows={flowPoolNode?.data?.artifacts?.records ?? []}
+                  rows={
+                    flowPoolNode?.data?.artifacts?.map(
+                      (artifact) => artifact.data
+                    ) ?? []
+                  }
                   columnMode="union"
                 />
               </div>