refactor: Simplify build_logs method in schema.py

This commit is contained in:
ogabrielluiz 2024-06-18 11:52:33 -03:00
commit e1f96dfb1f

View file

@ -91,9 +91,8 @@ def build_logs(vertex, result) -> dict:
case LogType.MESSAGE if hasattr(message, "message"):
message = message.message
case LogType.UNKNOWN if message is None:
case LogType.UNKNOWN:
message = ""
name = output.get("name", f"output_{index}")
logs |= {name: Log(message=message, type=_type).model_dump()}