Include downloaded files in the task block outputs within workflows so subsequent blocks can use them (#1797)

This commit is contained in:
Shuchang Zheng 2025-02-20 01:19:03 -08:00 committed by GitHub
parent 367473f930
commit 167f219a3e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 50 additions and 18 deletions

View file

@ -352,15 +352,17 @@ class TaskOutput(BaseModel):
extracted_information: list | dict[str, Any] | str | None = None
failure_reason: str | None = None
errors: list[dict[str, Any]] = []
downloaded_file_urls: list[str] | None = None
@staticmethod
def from_task(task: Task) -> TaskOutput:
def from_task(task: Task, downloaded_file_urls: list[str] | None = None) -> TaskOutput:
return TaskOutput(
task_id=task.task_id,
status=task.status,
extracted_information=task.extracted_information,
failure_reason=task.failure_reason,
errors=task.errors,
downloaded_file_urls=downloaded_file_urls,
)