Add aggregated extracted information in workflow run status response (#1836)

Co-authored-by: Muhammed Salih Altun <muhammedsalihaltun@gmail.com>
This commit is contained in:
Shuchang Zheng 2025-02-25 11:59:22 -08:00 committed by GitHub
parent 71307ce1ec
commit 13213d4523
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1032,6 +1032,12 @@ class WorkflowService:
outputs = None
if output_parameter_tuples:
outputs = {output_parameter.key: output.value for output_parameter, output in output_parameter_tuples}
extracted_information = {
output_parameter.key: output.value["extracted_information"]
for output_parameter, output in output_parameter_tuples
if isinstance(output.value, dict) and output.value["extracted_information"] is not None
}
outputs["extracted_information"] = extracted_information
total_steps = None
total_cost = None