jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

add temporary fix for message["type"]

Browse source
This commit is contained in:
Rodrigo 2024-06-09 23:08:15 -03:00
commit 4de1a3afd9
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/graph/schema.py
View file

@ -31,6 +31,11 @@ class ResultData(BaseModel):
if not values.get("logs") and values.get("artifacts"): if not values.get("logs") and values.get("artifacts"):
# Build the log from the artifacts # Build the log from the artifacts
message = values["artifacts"] message = values["artifacts"]
# ! Temporary fix
if not isinstance(message, dict):
message = {"message": message}
if "stream_url" in message and "type" in message: if "stream_url" in message and "type" in message:
stream_url = StreamURL(location=message["stream_url"]) stream_url = StreamURL(location=message["stream_url"])
values["logs"] = [Log(message=stream_url, type=message["type"])] values["logs"] = [Log(message=stream_url, type=message["type"])]