From 9c9a7f4aa10d6843021dc51518131a9a93c2420b Mon Sep 17 00:00:00 2001
From: italojohnny <italojohnnydosanjos@gmail.com>
Date: Tue, 28 May 2024 11:50:49 -0300
Subject: [PATCH] return artifact type to front

---
 src/backend/base/langflow/api/v1/chat.py       | 5 +++--
 src/backend/base/langflow/api/v1/schemas.py    | 1 +
 src/backend/base/langflow/graph/graph/base.py  | 4 ++--
 src/backend/base/langflow/graph/vertex/base.py | 5 ++++-
 4 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/backend/base/langflow/api/v1/chat.py b/src/backend/base/langflow/api/v1/chat.py
index 2ae9e7d9f..e4bccad8e 100644
--- a/src/backend/base/langflow/api/v1/chat.py
+++ b/src/backend/base/langflow/api/v1/chat.py
@@ -171,7 +171,7 @@ async def build_vertex(
                 result_dict,
                 log_message,
                 valid,
-                _,
+                log_type,
                 vertex,
             ) = await graph.build_vertex(
                 lock=lock,
@@ -186,6 +186,7 @@ async def build_vertex(
         except Exception as exc:
             logger.exception(f"Error building vertex: {exc}")
             log_message = format_exception_message(exc)
+            log_type = type(exc).__name__
             valid = False
             result_data_response = ResultDataResponse(results={})
 
@@ -193,7 +194,7 @@ async def build_vertex(
             # we need to clear the cache
             await chat_service.clear_cache(flow_id_str)
 
-        log_object = Log(message=log_message)
+        log_object = Log(message=log_message, type=log_type)
         result_data_response.logs.append(log_object)
 
         # Log the vertex build
diff --git a/src/backend/base/langflow/api/v1/schemas.py b/src/backend/base/langflow/api/v1/schemas.py
index b091b7ef1..dd0bac9ee 100644
--- a/src/backend/base/langflow/api/v1/schemas.py
+++ b/src/backend/base/langflow/api/v1/schemas.py
@@ -247,6 +247,7 @@ class VerticesOrderResponse(BaseModel):
 
 class Log(TypedDict):
     message: str
+    type: str
 
 
 class ResultDataResponse(BaseModel):
diff --git a/src/backend/base/langflow/graph/graph/base.py b/src/backend/base/langflow/graph/graph/base.py
index 37215bdeb..781b76141 100644
--- a/src/backend/base/langflow/graph/graph/base.py
+++ b/src/backend/base/langflow/graph/graph/base.py
@@ -735,16 +735,16 @@ class Graph:
 
             if vertex.result is not None:
                 params = vertex._built_object_repr()
+                log_type = vertex.artifacts_type
                 valid = True
                 result_dict = vertex.result
-                artifacts = vertex.artifacts
             else:
                 raise ValueError(f"No result found for vertex {vertex_id}")
 
             next_runnable_vertices, top_level_vertices = await self.get_next_and_top_level_vertices(
                 lock, set_cache_coro, vertex
             )
-            return next_runnable_vertices, top_level_vertices, result_dict, params, valid, artifacts, vertex
+            return next_runnable_vertices, top_level_vertices, result_dict, params, valid, log_type, vertex
         except Exception as exc:
             logger.exception(f"Error building vertex: {exc}")
             raise exc
diff --git a/src/backend/base/langflow/graph/vertex/base.py b/src/backend/base/langflow/graph/vertex/base.py
index 48cfa82ce..6ad5bc826 100644
--- a/src/backend/base/langflow/graph/vertex/base.py
+++ b/src/backend/base/langflow/graph/vertex/base.py
@@ -9,7 +9,7 @@ from typing import TYPE_CHECKING, Any, AsyncIterator, Callable, Dict, Iterator,
 from loguru import logger
 
 from langflow.graph.schema import INPUT_COMPONENTS, OUTPUT_COMPONENTS, InterfaceComponentTypes, ResultData
-from langflow.graph.utils import UnbuiltObject, UnbuiltResult
+from langflow.graph.utils import UnbuiltObject, UnbuiltResult, ArtifactType
 from langflow.graph.vertex.utils import log_transaction
 from langflow.interface.initialize import loading
 from langflow.interface.listing import lazy_load_dict
@@ -63,6 +63,7 @@ class Vertex:
         self._built_result = None
         self._built = False
         self.artifacts: Dict[str, Any] = {}
+        self.artifacts_type: Optional[str] = None
         self.steps: List[Callable] = [self._build]
         self.steps_ran: List[Callable] = []
         self.task_id: Optional[str] = None
@@ -625,6 +626,8 @@ class Vertex:
                 self._built_object, self.artifacts = result
             elif len(result) == 3:
                 self._custom_component, self._built_object, self.artifacts = result
+                self.artifacts_type = self.artifacts.get("type") or ArtifactType.UNKNOWN.value
+
         else:
             self._built_object = result