Ykeremy/workflow execution updates (#314)

This commit is contained in:
Kerem Yilmaz
2024-05-14 00:35:37 -07:00
committed by GitHub
parent 1953874717
commit a106452034
4 changed files with 71 additions and 14 deletions

View File

@@ -208,5 +208,23 @@ class TaskResponse(BaseModel):
errors: list[dict[str, Any]] = []
class TaskOutput(BaseModel):
task_id: str
status: TaskStatus
extracted_information: list | dict[str, Any] | str | None = None
failure_reason: str | None = None
errors: list[dict[str, Any]] = []
@staticmethod
def from_task(task: Task) -> TaskOutput:
return TaskOutput(
task_id=task.task_id,
status=task.status,
extracted_information=task.extracted_information,
failure_reason=task.failure_reason,
errors=task.errors,
)
class CreateTaskResponse(BaseModel):
task_id: str