From 4de1a3afd9d3574b22772f4ff33b0ca55ca4573d Mon Sep 17 00:00:00 2001 From: Rodrigo Date: Sun, 9 Jun 2024 23:08:15 -0300 Subject: [PATCH] add temporary fix for message["type"] --- src/backend/base/langflow/graph/schema.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/backend/base/langflow/graph/schema.py b/src/backend/base/langflow/graph/schema.py index 72b810555..766575364 100644 --- a/src/backend/base/langflow/graph/schema.py +++ b/src/backend/base/langflow/graph/schema.py @@ -31,6 +31,11 @@ class ResultData(BaseModel): if not values.get("logs") and values.get("artifacts"): # Build the log from the artifacts message = values["artifacts"] + + # ! Temporary fix + if not isinstance(message, dict): + message = {"message": message} + if "stream_url" in message and "type" in message: stream_url = StreamURL(location=message["stream_url"]) values["logs"] = [Log(message=stream_url, type=message["type"])]