This commit is contained in:
italojohnny 2024-06-14 16:52:49 -03:00
commit 09c21fca06
3 changed files with 4 additions and 3 deletions

View file

@ -187,7 +187,8 @@ async def build_vertex(
logger.exception(f"Error building vertex: {exc}")
params = format_exception_message(exc)
valid = False
logs = {vertex.outputs[0]["name"]: [Log(message=params, type="error")]}
output_label = vertex.outputs[0]["name"] if vertex.outputs else "output"
logs = {output_label: [Log(message=params, type="error")]}
result_data_response = ResultDataResponse(results={})
artifacts = {}
# If there's an error building the vertex

View file

@ -245,7 +245,7 @@ class VerticesOrderResponse(BaseModel):
class ResultDataResponse(BaseModel):
results: Optional[Any] = Field(default_factory=dict)
logs: dict[str, Log] = Field(default_factory=dict)
logs: dict[str, List[Log]] = Field(default_factory=dict)
message: Optional[Any] = Field(default_factory=dict)
artifacts: Optional[Any] = Field(default_factory=dict)
timedelta: Optional[float] = None

View file

@ -29,7 +29,7 @@ def build_logs_from_artifacts(artifacts: dict) -> dict:
elif _type:
log = Log(message=message, type=_type)
logs[key] = log
logs[key] = [log]
return logs