Skip to content

Record OpenAI Responses token usage and response id in XCom - #72151

Open
Lee-W wants to merge 1 commit into
apache:mainfrom
astronomer:openai-response-usage-xcom
Open

Record OpenAI Responses token usage and response id in XCom#72151
Lee-W wants to merge 1 commit into
apache:mainfrom
astronomer:openai-response-usage-xcom

Conversation

@Lee-W

@Lee-W Lee-W commented Aug 27, 2026

Copy link
Copy Markdown
Member

OpenAIResponseOperator returned only the aggregated output text, so the response id needed to chain a follow-up call and the token counts that are the run's only cost signal were discarded on every task. Both are now available to downstream tasks without dropping to the hook. The counts are tokens, not money: the OpenAI response carries no cost field.


Was generative AI tooling used to co-author this PR?
  • Yes (please specify the tool below)

Generated-by: [Claude] following the guidelines


  • Read the Pull Request Guidelines for more information. Note: commit author/co-author name and email in commits become permanently public when merged.
  • For fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
  • When adding dependency, check compliance with the ASF 3rd Party License Policy.
  • For significant user-facing changes create newsfragment: {pr_number}.significant.rst, in airflow-core/newsfragments. You can add this file in a follow-up commit after the PR is created so you know the PR number.

OpenAIResponseOperator returned only the aggregated output text, so the response
id needed to chain a follow-up call and the token counts that are the run's only
cost signal were discarded on every task. Both are now available to downstream
tasks without dropping to the hook. The counts are tokens, not money: the OpenAI
response carries no cost field.
@Lee-W
Lee-W force-pushed the openai-response-usage-xcom branch from 432b256 to cd7d399 Compare August 27, 2026 14:17
@Lee-W
Lee-W marked this pull request as ready for review August 27, 2026 14:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant