From 33af41d328678c1fb56c04c4feb6004673347ac1 Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Tue, 18 Jun 2024 10:33:29 -0300
Subject: [PATCH] refactor: Improve useValidationStatusString hook to handle
 multiple log items

---
 .../CustomNodes/hooks/use-validation-status-string.tsx    | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/frontend/src/CustomNodes/hooks/use-validation-status-string.tsx b/src/frontend/src/CustomNodes/hooks/use-validation-status-string.tsx
index e29cbc1b8..1f957df55 100644
--- a/src/frontend/src/CustomNodes/hooks/use-validation-status-string.tsx
+++ b/src/frontend/src/CustomNodes/hooks/use-validation-status-string.tsx
@@ -1,12 +1,16 @@
 import { useEffect } from "react";
+import { LogType, VertexBuildTypeAPI } from "../../types/api";
 
-const useValidationStatusString = (validationStatus, setValidationString) => {
+const useValidationStatusString = (validationStatus: VertexBuildTypeAPI, setValidationString) => {
   useEffect(() => {
     if (validationStatus?.data?.logs) {
       // if it is not a string turn it into a string
       console.log("validationStatus", validationStatus);
       let newValidationString = "";
-      Object.values(validationStatus?.data?.logs).forEach((log: any) => {
+      Object.values(validationStatus?.data?.logs).forEach((log: LogType | LogType[]) => {
+        if (!Array.isArray(log)) {
+          log = [log];
+        }
         log.forEach((logItem) => {
           if (logItem.type === "error" || logItem.type === "ValueError") {
             newValidationString += `${logItem.message}\n`;