Skip to content

Fix pulling AgentOperator output for list, marker and function types - #74729

Open
FrankYang0529 wants to merge 3 commits into
apache:mainfrom
FrankYang0529:common-ai-agent-unregistered-model-as-dict
Open

FrankYang0529 wants to merge 3 commits into
apache:mainfrom
FrankYang0529:common-ai-agent-unregistered-model-as-dict

Conversation

@FrankYang0529

Copy link
Copy Markdown
Member

Follow-up for #74537 (review).

  • Move the condition that Fix agent HITL review for list, marker and function output types #74537 added to the HITL review path into a new method, _convert_output_for_xcom. Both returns of _complete_run now call this method. Without review, AgentOperator now also pushes a model as a dict when the worker would not register the model's class, for example with output_type=[Summary, int].
  • Rename test_execute_with_hitl_rehydrates_base_model to test_execute_with_hitl_returns_the_model_instance and remove its outdated comment about the JSON string.

Was generative AI tooling used to co-author this PR?
  • Yes - Claude Code

Comment thread providers/common/ai/src/airflow/providers/common/ai/operators/agent.py Outdated
Comment thread providers/common/ai/src/airflow/providers/common/ai/operators/agent.py Outdated
@FrankYang0529
FrankYang0529 force-pushed the common-ai-agent-unregistered-model-as-dict branch from 4635497 to f1806a4 Compare October 16, 2026 03:30
Comment thread providers/common/ai/src/airflow/providers/common/ai/operators/agent.py Outdated
Comment thread providers/common/ai/docs/structured_output.rst
Comment thread providers/common/ai/tests/unit/common/ai/operators/test_agent.py
@FrankYang0529
FrankYang0529 force-pushed the common-ai-agent-unregistered-model-as-dict branch from f1806a4 to f5d341a Compare October 20, 2026 04:15
@FrankYang0529
FrankYang0529 force-pushed the common-ai-agent-unregistered-model-as-dict branch from f5d341a to edf86d0 Compare October 20, 2026 10:35

This branch has not been deployed

No deployments
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.

2 participants