From 122010f1bf22f0172e7077860c10eb6a29fa5d59 Mon Sep 17 00:00:00 2001 From: HeZzz Date: Wed, 2 Sep 2026 01:10:53 +0800 Subject: [PATCH 1/2] Python: preserve response update metadata in WorkflowAgent forwarding Forward finish_reason, continuation_token, agent_id and a shallow copy of additional_properties when WorkflowAgent reconstructs AgentResponseUpdate objects from workflow events, matching the documented as-is forwarding contract. --- .../core/agent_framework/_workflows/_agent.py | 4 ++ .../tests/workflow/test_workflow_agent.py | 38 +++++++++++++++++++ 2 files changed, 42 insertions(+) diff --git a/python/packages/core/agent_framework/_workflows/_agent.py b/python/packages/core/agent_framework/_workflows/_agent.py index 353a1a2efb..2cd169453a 100644 --- a/python/packages/core/agent_framework/_workflows/_agent.py +++ b/python/packages/core/agent_framework/_workflows/_agent.py @@ -614,9 +614,13 @@ def _convert_workflow_event_to_agent_response_updates( contents=list(data.contents), role=data.role, author_name=data.author_name or executor_id, + agent_id=data.agent_id, response_id=data.response_id, message_id=data.message_id, created_at=data.created_at, + finish_reason=data.finish_reason, + continuation_token=data.continuation_token, + additional_properties=dict(data.additional_properties) if data.additional_properties else None, raw_representation=data.raw_representation, ) ] diff --git a/python/packages/core/tests/workflow/test_workflow_agent.py b/python/packages/core/tests/workflow/test_workflow_agent.py index 70fd94468c..200e7ca24e 100644 --- a/python/packages/core/tests/workflow/test_workflow_agent.py +++ b/python/packages/core/tests/workflow/test_workflow_agent.py @@ -852,6 +852,44 @@ async def yielding_executor(messages: list[Message], ctx: WorkflowContext[Never, assert "first output" in texts assert "second output" in texts + async def test_workflow_as_agent_stream_preserves_response_update_metadata(self) -> None: + """Test that streaming forwards finish_reason, continuation_token and additional_properties. + + This validates the fix for issue #7952: AgentResponseUpdate metadata should be + forwarded as-is when the workflow is wrapped via .as_agent(). + """ + + @executor + async def metadata_executor(messages: list[Message], ctx: WorkflowContext[Never, AgentResponseUpdate]) -> None: # type: ignore[valid-type] + await ctx.yield_output( + AgentResponseUpdate( + contents=[Content.from_text(text="payload")], + role="assistant", + agent_id="source-agent", + response_id="source-response", + message_id="source-message", + finish_reason="stop", + continuation_token="resume-token", + additional_properties={"provider_marker": "preserve-me"}, + ) + ) + + workflow = WorkflowBuilder(start_executor=metadata_executor).build() + agent = workflow.as_agent("metadata-test-agent") + + updates: list[AgentResponseUpdate] = [] + async for update in agent.run("hello", stream=True): + updates.append(update) + + metadata_updates = [u for u in updates if u.response_id == "source-response"] + assert len(metadata_updates) == 1 + update = metadata_updates[0] + assert update.text == "payload" + assert update.agent_id == "source-agent" + assert update.finish_reason == "stop" + assert update.continuation_token == "resume-token" + assert update.additional_properties == {"provider_marker": "preserve-me"} + async def test_workflow_as_agent_yield_output_with_content_types(self) -> None: """Test that yield_output preserves different content types (Content, Content, etc.).""" From 697a0fc8176d5c2a1a2372d75ab7e46c618e5937 Mon Sep 17 00:00:00 2001 From: HeZzz Date: Wed, 2 Sep 2026 01:20:37 +0800 Subject: [PATCH 2/2] Python: keep empty additional_properties dict when forwarding Use an explicit None check instead of a truthiness check so an explicitly empty additional_properties dict is not converted to None, preserving the forwarding contract. --- .../core/agent_framework/_workflows/_agent.py | 2 +- .../tests/workflow/test_workflow_agent.py | 25 +++++++++++++++++++ 2 files changed, 26 insertions(+), 1 deletion(-) diff --git a/python/packages/core/agent_framework/_workflows/_agent.py b/python/packages/core/agent_framework/_workflows/_agent.py index 2cd169453a..750ab17bbe 100644 --- a/python/packages/core/agent_framework/_workflows/_agent.py +++ b/python/packages/core/agent_framework/_workflows/_agent.py @@ -620,7 +620,7 @@ def _convert_workflow_event_to_agent_response_updates( created_at=data.created_at, finish_reason=data.finish_reason, continuation_token=data.continuation_token, - additional_properties=dict(data.additional_properties) if data.additional_properties else None, + additional_properties=dict(data.additional_properties) if data.additional_properties is not None else None, raw_representation=data.raw_representation, ) ] diff --git a/python/packages/core/tests/workflow/test_workflow_agent.py b/python/packages/core/tests/workflow/test_workflow_agent.py index 200e7ca24e..87eac2b700 100644 --- a/python/packages/core/tests/workflow/test_workflow_agent.py +++ b/python/packages/core/tests/workflow/test_workflow_agent.py @@ -890,6 +890,31 @@ async def metadata_executor(messages: list[Message], ctx: WorkflowContext[Never, assert update.continuation_token == "resume-token" assert update.additional_properties == {"provider_marker": "preserve-me"} + async def test_workflow_as_agent_stream_preserves_empty_additional_properties(self) -> None: + """Test that an explicitly empty additional_properties dict is not converted to None.""" + + @executor + async def empty_props_executor(messages: list[Message], ctx: WorkflowContext[Never, AgentResponseUpdate]) -> None: # type: ignore[valid-type] + await ctx.yield_output( + AgentResponseUpdate( + contents=[Content.from_text(text="payload")], + role="assistant", + response_id="empty-props-response", + additional_properties={}, + ) + ) + + workflow = WorkflowBuilder(start_executor=empty_props_executor).build() + agent = workflow.as_agent("empty-props-test-agent") + + updates: list[AgentResponseUpdate] = [] + async for update in agent.run("hello", stream=True): + updates.append(update) + + forwarded = [u for u in updates if u.response_id == "empty-props-response"] + assert len(forwarded) == 1 + assert forwarded[0].additional_properties == {} + async def test_workflow_as_agent_yield_output_with_content_types(self) -> None: """Test that yield_output preserves different content types (Content, Content, etc.)."""