From d05e7c12f4c2659082a7f64b629042a49d8d009f Mon Sep 17 00:00:00 2001
From: italojohnny <italojohnnydosanjos@gmail.com>
Date: Wed, 29 May 2024 16:12:53 -0300
Subject: [PATCH] add new type to Log

---
 src/backend/base/langflow/api/v1/schemas.py |  2 +-
 src/backend/base/langflow/graph/utils.py    | 11 ++++++-----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/backend/base/langflow/api/v1/schemas.py b/src/backend/base/langflow/api/v1/schemas.py
index 123be10ae..4030e8637 100644
--- a/src/backend/base/langflow/api/v1/schemas.py
+++ b/src/backend/base/langflow/api/v1/schemas.py
@@ -246,7 +246,7 @@ class VerticesOrderResponse(BaseModel):
 
 
 class Log(TypedDict):
-    message: str
+    message: Union[dict, str]
     type: str
 
 
diff --git a/src/backend/base/langflow/graph/utils.py b/src/backend/base/langflow/graph/utils.py
index 795e8f02c..f6a574114 100644
--- a/src/backend/base/langflow/graph/utils.py
+++ b/src/backend/base/langflow/graph/utils.py
@@ -19,6 +19,7 @@ class UnbuiltResult:
 class ArtifactType(str, Enum):
     TEXT = "text"
     RECORD = "record"
+    OBJECT = "object"
     UNKNOWN = "unknown"
 
 
@@ -60,16 +61,16 @@ def serialize_field(value):
     return value
 
 
-def get_artifact_type(build_result: Any) -> str:
-    result = None
-    match build_result:
+def get_artifact_type(value: Any) -> str:
+    result = ArtifactType.UNKNOWN
+    match value:
         case Record():
             result = ArtifactType.RECORD
 
         case str():
             result = ArtifactType.TEXT
 
-        case _:
-            result = ArtifactType.UNKNOWN
+        case dict():
+            result = ArtifactType.OBJECT
 
     return result.value