add temporary fix for message["type"]

This commit is contained in:
Rodrigo 2024-06-09 23:08:15 -03:00
commit 4de1a3afd9

View file

@ -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"])]